比赛 CSP2023-J模拟赛 评测结果 WWWWWWWWWW
题目名称 新建题目 最终得分 0
用户昵称 rain 运行时间 0.000 s
代码语言 C++ 内存使用 0.00 MiB
提交时间 2023-10-18 21:57:52
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
int n,p[2010],dp[2010],ans,cj=1,t;
bool f[2010];
vector<int>vt[2010];
struct pt{
	int w,dep,m;
}a[2010];
void fnd(int x,int d){
	a[x].dep=d,dp[x]=d;
	for(int i=0;i<vt[x].size();i++){
		int y=vt[x][i];
		if(a[y].dep==0&&y!=1){
			if(a[y].w<=a[x].w)f[y]=1;
			fnd(y,d+1);
		}
	}
}
bool cmp(pt x,pt y){
	return x.dep>y.dep;
}
int main(){
	freopen("touch.in","r",stdin);
	freopen("touch.out","w",stdout);
	cin>>n;
	for(int i=1;i<=n;i++)
	cin>>a[i].w,a[i].m=i;
	for(int i=1;i<n;i++){
		int x,y;
		cin>>x>>y;
		if(x>y)swap(x,y);
		vt[x].push_back(y);
		vt[y].push_back(x);
	}
	fnd(1,0);f[1]=1;
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=n;i++){
		int x=a[i].m;t=0;
		for(int j=0;j<vt[x].size();j++){
			if(dp[vt[x][j]]>dp[x]&&f[vt[x][j]]==0)
			p[x]+=p[vt[x][j]]+1;
			if(x==1&&t>=2){
				cj*=p[vt[x][j]]+1;
				t=-1;
			}
		}
		ans+=p[x];
	}
	if(t==-1)cout<<ans+cj;
	else cout<<ans;
	return 0;
}