CodeForces - 492E Vanya and Field(逆元)

CodeForces - 492E Vanya and Field(逆元)

题目大意

给出一个$n\times n$的坐标纸，其上有m棵苹果树，给出$dx,dy$假设当前位置为$x,y$则每次行动之后的位置会变成$(x+dx)modn,(y+dy)modn$问从哪个位置开始可以经过尽可能多的苹果树

解题思路

对于一个给定的位置$(x_0,y_0)$其能经过的所有点都可以表示为
$$\begin{gathered} x_0+k\ast dx\equiv x(modn) \\ {y}_0+k\ast dy\equiv y(modn) \end{gathered}$$
其中$x,y$为从初始位置开始所能经过所有的点的位置的坐标

由于$gcd(dx,n)=1,gcd(dy,n)=1$因此可对上二式除dx,dy(且一定有逆元),则变成
$$\begin{gathered} x_0\ast inv(dx)+k\equiv x\ast inv(dx)(modn) \\ {y}_0\ast inv(dy)+k\equiv y\ast inv(dy)(modn) \end{gathered}$$
上下两个式子做差,则有
$$x_0\ast inv(dx)-y_0\ast inv(dy)\equiv x\ast inv(dx)-y\ast inv(dy)(modn)$$
其中$x_0\ast inv(dx)-y_0\ast inv(dy)$为一常数，由此可得所有$(x\ast inv(dx)-y\ast inv(dy))modn$相等的点之间都可以互相到达.因此依之进行统计即可

AC代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef pair<int,int> pii;
typedef long long LL;
unordered_map<LL,int> mpn;
LL exgcd(LL a,LL b,LL &x,LL &y)
{
    if(b==0)
    {
        x=1,y=0;
        return a;
    }
    LL ret=exgcd(b,a%b,y,x);
    y-=a/b*x;
    return ret;
}
LL getInv(int a,int mod){
    LL x,y;
    LL d=exgcd(a,mod,x,y);
    return d==1?(x%mod+mod)%mod:-1;
}
int32_t main()
{
	int n,m,dx,dy;
	cin>>n>>m>>dx>>dy;
	int invdx=getInv(dx,n),invdy=getInv(dy,n);
	int x,y;
	int nums=0;
	pii ans;
	while(m--)
	{
		cin>>x>>y;
		int xy=((1LL*x*invdx-1LL*y*invdy)%n+n)%n;
		if(!mpn.count(xy))
		{
			mpn[xy]=1;
		}
		else mpn[xy]++;
		if(mpn[xy]>nums) nums=mpn[xy],ans=pii(x,y);
	}
	cout<<ans.first<<' '<<ans.second<<endl;
}