P7298 [USACO21JAN] Dance Mooves G

题目

题目

思路

赛时刚T4就是赌博，赢了皆大欢喜，输了一败涂地。
结果赌了2天天天涂地……
总分77.7pts……
myd从来不赌，结果200+……
也许我也该戒赌了？？
提一嘴wj，52pts
本题非常好想，非常不好写……
可以发现通过置换构造环，然后考虑次数，如果做得够多，可以遍历整个环。
否则GG滑动窗口搞
code:

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
int x[200005],zh[200005],ans[200010],sum[200010],y[200005],to[200010];
bool p[200010];
int q[200010],r,l;
vector<pair<int,int> > c[200010];
int n,k;
long long m,m1,m2;
inline long long readll()
{
    long long ret,c,f=1;
    while (((c=getchar())> '9'||c< '0')&&c!='-');
    if (c=='-') f=-1,ret=0;
    else ret=c-'0';
    while ((c=getchar())>='0'&&c<='9') ret=(ret<<1)+(ret<<3)+c-'0';
    return ret*f;
}
inline int read()
{
    int ret,c,f=1;
    while (((c=getchar())> '9'||c< '0')&&c!='-');
    if (c=='-') f=-1,ret=0;
    else ret=c-'0';
    while ((c=getchar())>='0'&&c<='9') ret=(ret<<1)+(ret<<3)+c-'0';
    return ret*f;
}
int main()
{
	n=read(),k=read(),m=readll();
	for (int i=1;i<=n;i++) to[i]=i,c[i].push_back(make_pair(0,i));
	m1=m/k,m2=m%k;
	for (int i=1;i<=k;i++)
	{
		x[i]=read(),y[i]=read();
		c[to[x[i]]].push_back(make_pair(i,y[i]));
		c[to[y[i]]].push_back(make_pair(i,x[i]));
		swap(to[x[i]],to[y[i]]);
	}
	for (int i=1;i<=n;i++)
	{
		zh[to[i]]=i;
	}
	int siz=0;
	for (int i=1;i<=n;i++)
	{
		if (p[i]) continue;
		p[i]=1;
		bool flag=0;
		l=1,r=0;
		for (int j=1,kk=i;j<=m1;j++,kk=zh[kk])
		{
			if (kk==i&&j!=1)
			{
				flag=1;
				break;
			}
			q[++r]=kk;
			for (vector<pair<int,int> >::iterator it=c[kk].begin();it!=c[kk].end();it++)
			{
				if ((*it).first>k)
				{
					break;
				}
				sum[(*it).second]++;
				if (sum[(*it).second]==1) siz++;
			}
		}
		if (flag)
		{
			ans[i]=siz;
			p[i]=1;
			for (vector<pair<int,int> >::iterator it=c[i].begin();it!=c[i].end();it++)
			{
				if ((*it).first>k)
				{
					break;
				}
				sum[(*it).second]--;
				if (sum[(*it).second]==0) siz--;
			}
			for (int kk=zh[i];kk!=i;ans[kk]=ans[i],p[kk]=1,kk=zh[kk]) for (vector<pair<int,int> >::iterator it=c[kk].begin();it!=c[kk].end();it++)
			{
				if ((*it).first>k)
				{
					break;
				}
				sum[(*it).second]--;
				if (sum[(*it).second]==0) siz--;
			}
			continue;
		}
		zh[0]=i;
		for (int jk=i,kkk=1;jk!=i||kkk==1;p[jk]=1,kkk++,jk=zh[jk])
		{
			int kk=zh[q[r]];
			for (vector<pair<int,int> >::iterator it=c[kk].begin();it!=c[kk].end();it++)
			{
				if ((*it).first>m2)
				{
					break;
				}
				sum[(*it).second]++;
				if (sum[(*it).second]==1) siz++;
			}
			ans[jk]=siz;
			for (vector<pair<int,int> >::iterator it=c[kk].begin();it!=c[kk].end();it++)
			{
				if ((*it).first>m2)
				{
					break;
				}
				sum[(*it).second]--;
				if (sum[(*it).second]==0) siz--;
			}
			for (vector<pair<int,int> >::iterator it=c[kk].begin();it!=c[kk].end();it++)
			{
				if ((*it).first>k)
				{
					break;
				}
				sum[(*it).second]++;
				if (sum[(*it).second]==1) siz++;
			}
			for (vector<pair<int,int> >::iterator it=c[jk].begin();it!=c[jk].end();it++)
			{
				if ((*it).first>k)
				{
					break;
				}
				sum[(*it).second]--;
				if (sum[(*it).second]==0) siz--;
			}
			l++,q[++r]=kk;
		}
		for (int j=l;j<=r;j++)
		{
			int kk=q[j];
			for (vector<pair<int,int> >::iterator it=c[kk].begin();it!=c[kk].end();it++)
			{
				if ((*it).first>k)
				{
					break;
				}
				sum[(*it).second]--;
				if (sum[(*it).second]==0) siz--;
			}
		}
	}
	for (int i=1;i<=n;i++) printf("%lld\n",ans[i]);
	return 0;
}