Harbour.Space Scholarship Contest 2021-2022 (open for everyone, rated, Div. 1 + Div. 2) E

题意

https://codeforces.com/contest/1553/problem/E

给一组排列n，最多m次交换机会。

问：交换后的排列，可能与1,2,3,4,5....n左移多少位相同。输出可能的个数以及移动的位数。

 代码

#include <bits/stdc++.h>
using namespace std;
int cy_cnt(vector<int> q, int n)//长度n排列a->排列b 最小交换次数等于 ai与bi建无向边 交换最小数量=n-环数 
{
	for(int i=0;i<n;i++)q[i]--;
	vector<int>used(n);
	int ans=0;
	for(int i=0;i<n;i++)
	{
		if(used[i]==1)continue;
		ans++;
		int j=i;
		while(used[j]==0){
			used[j]=1;
			j=q[j];
		}
	}
	return ans;
}
bool check(int n, int m, int k, vector<int> p) 
{
	vector<int>q;
	for(int i=k;i<n;i++)q.push_back(p[i]);
	for(int i=0;i<k;i++)q.push_back(p[i]);
	return n-cy_cnt(q,n)<=m;
}
void solve()
{
	int n,m;
	scanf("%d%d",&n,&m);
	vector<int>p(n);
	for(int i=0;i<n;i++)scanf("%d",&p[i]);
	vector<int>cnt(n);
	for(int i=0;i<n;i++)
	{
		int of=i-p[i]+1;//pi-1=(i-of)mod n of移动次数
		if(of<0)of+=n;
		cnt[of]++;//移动of次时，正确的位置有多少个。 
	}
	vector<int>ans;
	for(int i=0;i<n;i++)
	{
		if(cnt[i]+2*m>=n&&check(n,m,i,p))//当前一个条件不符合时可以节省check的时间 
		ans.push_back(i);
	}
	printf("%d",ans.size());
	for(int i=0;i<ans.size();i++){
		printf(" %d",ans[i]);
	}puts("");
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--){
		solve();
	}
}