P1966 火柴排队

题目描述

涵涵有两盒火柴，每盒装有 n 根火柴，每根火柴都有一个高度。 现在将每盒中的火柴各自排成一列， 同一列火柴的高度互不相同， 两列火柴之间的距离定义为：∑(ai​−bi​)^2

其中ai​ 表示第一列火柴中第 i个火柴的高度，bi​表示第二列火柴中第 i 个火柴的高度。

每列火柴中相邻两根火柴的位置都可以交换，请你通过交换使得两列火柴之间的距离最小。请问得到这个最小的距离，最少需要交换多少次？如果这个数字太大，请输出这个最小交换次数对 99,999,997取模的结果。

输入格式

共三行，第一行包含一个整数n ，表示每盒中火柴的数目。

第二行有n个整数，每两个整数之间用一个空格隔开，表示第一列火柴的高度。

第三行有 n 个整数，每两个整数之间用一个空格隔开，表示第二列火柴的高度。

输出格式

一个整数，表示最少交换次数对 99,999,997 取模的结果。

输入输出样例

输入 #1

4
2 3 1 4
3 2 1 4

输出 #1

1

输入 #2

4
1 3 4 2
1 7 2 4

输出 #2

2

说明/提示

【输入输出样例说明1】

最小距离是0，最少需要交换 1 次，比如：交换第1列的前2根火柴或者交换第 2 列的前2根火柴。

【输入输出样例说明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≤火柴高度≤maxlongint

思路

本蒟蒻犯了两个“令人发指”的错误：

·N和Mod值赋反了

·cmp函数里return没打。。。

代码：

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

const int N=100010;
const int Mod=99999997;

long long ans;
int n,c[N],d[N];

struct no {
	int x,num;
} a[N],b[N];

bool cmp(no a,no b) {
	if(a.x==b.x)
		return a.num<b.num;
	else
		return a.x<b.x;
}

void solve(int l,int r) {
	if(r<=l)
		return;
	int mid=(l+r)/2;
	int i=l,j=mid+1,k=l;
	solve(l,mid);
	solve(mid+1,r);
	while(i<=mid&&j<=r)
		if(c[i]<=c[j])
			d[k++]=c[j++];
		else {
			d[k++]=c[i++];
			ans=(ans+r-j+1)%Mod;
		}
	while(i<=mid)
		d[k++]=c[i++];
	while(j<=r)
		d[k++]=c[j++];
	for(i=l; i<=r; i++)
		c[i]=d[i];
}

int main () {
	scanf("%d",&n);
	for(int i=1; i<=n; i++) {
		scanf("%d",&a[i].x);
		a[i].num=i;
	}
	for(int i=1; i<=n; i++) {
		scanf("%d",&b[i].x);
		b[i].num=i;
	}
	sort(a+1,a+1+n,cmp);
	sort(b+1,b+1+n,cmp);
	for(int i=1; i<=n; i++)
		c[b[i].num]=a[i].num;
	solve(1,n);
	printf("%lld\n",ans);
	return 0;
}