比赛 |
CSP2023-J模拟赛 |
评测结果 |
WWWWWWWWWW |
题目名称 |
新建题目 |
最终得分 |
0 |
用户昵称 |
HXF |
运行时间 |
4.455 s |
代码语言 |
C++ |
内存使用 |
5.29 MiB |
提交时间 |
2023-10-18 21:59:54 |
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
const int N=3010;
int n,idx,ans;
int dep[N],head[N],nxt[N],ver[N];
bool mk[N];
long long w[N],maxx[N];
struct node{
int hao,zhi;
}a[N],shi[N];
inline int read(){
int t=0,f=1;
register char c=getchar();
while (c<48||c>57) f=(c=='-')?(-1):(f),c=getchar();
while (c>=48&&c<=57)t=(t<<1)+(t<<3)+(c^48),c=getchar();
return f*t;
}
inline long long readl(){
long long t=0,f=1;
register char c=getchar();
while (c<48||c>57) f=(c=='-')?(-1):(f),c=getchar();
while (c>=48&&c<=57)t=(t<<1)+(t<<3)+(c^48),c=getchar();
return f*t;
}
bool cmp(node x,node y){
return x.zhi>y.zhi;
}
void add(int u,int v){
nxt[++idx]=head[u];
head[u]=idx;
ver[idx]=v;
}
void dfsdep(int d,int s){
dep[d]=s;
for(int i=head[d];i;i=nxt[i]){
int dao=ver[i];
if(dep[dao]) continue;
dfsdep(dao,s+1);
}
}
void dfs(int d,int ed,int ci){
a[ci].hao=d,a[ci].zhi=w[d];
if(d==ed){
bool flag=false;
for(int i=1;i<=ci;i++) shi[i]=a[i];
sort(shi+1,shi+1+ci,cmp);
for(int i=2;i<=ci;i++){
// cout<<a[i].hao<<" ";
if(dep[a[i].hao]<dep[a[i-1].hao]){
// cout<<a[i].hao<<endl;
flag=true;
break;
}
}
// cout<<endl;
if(!flag) ans++;
return;
}
for(int i=head[d];i;i=nxt[i]){
int dao=ver[i];
if(mk[dao]) continue;
// cout<<dao<<endl;
mk[dao]=1;
dfs(dao,ed,ci+1);
}
}
int main(){
freopen("touch.in","r",stdin);
freopen("touch.out","w",stdout);
n=read();
for(int i=1;i<=n;i++) w[i]=readl();
for(int i=1;i<n;i++){
int x=read(),y=read();
add(x,y);
add(y,x);
}
dfsdep(1,1);
for(int i=1;i<=n;i++){
for(int j=i+1;j<=n;j++){
memset(maxx,0,sizeof(maxx));
memset(mk,0,sizeof(mk));
mk[i]=1;
dfs(i,j,1);
}
}
cout<<ans<<'\n';
return 0;
}