分析题目可知
min{∑(ai-bi)^2 (1<=i<=n)}
展开:min{∑(ai^2+bi^2-2*ai*bi)}=min{∑ai^2+∑bi^2-∑2*ai*bi}
仔细观察,可以发现∑ai^2和∑bi^2的值是不会变的,所以只能在∑2*ai*bi上做文章。
为使得和最小,那么∑2*ai*bi要最大,本题的模型就转变为max{∑ai*bi}。
让b数组中第i小的数和a数组中第i小的数在同一个位置一定最优
a<b c<d
根据以上猜测,则ac+bd一定是最大的。利用反证法。
若ac+bd不是最大的,那么一定有比它更大的,只有ad+bc
ac+bd<ad+bc
ac-ad<bc-bd
a*(c-d)<b*(c-d)//c-d<0
a>b(???)
得出矛盾。
#include <bits/stdc++.h>
using namespace std;
int n,ans,c[1000001],t[1000001];
pair<int,int> a[1000001],b[1000001];
int que(int k)
{ int ans=0;while(k>0) ans+=t[k],k-=k&-k;return ans;}
void add(int k)
{ while(k<=n) ++t[k],k+=k&-k;}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i].first),a[i].second=i;
for(int i=1;i<=n;i++)
scanf("%d",&b[i].first),b[i].second=i;
sort(a+1,a+n+1);sort(b+1,b+n+1);//1.讲两数组排序
for(int i=1;i<=n;i++)
c[a[i].second]=b[i].second;//2.开辟一个新的数组c,c的下标为第一个数组排序后的标号,c的值为第二个数组排序后的标号。
for(int i=1;i<=n;i++)
//ans+=que(n-c[i]),add(n-c[i]+1),ans%=99999997;
add(c[i]),ans+=i-que(c[i]),ans%=99999997;//求逆序数。
printf("%d",ans);
return 0;
}
例如
输入 数组一:1 3 4 2 数组一:1 2 3 4
标号 :1 2 3 4 排序后----------> 标号 :1 4 2 3 ------------------> c1=1 c4=3 c2=4 c3=2
数组二:1 7 2 4 数组二:1 2 4 7 c1=1 c2=4 c3=2 c4=3
标号 :1 2 3 4 标号 :1 3 4 2
此时的问题是如何使1 2 3 4(c的下标) 排成1 4 2 3(c的值)的顺序 即为求逆序对