Pagini recente » Cod sursa (job #564009) | Cod sursa (job #1995688) | Cod sursa (job #2172528) | Cod sursa (job #1128197) | Cod sursa (job #2036707)
#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;
}