题目地址:http://acm.zju.edu.cn/onlinejudge/showContestProblem.do?problemId=5971
题意
给出一个数字序列原本的x、y值,和交换两个数字之后新的序列,问有多少种交换方法使新的序列恢复到原来的序列,注意:给出的x和y可能并不是原来序列的实际对应值,当然这种情况下就没办法将新序列恢复到原来的序列,方法数就是0了。
解题思路
根据题意可以列出如下式子(还是手写方便ε-(´∀`; ))
可以导出下面的式子
可知最后要交换的两个数满足7⃣️式
所以题目就转化为在一个序列中找到两个数字ai和aj使得7式,并且交换之后的恢复的序列要满足x和y值(这就是为什么第一个样例ai+aj=5,但是只交换和第2和第3个、第5和第6个,因为其他位置上的1和4交换不满足x和y值这个限定条件)
分情况讨论:
1. y-y'不能整除x-x',那么输出0
2. 分母x-x'=0,即新序列和原始序列相同
如果y!=y'的话说明给出y是不准确的,不能恢复到原始序列,输出0;
如果y==y',说明新序列和原始序列相同,那么只要在输入序列的时候预处理相同的数字有多少个,从中选取任意两个交换后
的序列就是原始序列,方法数相加就是结果。
3. 能整除且不是新序列不是原始序列
遍历序列:定ai和i, 由7式求出aj,,由5式推出aj对应的下标j,还要判断计算出来aj和a[j]是否相同!并判断此时的ai和aj交换后序列是否满足x和y值,满足的话计数器++
注意:在遍历新序列是求得的i和j是要交换的两个数的位置,恢复到原始序列实际上对应的是i-aj,j-ai,而上述思路和公式都是建立在原始序列中i-ai,j-aj的对应关系上的,一定要注意区别!!否则就错啦!
j会比较大,数组要开大一些
ac代码
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <ctype.h>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <sstream>
#define maxn 200005
typedef long long ll;
const ll mod=1e9+7;
using namespace std;
ll a[maxn],cnt[maxn],vis[maxn];
int main()
{
//freopen("/Users/zhangkanqi/Desktop/11.txt","r",stdin);
ll t,n,x,y;
scanf("%lld",&t);
while(t--)
{
ll num=0,div,xx=0,yy=0,ans=0;//xx,yy为新序列对应的x和y值
memset(cnt,0,sizeof(cnt));
memset(vis,0,sizeof(vis));
scanf("%lld %lld %lld",&n,&x,&y);
for(ll i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
if(cnt[a[i]]==0)
vis[num++]=a[i];//不记录重复出现的数字
cnt[a[i]]++;
xx+=i*a[i];
yy+=i*a[i]*a[i];
}
if(x==xx)
{
if(y !=yy)
{
printf("0\n");
continue;
}
else
{
for(ll i=0;i<num;i++)
{
ll r=cnt[vis[i]];
if(r>1) ans+=r*(r-1)/2;//任选2个
}
printf("%lld\n",ans);
continue;
}
}
else if(abs(y-yy)%abs(x-xx)!=0)
{
printf("0\n");
continue;
}
else
{
ll ai,aj,j;
div=(y-yy)/(x-xx);
for(ll i=1;i<=n;i++)
{
ai=a[i];
aj=div-ai;
if(ai-aj==0)continue;//除法要先判断分母是否为0
if(abs(x-xx)%abs(ai-aj)!=0) continue;
j=i+((x-xx)/(ai-aj));//注意在遍历新序列是求得的i和j是要交换的两个数的位置,恢复到原始序列实际上对应的是i-aj,j-ai,注意区别!
if(j<=i) continue;//一直往后找,避免重复
if( aj==a[j] && x==xx-i*ai-j*aj+i*aj+j*ai && y==yy-i*ai*ai-j*aj*aj+i*aj*aj+j*ai*ai)//后面两个判断可以省略
ans++;
}
printf("%lld\n",ans);
}
}
}