Codevs 2054疯狂的馒头

AC通道: http://www.lydsy.com/JudgeOnline/problem.php?id=2054

[分析]

对于任何一个馒头，只有最后一次刷才有意义。无论之前刷过多少次，它的颜色都是由最后一次刷决定。所以，我们可以考虑倒过来刷，如果已经刷过就不需要再刷了。我们可以用并查集维护这个链表，用fa[i]表示i右边第一个未刷过的的馒头。

核心代码:for(int j=find(l);j<=r;j=find(j)){color[j]=i;fa[j]=j+1;}

#include <iostream>#include <cstdio>using namespace std;int n,m,p,q;int fa[1000010];int color[1000010];int find(int x){int tmp=x,pre;while(tmp!=fa[tmp])tmp=fa[tmp];while(x!=tmp){pre=fa[x];fa[x]=tmp;x=pre;}return tmp;} int main(){scanf("%d%d%d%d",&n,&m,&p,&q);for(int i=1;i<=n;i++)fa[i]=i;for(int i=m;i>=1;i--){int l=(i*p+q)%n+1;int r=(i*q+p)%n+1;if(l>r){int t=l;l=r;r=t;}for(int j=find(l);j<=r;j=find(j)){if(j==0)break;   //注意：如果不加这一条会导致超时color[j]=i;fa[j]=j+1; }}for(int i=1;i<=n;i++)printf("%d\n",color[i]);return 0;}