提交时间:2023-12-09 09:10:22

运行 ID: 24057

#include<bits/stdc++.h> #define up(i,l,r) for(int i=(l);i<=(r);++i) #define down(i,l,r) for(int i=(l);i>=(r);--i) #define p_b push_back using namespace std; typedef long long ll; const int maxn=2e5+10,mod=998244353; inline ll read(){ ll x=0;short t=1; char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')t=-1;ch=getchar();} while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar(); return x*t; } int n,a[maxn],dep[maxn],res; vector<int>v[maxn]; void dfs(int u,int fa){ for(int x:v[u])if(x!=fa){ dep[x]=dep[u]+1;dfs(x,u); } }bool chk(vector<int>v){ if(!v.size())return 0; up(x,1,n){ dep[x]=0;dfs(x,0); bool f=1; for(int y:v)if(dep[y]>a[y]){f=0;break;} if(f)return 1; }return 0; } void DFS(int x,vector<int>v){ if(x>n){ if(chk(v))res+=(1<<int(v.size())); return; }DFS(x+1,v); v.p_b(x); DFS(x+1,v); v.pop_back(); } void slv(){ n=read();up(i,1,n)a[i]=read(); up(i,1,n-1){int x=read(),y=read();v[x].p_b(y),v[y].p_b(x);} assert(n<=8); vector<int>g; DFS(1,g); cout<<res<<'\n'; }int main(){ // freopen("tree.in","r",stdin); // freopen("tree.out","w",stdout); slv(); fclose(stdin); fclose(stdout); return 0; }