Cod sursa(job #1804674)

Utilizator adystar00Bunea Andrei adystar00 Data 12 noiembrie 2016 21:14:21
Problema Curcubeu Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.34 kb
#include <iostream>
#include <fstream>
#include <cstdlib>
using namespace std;
const int N = 1000010;
int dad[N],culoare[N],lim[N],a[N],b[N],c[N];
int Find (int x)
{
    if(dad[x]==x)
        return x;
    dad[x]=Find(dad[x]);
    return dad[x];
}
void Union (int x, int y)
{
    int i,j;
    i=Find(x);
    j=Find(y);
    if(rand()%2==1)
    {
        dad[i]=j;
        lim[j]=max(lim[i],lim[j]);
    }
    else
        {
        dad[j]=i;
        lim[i]=max(lim[i],lim[j]);
    }
}
int main()
{
    ifstream fin ("curcubeu.in");
    ofstream fout ("curcubeu.out");
    int n,i,j,left,right;
    fin>>n>>a[1]>>b[1]>>c[1];
    for(i=2;i<n;i++)
    {
        a[i]=(a[i-1]*i)%n;
        b[i]=(b[i-1]*i)%n;
        c[i]=(c[i-1]*i)%n;
    }
    for(i=n-1;i>=1;i--)
    {
        left=min(a[i],b[i]);
        right=max(a[i],b[i]);
        for(j=left;j<=right;j++)
        {
            if(dad[j]==0)
            {
                dad[j]=j;
                lim[j]=j;
                if(dad[j-1]!=0)
                    Union(j,j-1);
                culoare[j]=c[i];
            }
            else
            {
                if(dad[j-1]!=0)
                    Union(j,j-1);
                j=lim[dad[j]];
            }
        }
    }
    for(i=1;i<n;i++)
        fout<<culoare[i]<<"\n";
    return 0;
}