遞歸和分治:poj2299
/*
總體思想使用分治法,進行歸併排序,求出每次合併的逆序總數累加。說明一下:按照算導中方法寫的程序
有可能會超時,因爲每次merge都重新分配兩個數組,且需單獨循環計算逆序,操作較多。另外,最後結果sum使用long long,
在最壞情況下是n^2級的,n<500000,一個long(4B,2^32) 不夠 。 3692K 375MS
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
const int N = 500005;
long long sum;
int vec[N],back[N];
void merge(int s,int m,int e){
int i = s, j = m + 1, k = s;
while (i <= m && j <= e) {
if (vec[i] > vec[j]) {
sum += m - i + 1; //逆序數累加
back[k++] = vec[j++];
} else {
back[k++] = vec[i++];
}
}
while (i <= m)
back[k++] = vec[i++];
while (j <= e)
back[k++] = vec[j++];
memcpy(vec + s, back + s, sizeof(*back) * (e - s + 1));
}
void merge_sort(int p,int r){
if (p<r){
int q=(p+r)/2;
merge_sort(p,q);
merge_sort(q+1,r);
merge(p,q,r);
}
}
int main(){
int i,n;
scanf("%d",&n);
while(n!=0){
sum = 0;
for(i=1;i<=n;i++){
scanf("%d",vec+i);
}
merge_sort(1,n);
printf("%lld\n",sum);
scanf("%d",&n);
}
return 0;
}
總體思想使用分治法,進行歸併排序,求出每次合併的逆序總數累加。說明一下:按照算導中方法寫的程序
有可能會超時,因爲每次merge都重新分配兩個數組,且需單獨循環計算逆序,操作較多。另外,最後結果sum使用long long,
在最壞情況下是n^2級的,n<500000,一個long(4B,2^32) 不夠 。 3692K 375MS
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
const int N = 500005;
long long sum;
int vec[N],back[N];
void merge(int s,int m,int e){
int i = s, j = m + 1, k = s;
while (i <= m && j <= e) {
if (vec[i] > vec[j]) {
sum += m - i + 1; //逆序數累加
back[k++] = vec[j++];
} else {
back[k++] = vec[i++];
}
}
while (i <= m)
back[k++] = vec[i++];
while (j <= e)
back[k++] = vec[j++];
memcpy(vec + s, back + s, sizeof(*back) * (e - s + 1));
}
void merge_sort(int p,int r){
if (p<r){
int q=(p+r)/2;
merge_sort(p,q);
merge_sort(q+1,r);
merge(p,q,r);
}
}
int main(){
int i,n;
scanf("%d",&n);
while(n!=0){
sum = 0;
for(i=1;i<=n;i++){
scanf("%d",vec+i);
}
merge_sort(1,n);
printf("%lld\n",sum);
scanf("%d",&n);
}
return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.