(match.cpp/c/pas)
【问题描述】
涵涵有两盒火柴,每盒装有 n 根火柴,每根火柴都有一个高度。现在将每盒中的火柴各自排成一列,同一列火柴的高度互不相同,两列火柴之间的距离定义为:,其中 ai 表示第一列火柴中第 i 个火柴的高度,bi 表示第二列火柴中第 i 个火柴的高度。
每列火柴中相邻两根火柴的位置都可以交换,请你通过交换使得两列火柴之间的距离最小。请问得到这个最小的距离,最少需要交换多少次?如果这个数字太大,请输出这个最小交换次数对 99,999,997 取模的结果。
【输入】
输入文件为 match.in。
共三行,第一行包含一个整数 n,表示每盒中火柴的数目。
第二行有 n 个整数,每两个整数之间用一个空格隔开,表示第一列火柴的高度。
第三行有 n 个整数,每两个整数之间用一个空格隔开,表示第二列火柴的高度。
【输出】
输出文件为 match.out。
输出共一行,包含一个整数,表示最少交换次数对99,999,997取模的结果。
【输入输出样例1】
match.in |
match.out |
4 2 3 1 4 3 2 1 4 |
1 |
【输入输出样例说明】
最小距离是 0,最少需要交换 1 次,比如:交换第 1 列的前 2 根火柴或者交换第 2 列的前 2 根火柴。
【输入输出样例2】
match.in |
match.out |
4 1 3 4 2 1 7 2 4 |
2 |
【输入输出样例说明】
最小距离是 10,最少需要交换 2 次,比如:交换第 1 列的中间 2 根火柴的位置,再交换第 2 列中后 2 根火柴的位置。
【数据范围】
对于 10%的数据, 1 ≤ n ≤ 10;
对于 30%的数据,1 ≤ n ≤ 100;
对于 60%的数据,1 ≤ n ≤ 1,000;
对于 100%的数据,1 ≤ n ≤ 100,000,0 ≤火柴高度≤ 231 − 1。
【思路】
很容易想到:如果距离最小,那么每对ab对应的排名肯定是相同的。
将两个序列排序,得出关于原来标号的新序列A’B’,此时有数的排名对应但是标号不同,问题转化为从原来的标号相同转化到标号不同最少交换,其实就是从A’转化到B’的最少交换数。按照B’将A’排序后,问题进一步转化为将A’排序所需要的最少交换数。
每个逆序对都必须进行一次交换且这种方法交换次数最少,所以答案就是A’的逆序对个数。
【代码】
1 #include<iostream> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 6 const int MOD = 99999997; 7 const int maxn = 100000; 8 struct Node{ 9 int v,pos; 10 bool operator <(const Node& rhs) const{ 11 return v<rhs.v || (v==rhs.v && pos<rhs.pos); 12 } 13 }A[maxn],B[maxn]; 14 int U[maxn],T[maxn]; 15 16 int ans=0; 17 void merge_sort(int* A,int x,int y,int* T){ //[x,y) 18 if(y-x<=1) return; 19 int m=x+(y-x)/2; 20 int p=x,q=m,i=x; 21 merge_sort(A,x,m,T); 22 merge_sort(A,m,y,T); 23 while(p<m || q<y) { 24 if(q>=y||(p<m && A[p]<=A[q])) T[i++]=A[p++]; 25 else{ 26 T[i++]=A[q++]; 27 ans=(int)((long long)ans+m-p)%MOD; 28 } 29 } 30 for(int i=x;i<y;i++) A[i]=T[i]; 31 } 32 33 int n,maxlongint=-(1<<30); 34 int C[maxn]; 35 inline int lowbit(int x) {return x&(-x); } 36 inline void add(int u,int d) { 37 while(u<=maxlongint+1000) { 38 C[u]+=d; 39 u+=lowbit(u); 40 } 41 } 42 inline int sum(int u) { 43 int res=0; 44 while(u>0) { 45 res+= C[u]; 46 u-=lowbit(u); 47 } 48 return res; 49 } 50 51 int main() { 52 ios::sync_with_stdio(false); 53 cin>>n; 54 int U[maxn]; 55 for(int i=0;i<n;i++) { cin>>A[i].v; A[i].pos=i;} 56 for(int i=0;i<n;i++) { cin>>B[i].v; B[i].pos=i;} 57 58 sort(A,A+n); sort(B,B+n); 59 60 for(int i=0;i<n;i++) { U[A[i].pos]=B[i].pos+1; maxlongint=max(maxlongint,U[A[i].pos]); } 61 merge_sort(U,0,n,T); 62 /* 63 for(int j=0;j<n;j++) { 64 ans=(ans+j-sum(U[j]))%MOD; 65 add(U[j],1); 66 } 67 */ 68 cout<<ans; 69 return 0; 70 }