分治法 求 逆序数

这是在BUPT acm上面的一个题。一开始写了个o(n^2)的算法，到程序里面跑，结果超时了，后来想办法写了两个o(n)的程序，后来发现逻辑是错的，想的不周到。网上查了一下发现了一种分治法可以以o(nlogn)的时间复杂度计算出逆序数。自己写了一个，记录一下。

 

1. 超时的程序

 

// 求逆序数 #include <stdio.h> #include <stdlib.h> int main() { int n; int i,j; int* a; int cnt=0; scanf("%d",&n); a = (int*) calloc(n, sizeof(int)); for(i=0; i<n; ++i) { scanf("%d",&a[i]); } for(i=0; i<n; ++i) for(j=i; j<n; ++j) { if(a[i] > a[j]) ++cnt; } printf("%d/n",cnt); system("pause"); return 0; } 

 

2,3. 自己想的方法，错误的~

 

// 求逆序数 #include <stdio.h> #include <stdlib.h> #include <string.h> int main() { int n; int i,j; int* a,*b; scanf("%d",&n); a = (int*) calloc(n, sizeof(int)); b = (int*) calloc(n, sizeof(int)); memset(b,0,n); for(i=0; i<n; ++i) { scanf("%d",&a[i]); } for(i=1; i<n; ++i) { if(a[i] < a[i-1]) { b[i] = 2*b[i-1]+1; } else { b[i] = b[i-1]; } } printf("%d/n",b[n-1]); // // system("pause"); return 0; } 

 

// 求逆序数 #include <stdio.h> #include <stdlib.h> #include <string.h> int main() { int n; int i,j; int* a,*b; scanf("%d",&n); a = (int*) calloc(n, sizeof(int)); b = (int*) calloc(n, sizeof(int)); memset(b,0,n); for(i=0; i<n; ++i) { scanf("%d",&a[i]); } for(i=1; i<n; ++i) { if(a[i] < a[i-1]) { b[i] = 2*b[i-1]+1; } else { // 去找比他大的那个，如果实在找不到在这样算 for(j=i-1;j!=0;--j ) { if(a[j]>a[i]) { b[i] = b[i-1]+b[j]+1; break;} else if(a[j] == a[i]) { b[i] = b[i-1]+b[j]; break;} } if(j==0) b[i] = b[i-1]; } } printf("%d/n",b[n-1]); // system("pause"); return 0; }  

 

4. 分治的程序，又WA了，找找原因（程序里面本来是一个memcpy函数，但是好像没起作用，不知道为什么，就拿for循环改了）

 

// o(nlogn) 逆序数 #include <stdio.h> #include <string.h> #include <stdlib.h> int Calc(int* a, int* b, int first, int last) { if(first >= last) return 0; // 一个元素的情况下没有逆序数 int i,j,k=0,mid,res=0; k=i=first; mid=(first+last)/2; j=mid+1; res=Calc(a,b,i,mid)+Calc(a,b,j,last); // 递归计算两边的逆序数 // 计算两边相关联的数据的逆序数 while((i<=mid) && (j<=last)) { if(a[i]<a[j]) b[k++]=a[i++],res+=j-mid-1; else b[k++]=a[j++]; // 是不是没有考虑a[i]=a[j]的问题 } while(i<=mid) b[k++]=a[i++],res+=j-mid-1; while(j<=last) b[k++]=a[j++]; // memcpy(a+first,b+first,sizeof(int)*(first-last+1)); for(i=first; i<=last;++i) a[i]=b[i]; return res; } int main() { int n; scanf("%d",&n); int* a=(int*) calloc(n, sizeof(int)); int* b=(int*) calloc(n, sizeof(int)); int i=0; for(i=0;i<n;++i) { scanf("%d",(a+i)); } printf("%d/n",Calc(a,b,0,n-1)); // system("pause"); return 0; }  

 

5. 按理说第一个程序虽然超时，但是结果应该是没有问题的。这样的话程序 4 出问题的原因就是没考虑数列中存在重复元素的情况。我把程序4修改后

// o(nlogn) 逆序数 #include <stdio.h> #include <string.h> #include <stdlib.h> int Calc(int* a, int* b, int first, int last) { if(first >= last) return 0; // 一个元素的情况下没有逆序数 int i,j,k=0,mid,res=0; k=i=first; mid=(first+last)/2; j=mid+1; res=Calc(a,b,i,mid)+Calc(a,b,j,last); // 递归计算两边的逆序数 // 计算两边相关联的数据的逆序数 while((i<=mid) && (j<=last)) { if(a[i]<=a[j]) b[k++]=a[i++],res+=j-mid-1; else b[k++]=a[j++]; // 是不是没有考虑a[i]=a[j]的问题 } while(i<=mid) b[k++]=a[i++],res+=j-mid-1; while(j<=last) b[k++]=a[j++]; // memcpy(a+first,b+first,sizeof(int)*(first-last+1)); for(i=first; i<=last;++i) a[i]=b[i]; return res; } int main() { int n; scanf("%d",&n); int* a=(int*) calloc(n, sizeof(int)); int* b=(int*) calloc(n, sizeof(int)); int i=0; for(i=0;i<n;++i) { scanf("%d",(a+i)); } printf("%d/n",Calc(a,b,0,n-1)); system("pause"); return 0; }  

 

此时已经可以得到和程序1一样的结果了，但是答案还是WA，why？

 

 

---

 

日啊，最后终于AC了，原来逆序数太大，要用long long才能存的下！！！！

 

// o(nlogn) 逆序数 #include <stdio.h> #include <string.h> #include <stdlib.h> long long Calc(int* a, int* b, int first, int last) { if(first >= last) return 0; // 一个元素的情况下没有逆序数 int i,j,k=0,mid; k=i=first; long long res=0; mid=(first+last)/2; j=mid+1; res=Calc(a,b,i,mid)+Calc(a,b,j,last); // 递归计算两边的逆序数 // 计算两边相关联的数据的逆序数 while((i<=mid) && (j<=last)) { if(a[i]<=a[j]) b[k++]=a[i++],res+=j-mid-1; else b[k++]=a[j++]; // 是不是没有考虑a[i]=a[j]的问题 } while(i<=mid) b[k++]=a[i++],res+=j-mid-1; while(j<=last) b[k++]=a[j++]; // memcpy(a+first,b+first,sizeof(int)*(first-last+1)); for(i=first; i<=last;++i) a[i]=b[i]; return res; } int main() { int n; scanf("%d",&n); int* a=(int*) calloc(n, sizeof(int)); int* b=(int*) calloc(n, sizeof(int)); int i=0; for(i=0;i<n;++i) { scanf("%d",(a+i)); } printf("%lld/n",Calc(a,b,0,n-1)); // system("pause"); return 0; }  

 

感动ing~