比赛 |
CSP2023-J模拟赛 |
评测结果 |
WWWWWWWWWW |
题目名称 |
新建题目 |
最终得分 |
0 |
用户昵称 |
李栋阳 |
运行时间 |
0.000 s |
代码语言 |
C++ |
内存使用 |
0.00 MiB |
提交时间 |
2023-10-18 18:58:54 |
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
int n;
int f[2005];
int size[2005];
int w[2005];
int ans;
int findx(int x){
if(f[x]==x){
return x;
}
return f[x]=findx(f[x]);
}
void merge(int a,int b){
int x=findx(a);
int y=findx(b);
if(x==y) return;
else{
if(size[x]>=size[y]){
size[x]+=size[y];
size[y]=0;
f[y]=x;
}
else{
size[y]+=size[x];
size[x]=0;
f[x]=y;
}
}
return;
}
vector<int> e[2005];
void dfs(int u,int fa){
for(int i=0;i<e[u].size();i++){
int v=e[u][i];
if(v==fa) continue;
if(w[v]>w[u]){
merge(u,v);
}
dfs(v,u);
}
return;
}
int main(){
freopen("touch.in","r",stdin);
freopen("touch.out","w",stdout);
cin>>n;
for(int i=1;i<=n;i++){
f[i]=i;
size[i]=1;
}
for(int i=1;i<=n;i++){
cin>>w[i];
}
for(int i=1;i<n;i++){
int x,y;
cin>>x>>y;
e[x].push_back(y);
e[y].push_back(x);
}
dfs(1,0);
for(int i=1;i<=n;i++){
ans+=size[i]*(size[i]-1)/2;
}
cout<<ans;
return 0;
}