Cod sursa(job #2278470)

Utilizator savulescustefanSavulescu Stefan savulescustefan Data 8 noiembrie 2018 07:48:52
Problema Heavy Path Decomposition Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.49 kb
#include <cstdio>
#include <vector>
using namespace std;
vector <int> m3[100004],v[100004];
int n,m,i,j,x,y,c,z,q,w,nr,nr3,poz1,poz2,Max,k,aux,nr4,Min,st,dr,nr2[100004],t[100004],a[100004],s[100004],l[100004],poz[100004],le[100004];
bool ok;
void query (int st, int dr, int poz1, int poz2, int nod)
{
    if ((st>=poz1) && (dr<=poz2))
    {
        if (Max<m3[i][nod])
            Max=m3[i][nod];
        return;
    }
    if (((st+dr)/2)>=poz1)
        query(st,(st+dr)/2,poz1,poz2,nod*2);
    if ((((st+dr)/2)+1)<=poz2)
        query(((st+dr)/2)+1,dr,poz1,poz2,(nod*2)+1);
    return;
}
void up (int st, int dr, int nod)
{
    if (st==dr)
    {
        m3[i][nod]=x;
        return;
    }
    if (((st+dr)/2)>=poz1)
        up(st,(st+dr)/2,nod*2);
    else
        up(((st+dr)/2)+1,dr,(nod*2)+1);
    if (m3[i][nod*2]>m3[i][(nod*2)+1])
        m3[i][nod]=m3[i][nod*2];
    else
        m3[i][nod]=m3[i][(nod*2)+1];
    return;
}
void g (int x)
{
    int i,k;
    k=v[x].size();
    k--;
    for (i=0;i<=k;i++)
    {
        if (le[v[x][i]]==0)
        {
            le[v[x][i]]=le[x]+1;
            g(v[x][i]);
        }
    }
    Max=-1;
    poz1=0;
    nr4=0;
    ok=true;
    for (i=0;i<=k;i++)
    {
        if (le[v[x][i]]>le[x])
        {
            if (s[v[x][i]]>Max)
            {
                Max=s[v[x][i]];
                poz1=v[x][i];
            }
            ok=false;
            nr4+=s[v[x][i]];
        }
    }
    if (ok==true)
    {
        s[x]=1;
        nr++;
        l[x]=nr;
        poz[x]=0;
        nr2[nr]=1;
        return;
    }
    l[x]=l[poz1];
    poz[x]=poz[poz1]+1;
    nr2[l[x]]++;
    s[x]=nr4+1;
    for (i=0;i<=k;i++)
    {
        if ((v[x][i]!=poz1) && (le[x]<le[v[x][i]]))
            t[l[v[x][i]]]=x;
    }
    return;
}
void f (int x,int y)
{
    while (l[x]!=l[y])
    {
        if (le[t[l[x]]]<le[t[l[y]]])
        {
            z=l[y];
            i=z;
            poz1=poz[y];
            k=nr2[z];
            k--;
            poz2=k;
            query(0,k,poz1,poz2,1);
            y=t[l[y]];
        }
        else
        {
            z=l[x];
            i=z;
            poz1=poz[x];
            poz2=nr2[z]-1;
            query(0,poz2,poz1,poz2,1);
            x=t[l[x]];
        }
    }
    z=l[x];
    i=z;
    poz1=poz[y];
    poz2=poz[x];
    if (poz1>poz2)
    {
        aux=poz1;
        poz1=poz2;
        poz2=aux;
    }
    k=nr2[z];
    k--;
    query(0,k,poz1,poz2,1);
    return;
}
int main()
{
    freopen ("heavypath.in","r",stdin);
    freopen ("heavypath.out","w",stdout);
    scanf ("%d %d", &n, &m);
    for (i=1;i<=n;i++)
        scanf ("%d", &a[i]);
    for (i=1;i<=(n-1);i++)
    {
        scanf ("%d %d", &x, &y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    le[1]=1;
    g(1);
    for (i=1;i<=nr;i++)
    {
        k=nr2[i];
        k*=3;
        for (j=1;j<=k;j++)
            m3[i].push_back(-1);
    }
    for (j=1;j<=n;j++)
    {
        x=a[j];
        i=l[j];
        poz1=poz[j];
        up(0,nr2[i]-1,1);
    }
    for (j=1;j<=m;j++)
    {
        scanf ("%d %d %d", &c, &x, &y);
        if (c==1)
        {
            Max=-1;
            f(x,y);
            printf ("%d\n", Max);
        }
        else
        {
            z=l[x];
            poz1=poz[x];
            i=z;
            x=y;
            k=nr2[z];
            k--;
            up(0,k,1);
        }
    }
    return 0;
}