比赛 CSP2023-J模拟赛 评测结果 AAAATTTTTT
题目名称 新建题目 最终得分 40
用户昵称 PCT 运行时间 6.022 s
代码语言 C++ 内存使用 21.28 MiB
提交时间 2023-10-18 18:32:08
显示代码纯文本
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int N=2005;
int n,w[N],u,v,ans;
vector<int>G[N];
int dep[N];
vector<int>num;
int dis[N][N];
bool flag;
int main(){
	freopen("touch.in","r",stdin);
	freopen("touch.out","w",stdout);
	cin.tie(0)->sync_with_stdio(0);
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>w[i];
	}
	memset(dis,9,sizeof(dis));
	for(int i=1;i<n;i++){
		cin>>u>>v;
		G[u].push_back(v); 
		G[v].push_back(u);
		dis[u][v]=dis[v][u]=1;
		dis[i][i]=0; 
	}
	dis[n][n]=0;
	for(int k=1;k<=n;k++){
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]); 
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=i+1;j<=n;j++){
			num.clear();
			for(int k=1;k<=n;k++){
				if(dis[i][k]+dis[k][j]==dis[i][j]){
					num.push_back(k);
				}
			}
			flag=0;
			for(int k=0;k<num.size();k++){
				for(int l=0;l<num.size();l++){
					if(dis[1][num[k]]>dis[1][num[l]]&&w[num[k]]<=w[num[l]]){
						flag=1; break;
					}
				}
			}
			if(flag==0){
				ans++;
			}
		}
	}
	cout<<ans;
return 0;
}