Just Shuffle

题目

传送门

分析

这题用到了置换群的知识，具体结论可以参考大佬论文。
有了前置结论，再用上扩展欧几里得求乘法逆元就ok

代码

#include

#define ll long long

using namespace std;

const int MAXN=1e5+100;
vector<ll> vec;

ll exgcd(ll a,ll b,ll &x,ll &y)//扩展欧几里得求同余方程
{
	if(b==0)
	{
		x=1;y=0;return a;
	}
	ll c=exgcd(b,a%b,y,x);
	y-=a/b*x;
	return c;
}

ll l,k,t[MAXN],ans[MAXN];
bool v[MAXN];

int main()
{
	scanf("%lld%lld",&l,&k);
	for(int i=1;i<=l;i++) scanf("%lld",&t[i]);
	for(int i=1;i<=l;i++)
	{
		if(v[i]) continue;
		v[i]=1;vec.clear();vec.push_back(i);
		for(int j=t[i];j!=i;j=t[j]) vec.push_back(j),v[j]=1;
		if(vec.size()==1)
		{
			ans[i]=t[i];continue;
		}
		ll inv,y;
		exgcd(k%vec.size(),vec.size(),inv,y);
		if(inv<0)  inv+=vec.size();
		for(int j=0;j<vec.size();j++) ans[vec[j]]=vec[(j+inv)%vec.size()];
	}
	for(int i=1;i<=l;i++) printf("%lld ",ans[i]);
}

----原创文章，仅供参考