比赛 |
CSP2023-J模拟赛 |
评测结果 |
AAAATTTTTT |
题目名称 |
新建题目 |
最终得分 |
40 |
用户昵称 |
心灵震荡 |
运行时间 |
6.000 s |
代码语言 |
C++ |
内存使用 |
14.89 MiB |
提交时间 |
2023-10-18 19:23:08 |
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
const int N = 2005;
int n, w[N], x, y, v, head[N], dep[N], mp[N][N], ans[N], res, tot;
bool vis[N], flag;
vector<int> g[N];
struct edge {int to, nxt;} e[N << 1];
void add_edge(int u, int v) {e[++tot] = {v, head[u]}, head[u] = tot;}
inline bool check(int i, int j)
{
return (dep[i] > dep[j] && w[i] <= w[j]);
}
void dfs1(int u, int fa)
{
dep[u] = dep[fa] + 1;
for(int i = head[u]; i; i = e[i].nxt)
if(e[i].to != fa) dfs1(e[i].to, u);
return;
}
void dfs2(int u, int t)
{
if(flag) return;
if(u == v)
{
flag = 1;
for(int i = 1; i <= t; i++)
for(int j = i + 1; j <= t; j++)
if(check(ans[i], ans[j]) || check(ans[j], ans[i])) return;
return (void) (res++);
}
for(int i = head[u]; i; i = e[i].nxt)
{
if(!vis[e[i].to])
{
vis[e[i].to] = 1, ans[t + 1] = e[i].to;
dfs2(e[i].to, t + 1);
vis[e[i].to] = 0;
}
}
return;
}
int main()
{
freopen("touch.in", "r", stdin);
freopen("touch.out", "w", stdout);
cin >> n;
for(int i = 1; i <= n; i++) cin >> w[i];
for(int i = 1; i < n; i++)
{
cin >> x >> y;
add_edge(x, y);
add_edge(y, x);
}
dfs1(1, 0);
for(int i = 1; i <= n; i++)
for(int j = i + 1; j <= n; j++)
if(!check(i, j)) g[i].push_back(j);
for(int i = 1; i <= n; i++)
for(auto j : g[i])
{
if(i == j) continue;
memset(vis, 0, sizeof vis);
v = j, ans[1] = i, flag = 0, vis[i] = 1;
dfs2(i, 1);
}
cout << res;
return 0;
}