Cod sursa(job #774099)

Utilizator batistaUPB-Oprea-Cosmin-Dumitru batista Data 3 august 2012 14:36:13
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.01 kb
#include<fstream>
#include<algorithm>
using namespace std;
struct nod{int info,c;nod*adr;}*v[16003],*p;
int n,i,x,y,val[16003];
short viz[16003];
int smax[16003];
int k,cm[16003];
void dfs(int nd)
{
	viz[nd]=1;
	nod*p=v[nd];
	smax[nd]=val[nd];
	while(p)
	{
		if(!viz[p->info])
		{
		 dfs(p->info);
		 k++;
		 if(smax[nd]+smax[p->info]>smax[nd])
		 {
		  smax[nd]+=smax[p->info];
		  cm[k]=smax[p->info];
		 }
		}
		p=p->adr;
	}
}
void dfs_dinamica(int nd)
{
	viz[nd]=0;
	nod*p=v[nd];
	while(p)
	{
		if(viz[p->info])
		{
		  dfs_dinamica(p->info);
		 k++;
		 smax[p->info]=max(smax[p->info],smax[nd]-cm[k]+smax[p->info]);
		}
		p=p->adr;
	}
}
int main()
{
	ifstream f("asmax.in");ofstream g("asmax.out");
	f>>n;
	for(i=1;i<=n;i++)f>>val[i];
	for(i=1;i<n;i++)
	{
		f>>x>>y;
		p=new nod;
		p->info=y; p->adr=v[x]; v[x]=p;
		p=new nod;
		p->info=x; p->adr=v[y]; v[y]=p;
	}
	dfs(1);
	k=0;
	dfs_dinamica(1);
	g<<*max_element(smax+1,smax+n+1)<<'\n';
	f.close();g.close();
return 0;}