Ultra-QuickSort poj-2299

原題:POJ-2299


這個題目是一個求逆序數的題目

可以用樹狀數組、線段樹、歸併排序來做


樹狀數組、線段樹只要先求和再更行就行了,插入法。在做的時候先要對序列進行離散,不然數字太大了


歸併排序的做法是在合併的時候求逆序,假如有序列A和序列B,當Ai<Bi時,那麼Bi的逆序數爲 (length(A)-i+1),然後一次求逆序數再加起來就行了


以下代碼是渣渣手寫的,應該很容易看懂QAQ,結果sum要定義爲 long long !


樹狀數組:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

struct data{
  int d,t;
}p[500050];
int cmp1(data a,data b){
   return a.d<b.d;
}
int cmp2(data a,data b){
   return a.t<b.t;
}
int c[500050];
void updata(int x,int v,int n){
    for(int i=x;i<n+10;i+=(i&(-i)))
       c[i]+=v;
}
int getsum(int x){
   int sum=0;
   for(int i=x;i>0;i-=(i&(-i)))
    sum+=c[i];
   return sum;
}
int main() {
    //freopen("C:\\Users\\adminjust\\Desktop\\test.txt","w",stdout);
    int n;
    while(cin>>n&&n){
        for(int i=1;i<=n;i++)
         {
             scanf("%d",&p[i].d);
             p[i].t=i;
         }
         sort(p+1,p+n+1,cmp1);
         for(int i=1;i<=n;i++)
            p[i].d=i;
         sort(p+1,p+n+1,cmp2);
         long long sum=0;
         memset(c,0,sizeof(c));
         for(int i=1;i<=n;i++){
           sum+=(i-1-getsum(p[i].d));
           updata(p[i].d,1,n);
         }
         cout<<sum<<endl;
    }
    return 0;
}

線段樹:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

struct data {
    int d,t;
} p[500050];
struct node {
    int l,r,v;
} q[50005000];
int cmp1(data a,data b) {
    return a.d<b.d;
}
int cmp2(data a,data b) {
    return a.t<b.t;
}
int c[50005000];

void build(int t,int l,int r) {
    if(l==r) {
        c[t]=0;
        return ;
    }
    int mid=(l+r)/2;
    build(t*2,l,mid);
    build(t*2+1,mid+1,r);
    c[t]=0;
}
void updata(int t,int x,int l,int r) {
    c[t]++;
    if(l==r&&l==x)
        return ;
    int mid=(l+r)/2;
    if(x<=mid)
        updata(t*2,x,l,mid);
    else
        updata(t*2+1,x,mid+1,r);
}
int getsum(int l,int r,int L,int R,int t) {
    if(l<=L&&r>=R)
        return c[t];
    int mid=(L+R)/2;
    if(r<=mid)
        return getsum(l,r,L,mid,t*2);
    else if(l>mid)
        return getsum(l,r,mid+1,R,t*2+1);
    else return getsum(l,mid,L,mid,t*2)+getsum(mid+1,r,mid+1,R,t*2+1);
}
int main() {
    //freopen("C:\\Users\\adminjust\\Desktop\\test.txt","w",stdout);
    int n;
    while(cin>>n&&n) {
        for(int i=1; i<=n; i++) {
            scanf("%d",&p[i].d);
            p[i].t=i;
        }
        sort(p+1,p+n+1,cmp1);
        for(int i=1; i<=n; i++)
            p[i].d=i;
        sort(p+1,p+n+1,cmp2);
        long long sum=0;
        memset(c,0,sizeof(c));
        build(1,1,n);
        for(int i=1; i<=n; i++) {
            sum+=(i-1-getsum(1,p[i].d,1,n,1));
            updata(1,p[i].d,1,n);
        }
        cout<<sum<<endl;
    }
    return 0;
}

歸併:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int c[500050];
long long sum;
void lianjie(int l,int r,int L,int R) {
    int *temp=new int[R+l+10];
    int i,j,p;
    for(i=l,j=L,p=0; i<=r&&j<=R; p++) {
        if(c[i]>c[j]) {
            sum+=(r-i+1);
            temp[p]=c[j++];
        } else temp[p]=c[i++];
    }
    while(i<=r)temp[p++]=c[i++];
    while(j<=R)temp[p++]=c[j++];
    for(int k=l; k<=R; k++)
        c[k]=temp[k-l];
}
void MergeSort(int l,int r) {
    if(l==r)
        return ;
    int mid=(l+r)/2;
    MergeSort(l,mid);
    MergeSort(mid+1,r);
    lianjie(l,mid,mid+1,r);
}
int main() {
  //  freopen("C:\\Users\\adminjust\\Desktop\\test.txt","w",stdout);
    int n;
    while(cin>>n&&n) {
        for(int i=1; i<=n; i++)
            scanf("%d",&c[i]);
        sum=0;
        MergeSort(1,n);
        cout<<sum<<endl;
    }
    return 0;
}






發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章