Cod sursa(job #1306692)

Utilizator vlady1997Vlad Bucur vlady1997 Data 31 decembrie 2014 13:32:22
Problema Asmax Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.35 kb
        #include <cstdio>
        #include <cstring>
        #include <vector>
        using namespace std;
        vector <int> g[16001];
        int a[16001];
        bool sel[16001];
        int main()
        {
            int n, i, j, x, y, s, Max=0;
            freopen("asmax.in","r",stdin);
            freopen("asmax.out","w",stdout);
            scanf("%d",&n);
            for (i=1; i<=n; i++)
            {
                scanf("%d",&a[i]); Max+=a[i];}
            for (i=1; i<n; i++)
            {
                scanf("%d%d",&x,&y);
                g[x].push_back(y);
                g[y].push_back(x);
            }
            for (i=1; i<=n; i++)
            {
                memset(sel,0,sizeof(sel)); sel[i]=true; s=0;
                for (j=0; j<g[i].size(); j++)
                {
                    int y=g[i][j];
                    if (g[y].size()<=1)
                    {
                        sel[i]=true;
                    }
                }
                for (j=1; j<=n; j++)
                {
                    if (sel[j]==false)
                    {
                        s+=a[j];
                    }
                }
                if (s>Max) Max=s;
            }
            printf("%d\n",Max);
            fclose(stdin);
            fclose(stdout);
            return 0;
        }