[NOIP模拟]鸭舌

题目描述：

样例输入1：
5
1 3 1 3 2
2 5
3 4
4 5
1 5
4
样例输出1：
6
样例输入 2：
3
2 1 1
3 2
1 2
3
样例输出 2：
2
数据规模：
对于 30% 的数据：n≤5；ai≤5；
对于 100% 的数据：1≤n≤100000；0≤ai≤109；1≤x≤n。

附代码：

#include<iostream>#include<cstring>#include<cstdlib>#include<cstdio>#include<ctime>#include<queue>#include<string>#include<cmath>#include<cctype>#include<set>#include<map>#include<iomanip>#include<algorithm>#include<vector>using namespace std;const int N=100010;long long tot,t,n,f[N],a[N],go[N*2],next[N*2],first[N];void create(long long a,long long b){    tot++;    next[tot]=first[a];    first[a]=tot;    go[tot]=b;}void work(long long x,long long fa){    long long j,tot=0;    vector<long long>vec;    if(a[x]==0) return;    for(long long i=first[x];i;i=next[i])    if(go[i]!=fa)    {        work(go[i],x);        vec.push_back(f[go[i]]);        tot++;    }    sort(vec.begin(),vec.end());    a[x]--;    f[x]=1;    for(long long i=tot;i>=1;i--)    {        j=vec[i-1];        if(a[x]==0) break;        if(j!=0)        {            f[x]+=j+1;            a[x]--;        }        else break;    }    for(long long i=first[x];i;i=next[i])        if(go[i]!=fa)        {            j=min(a[x],a[go[i]]);            a[x]-=j;            f[x]+=2*j;        }}int main(){    //freopen("tongue.in","r",stdin);    //freopen("tongue.out","w",stdout);    scanf("%I64d",&n);    for(long long i=1;i<=n;i++)        scanf("%I64d",&a[i]);    for(long long i=1;i<=n-1;i++)    {        long long x,y;        scanf("%I64d%I64d",&x,&y);        create(x,y);        create(y,x);    }       scanf("%I64d",&t);    a[t]++;    work(t,0);    printf("%I64d",f[t]-1);    return 0;}