Cod sursa(job #3169970)

Utilizator Giulian617Buzatu Giulian Giulian617 Data 16 noiembrie 2023 15:00:37
Problema Curcubeu Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("curcubeu.in");
ofstream g("curcubeu.out");
int n,pos,mn,maxx;
vector<int>a,b,c,parent,color;
int findRoot(int node)
{
    if(parent[node]==node)
        return node;
    return parent[node]=findRoot(parent[node]);
}
int main()
{
    f>>n;
    a.resize(n+1);
    b.resize(n+1);
    c.resize(n+1);
    parent.resize(n+1);
    color.resize(n+1);
    f>>a[1]>>b[1]>>c[1];
    for(int i=2; i<=n; i++)
    {
        a[i]=(1LL*a[i-1]*i)%n;
        b[i]=(1LL*b[i-1]*i)%n;
        c[i]=(1LL*c[i-1]*i)%n;
    }
    for(int i=0; i<n; i++)
        parent[i]=i;
    for(int i=n; i>=1; i--)
    {
        mn=min(a[i],b[i]);
        maxx=max(a[i],b[i]);
        if(mn==0 && maxx==0)
            continue;
        if(mn==0)
            mn=1;
        pos=mn;
        if(color[pos])
            pos=findRoot(pos);
        while(!color[pos] && pos<=maxx)
        {
            color[pos]=c[i];
            pos=findRoot(pos);
        }
        parent[mn]=pos;
    }
    for(int i=1; i<n; i++)
        g<<color[i]<<'\n';
    return 0;
}