FZU2168 防守阵地 I

Accept: 28 Submit: 97

Time Limit: 1000 mSec Memory Limit : 32768 KB

Problem Description

YL是shadow国的国王,shadow国有N个城市。为了节省开支,shadow国只有N-1条道路,这N-1条道路使得N个城市连通。某一年,shadow国发生了叛乱,叛军占领了多个城市,王都岌岌可危。王都为编号为1的城市，除了王都外有K个城市有YL的军队。现在这K支军队要向王都进军，并且消灭沿途经过的城市中的叛军。现给出N个城市的道路情况以及城市的叛军数量，问总共需要消灭多少叛军？

Input

第一行输入两个整数N，K，接下来输入N(1<=N<=100000)个整数Ai(0<=Ai<=10000)，表示第i个城市的叛军数量。接下来输入K个大于等于1且小于等于N的整数，表示有军队的城市的编号。数据保证王都以及有军队的城市没有叛军。接下来输入N-1行，每行两个整数u、v，表示连接u和v的一条道路。每支军队只能沿着道路走，并且是其所在城市与王都之间的最短路线走。

Output

输出一行一个整数表示消灭的叛军数量。

Sample Input

4 2

0 3 0 0

3 4

1 2

2 3

2 4

Sample Output

3

#include<stdio.h>
#include<malloc.h>
typedef struct nnn
{
    int id;
    struct nnn *next;
}*NODE,node;
NODE edg[200005];
int dp[200005],flog[200005],value[200005],vist[200005];
void dfs(int p)
{
    NODE NEXT,q;
    dp[p]=value[p]; vist[p]=1;
    for(NEXT=edg[p]->next; NEXT!=NULL; )
    {
        int son=NEXT->id;
        if(vist[son]){
          NEXT=NEXT->next;  continue;
        }
        dfs(son);
        if(flog[son])
        {
            flog[p]=1; dp[p]+=dp[son];
        }
        q=NEXT;
        NEXT=NEXT->next;
    }
}
int main()
{
    int n,k,a,b,i;
    NODE p;
    while(scanf("%d%d",&n,&k)>0)
    {
        for(i=0;i<=n;i++)
        {
            flog[i]=0; vist[i]=0;
            edg[i]=(NODE)malloc(sizeof(node));
            edg[i]->next=NULL;
        }
        for(i=1; i<=n; i++)
            scanf("%d",&value[i]);
        for(i=1; i<=k; i++)
        {
            scanf("%d",&a);
            flog[a]=1;
        }
        for(i=1; i<n;i++)
        {
            scanf("%d%d",&a,&b);
            p=(NODE)malloc(sizeof(node));
            p->id=b; p->next=edg[a]->next;
            edg[a]->next=p;
             p=(NODE)malloc(sizeof(node));
            p->id=a; p->next=edg[b]->next;
            edg[b]->next=p;
        }
        dfs(1);
        printf("%d\n",dp[1]);
    }
}