Cod sursa(job #585848)

Utilizator S7012MYPetru Trimbitas S7012MY Data 30 aprilie 2011 12:12:50
Problema Guvern Scor 0
Compilator cpp Status done
Runda Algoritmiada 2011, Runda Finală, Clasele 10-12 Marime 1.27 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <bitset>
#include <algorithm>
#define DN 200005
using namespace std;

typedef vector<int>::iterator it;
vector<int> gr[DN];
int n,euler[DN],sze,val[DN],lg,q[DN],nrf[DN];
bitset<DN> viz;

bool cmp(const int &a, const int &b) {
    return nrf[a]<nrf[b];
}

void add(int x,int st,int dr) {
    if(st==dr) {
        if(dr>lg) ++lg;
        q[st]=x;
        return;
    }
    int m=(st+dr)/2;
    if(x<=q[m]) add(x,st,m);
    else add(x,m+1,dr);
}

void dfs(int s) {
    viz[s]=1;
    euler[++sze]=val[s];
    for(it i=gr[s].begin(); i!=gr[s].end(); ++i) if(0==viz[*i]) dfs(*i);
}

int count(int s) {
    viz[s]=1;
    for(it i=gr[s].begin(); i!=gr[s].end(); ++i) if(0==viz[*i]) nrf[s]+=count(*i);
    return nrf[s]+1;
}

int main()
{
    ifstream f("guvern.in");
    ofstream g("guvern.out");
    f>>n;
    for(int i=1; i<n; ++i) {
        int a,b;
        f>>a>>b;
        gr[a].push_back(b);
        gr[b].push_back(a);
    }
    count(1);
    viz&=0;
    for(int i=1; i<=n; ++i) sort(gr[i].begin(),gr[i].end(),cmp);
    for(int i=1; i<=n; ++i) f>>val[i];
    dfs(1);
    //for(int i=1; i<=n; ++i) cerr<<euler[i]<<' ';
    reverse(euler+1,euler+n+1);
    for(int i=1; i<=n; ++i) add(euler[i],1,lg+1);
    g<<lg;
    return 0;
}