Cod sursa(job #2036707)

Utilizator usureluflorianUsurelu Florian-Robert usureluflorian Data 10 octombrie 2017 23:19:17
Problema Reconst Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.17 kb
#include <fstream>
#include <vector>
#include <bitset>
#include <algorithm>
#include <climits>
using namespace std;
ifstream f ("petrica.in");
ofstream g ("petrica.out");
const int nmax=2*1e2+3;
int n,value,dad[nmax],population[nmax],a,b,sum,country[nmax],sol,usurelu[5],order[5],k,val,valusu[nmax];
struct
{
    int st,dr;
}usu[nmax];
bitset <nmax> viz;
vector <int> v[nmax];
bool apartine(int a,int b)
{
    return usu[a].st<=usu[b].st&&usu[a].dr>=usu[b].dr;
}
void clean()
{
    for(int i=1;i<=n;++i) viz[i]=false;
}
void dfs(int node)
{
    viz[node]=true;
    sum+=population[node];
    for(int i=0;i<v[node].size();++i)
    {
        if(viz[v[node][i]]==false&&dad[v[node][i]]>dad[node]) dfs(v[node][i]);
    }
}
void atribution(int node)
{
    viz[node]=true;
    dad[node]=++value;
    for(int i=0;i<v[node].size();++i)
    {
        if(viz[v[node][i]]==false) atribution(v[node][i]);
    }
}
void calculate(int nod)
{
    viz[nod]=true;
    val+=population[nod];
    for(int i=0;i<v[nod].size();++i)
    {
        if(viz[v[nod][i]]==false&&dad[v[nod][i]]>dad[nod]) calculate(v[nod][i]);
    }
}
void maimute(int nod)
{
    viz[nod]=true;
    usu[nod].st=++k;
    for(int i=0;i<v[nod].size();++i)
    {
        if(viz[v[nod][i]]==false) maimute(v[nod][i]);
    }
    usu[nod].dr=++k;
}
int main()
{
    f>>n;
    sol=INT_MAX;
    for(int i=1;i<=n;++i) f>>population[i];
    for(int i=1;i<n;++i)
    {
        f>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    atribution(1);
    clean();
    for(int i=1;i<=n;++i)
    {
        val=0;
        calculate(i);
        valusu[i]=val;
        clean();
    }
    maimute(1);
    for(int i=2;i<n-1;++i)
    {
        for(int j=i+1;j<n;++j)
        {
            for(int k=j+1;k<=n;++k)
            {
                order[1]=i;
                order[2]=j;
                order[3]=k;
                order[4]=1;
                for(int i1=1;i1<4;++i1)
                {
                    for(int i2=i1+1;i2<=4;++i2) if(apartine(order[i1],order[i2])) swap(order[i1],order[i2]);
                }
                //g<<order[1]<<' '<<order[2]<<' '<<order[3]<<' '<<order[4]<<'\n';
                usurelu[1]=valusu[order[1]];
                usurelu[2]=valusu[order[2]];
                usurelu[3]=valusu[order[3]];
                usurelu[4]=valusu[order[4]];
                //g<<order[1]<<' '<<order[2]<<' '<<order[3]<<' '<<order[4]<<" -> de valori: "<<usurelu[order[1]]<<' '<<usurelu[order[2]]<<' '<<usurelu[order[3]]<<' '<<usurelu[order[4]]<<'\n';
                for(int i1=1;i1<4;++i1)
                {
                    for(int i2=i1+1;i2<=4;++i2)
                    {
                        if(apartine(order[i2],order[i1])) usurelu[i2]-=usurelu[i1];
                    }
                }
                //g<<order[1]<<' '<<order[2]<<' '<<order[3]<<' '<<order[4]<<" -> de valori: "<<usurelu[order[1]]<<' '<<usurelu[order[2]]<<' '<<usurelu[order[3]]<<' '<<usurelu[order[4]]<<'\n';
                sort(usurelu+1,usurelu+4+1);
                sol=min(sol,usurelu[4]-usurelu[1]);
            }
        }
    }
    g<<sol;
    return 0;
}