归并排序刷题

POJ 2299 超快速排序

两两交换生成有序序列  ----》 数逆序元素的个数  逆序对

mid-i+1 第一个元素的下标 i 

子序列之后要有序

在合并的时候数一数逆序对的个数 

110 = 6     6/2 = 3    右移一位 011=3  左移动 1100=12

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=500010;
int a[maxn], b[maxn];
// 不用动态的分配空间 开辟静态数组50个数组或者1w的时间差距并不大，动态数组时间差距比较大。频繁的开辟动态数组会影响算法的效率
long long cnt;

void Merge(int low, int mid, int high){
    int i=low, j=mid+1, k=low;
    while (i<=mid && j<=high) {  // 按照从小到大存放到辅助书B[]中
        if(a[i]<=a[j])
            b[k++] = a[i++];
        else{
            b[k++] = a[j++];
            cnt += (long long)(mid-i+1);
        }
    }
    while(i<=mid) b[k++]=a[i++];
    while(j<=high) b[k++]=a[j++];
    for(k=low; k<=high; k++)
        a[k] = b[k];
}

void MergeSort(int low, int high){
    if(low<high){
        int mid = (low + high) >> 1;  //右移操作比除法快
        // 分数组
        MergeSort(low, mid);
        MergeSort(mid+1, high);
        // 合并
        Merge(low, mid, high);
    }
}
int main(){
    int n;
    while (scanf("%d", &n) && n) {
        cnt = 0;
        for(int i=0; i<n; i++)
            scanf("%d", &a[i]);
        MergeSort(0, n-1);
        printf("%lld
", cnt);  // %I64d 杭电不让用lld可以使用%I64d代替
    }
    return 0;
}

HDU 1031

https://vjudge.net/problem/HDU-1031

对满意度之和排序递减排序（从大到小），满意度相等按id递增（从小到大），取前k个，按照缩影递减输出（从大到小），输出的是索引。

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

struct data{
    int id;
    double value;
}a[10010];

void Merge(data A[], int low, int mid, int high, bool flag){  // 合并函数
    data *B= new data[high-low+1];  // 辅助数组
    int i=low, j=mid+1, k=0;
    double x, y;
    // 按照满意度递减、按照索引递减 用flag区别
    while (i<=mid && j<=high) {
        
        if(!flag){
            x = A[i].value;
            y = A[j].value;
        }else{
            x = A[i].id;
            y = A[j].id;
        }
        
        if(x>=y)  // 保证了id小的在前
            B[k++] = A[i++];
        else
            B[k++] = A[j++];
    }
    while(i<=mid) B[k++]=A[i++];
    while(j<=high) B[k++]=A[j++];
    for(i=low, k=0; i<=high; i++, k++)
        A[i] = B[k];
    delete []B;
}

void MergeSort(data A[], int low, int high, bool flag){
    if(low < high){
        int mid = (low+high)>>1;
        MergeSort(A, low, mid, flag);
        MergeSort(A, mid+1, high, flag);
        Merge(A, low, mid, high, flag);
    }
}

int main(){
    int n, m, k;
    double x;
    while (scanf("%d%d%d", &n, &m, &k)!=EOF) {
        for(int i=0; i<m; i++){
            a[i].id = i+1;
            a[i].value = 0.0;
        }
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                scanf("%lf", &x);
                a[j].value += x;
            }
        }
        MergeSort(a, 0, m-1, 0);  // 按值非递增
        MergeSort(a, 0, k-1, 1);  // 按序号非递增 我们只要钱前k个
        for(int i=0; i<k-1; i++)
            printf("%d ", a[i].id);
        printf("%d
", a[k-1].id);
    }
    return 0;
}

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

struct data{
    int id;
    double value;
}a[10010];

bool cmp1(data a, data b){
    if(a.value == b.value)
        return a.id < b.id;
    return a.value>b.value;
}

bool cmp2(data a, data b){
    return a.id > b.id;
}

int main(){
    int n, m, k;
    double x;
    while (scanf("%d%d%d", &n, &m, &k)!=EOF) {
        for(int i=0; i<m; i++){
            a[i].id = i+1;
            a[i].value = 0.0;
        }
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                scanf("%lf", &x);
                a[j].value += x;
            }
        }
        sort(a, a+m, cmp1);  // 按值降序
        sort(a, a+k, cmp2);  // 按序号降序
        for(int i=0; i<k-1; i++)
            printf("%d ", a[i].id);
        printf("%d
", a[k-1].id);
    }
    return 0;
}