Cod sursa(job #1135736)

Utilizator andi23roscaRosca Andrei andi23rosca Data 8 martie 2014 12:33:47
Problema Asmax Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <iostream>
#include <fstream>
using namespace std;
int n,v[16001],g[16001],m[16001][16001],i,j,x,y,ok=0,s=-1001;
int main() {
ifstream in("asmax.in");
ofstream out("asmax.out");
in>>n;
for(i=1;i<=n;i++) {
    in>>v[i];
    if(v[i]>0) ok=1;
}
for(i=1;i<n;i++) {
    in>>x>>y;
    m[x][y]=1;
    m[y][x]=1;
    g[x]++;
    g[y]++;
}
if (ok==0) {
    for(i=1;i<=n;i++) {
        if(v[i]>s) s=v[i];
    }
    out<<s;
    return 0;
}
ok=0;
while(ok==0) {
    ok=1;
    for(i=1;i<=n;i++) {
        if(g[i]==1) {
            if(v[i]<0) {
                g[i]=0;
                ok=0;
                for(j=i;j<n;j++) {
                    if(m[i][j]==1) {
                        m[i][j]=0;
                        m[j][i]=0;
                        g[j]--;
                    }
                }
            }
        }
    }
}
s=0;
for(i=1;i<=n;i++) {
    if(g[i]!=0) s=s+v[i];
}
out<<s;
return 0;
}