Cod sursa(job #1804691)

Utilizator adystar00Bunea Andrei adystar00 Data 12 noiembrie 2016 21:27:43
Problema Curcubeu Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.44 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)
{
    int node=x,aux;
    while(x!=dad[x])
        x=dad[x];
    while(node!=x)
    {
        aux=dad[node];
        dad[node]=x;
        node=aux;
    }
    return x;
}
void Union (int x, int y)
{
    if(rand()%2==1)
    {
        dad[x]=y;
        lim[y]=max(lim[x],lim[y]);
    }
    else
        {
        dad[y]=x;
        lim[x]=max(lim[y],lim[x]);
    }
}
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]=((long long)a[i-1]*i)%n;
        b[i]=((long long)b[i-1]*i)%n;
        c[i]=((long long)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(Find(j),Find(j-1));
                culoare[j]=c[i];
            }
            else
            {
                if(dad[j-1]!=0)
                    Union(Find(j),Find(j-1));
                j=lim[Find(j)];
            }
        }
    }
    for(i=1;i<n;i++)
        fout<<culoare[i]<<"\n";
    return 0;
}