ZOJ4101 Element Swapping

The 16th Zhejiang Provincial Collegiate Programming Contest Promlem B Element Swapping

题目链接： http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=4101
题意：

题目求有多少种交换方案，使得交换数列中两个元素${\alpha }_i,{\alpha }_j(i!=j)$，有交换后得到$x_2={\sum }_1^{n}k{\alpha }_k,y_2={\sum }_1^{n}k{\alpha }_k^2$题目所给的$x_1,y_1$,相等。

解法:
记交换的元素${\alpha }_i,{\alpha }_j(i!=j)$，则经过化简可以得到条件:
①$x_1-x_2=(i-j)({\alpha }_j-{\alpha }_i)$
②$y_1-y_2=(i-j)({\alpha }_j-{\alpha }_i)({\alpha }_j+{\alpha }_i)$
其中②可以转化为$(y_1-y_2)/(x_1-x_2)={\alpha }_i+{\alpha }_j.$

现可以一边读取一边处理得到相应的$x=x_1-x_2,y=y_1-y_2$,
（1）在$x！=0$的情况下，可以得到$sum=(y_1-y_2)/(x_1-x_2)=y/x={\alpha }_i+{\alpha }_j$,。遍历数组，当前为${\alpha }_i$，则对应的$target={\alpha }_j=sum-{\alpha }_i$，通过条件①可以得到$j=i+(x_1-x_2)/({\alpha }_j-{\alpha }_i)$，这时只需要检查数组中${\alpha }_j$是否等于$target$，若等于答案加一，很容易想到这样遍历之后会计两遍（i和j换 j和i换，其中i < j），故最后答案还要除以2。
（2）在$x=0$的情况下，由①②可知$y$一定也为0，所以y不为0的情况下交换次数为0，而y=0时能交换的只有数值大小相同的元素。

代码：

	#include<bits/stdc++.h>
	using namespace std;
	typedef long long ll;
	const int MAX = 1e5 + 10;
	int T, N, a[MAX];
	ll x, y;
	map<int, int> mp;//储存num出现了几次
	int main() {
		scanf("%d", &T);
		while (T--) {
			mp.clear();
			ll cnt = 0;
			scanf("%d%lld%lld", &N, &x, &y);
			for (int i = 1; i <= N; i++) {
				scanf("%d", &a[i]);
				mp[a[i]]++;
				x -= 1ll * i * a[i];
				y -= 1ll * i * a[i] * a[i];
			}
			if (x == 0) {//说明a[i] == a[j]，则y必然为0
				if (y) {//y不为0 不能交换
					printf("0\n");
					continue;
				}
				for (int i = 1; i <= N; i++)//相等的数可以互换
					cnt += mp[a[i]] - 1;
			}
			else {//x不为0
				if (y % x) {//不能整除说明a[i]+a[j]不是整数 不能交换
					printf("0\n");
					continue;
				}
				ll sum = y / x;
				for (int i = 1; i <= N; i++) {
					ll target = sum - a[i];
					if (a[i] == target)continue;//防止除数为0
					int j = i + x / (a[i] - target);
					if (j >= 1 && j <= N && a[j] == target)cnt++;
				}
				
			}
			printf("%lld\n", cnt / 2);
		}
		return 0;
	}
	```