Submission #1626641
Source Code Expand
#include <bits/stdc++.h> #define xx first #define yy second #define mp make_pair #define pb push_back #define mset(x, y) memset(x, y, sizeof x) #define mcpy(x, y) memcpy(x, y, sizeof x) using namespace std; typedef long long LL; typedef pair <int, int> pii; inline int Read() { int x = 0, f = 1, c = getchar(); for (; !isdigit(c); c = getchar()) if (c == '-') f = -1; for (; isdigit(c); c = getchar()) x = x * 10 + c - '0'; return x * f; } const int MAXN = 3005; int n, a[MAXN], par[MAXN]; vector <int> adj[MAXN]; bool f[MAXN], g[MAXN]; inline void Dfs(int x, int p) { if (p) adj[x].erase(find(adj[x].begin(), adj[x].end(), p)); par[x] = p; for (auto y : adj[x]) Dfs(y, x), f[x] |= !f[y] && a[x] > a[y]; } inline void Dfs(int x) { int pre = 0, suf = 0; for (auto y : adj[x]) suf += !f[y] && a[x] > a[y]; for (int i = 0; i < adj[x].size(); i ++) { if (i) pre += !f[adj[x][i - 1]] && a[x] > a[adj[x][i - 1]]; suf -= !f[adj[x][i]] && a[x] > a[adj[x][i]]; if ((!g[x] && a[x] > a[par[x]]) || pre || suf) g[adj[x][i]] = 1; Dfs(adj[x][i]); } } int main() { #ifdef wxh010910 freopen("data.in", "r", stdin); #endif n = Read(); for (int i = 1; i <= n; i ++) a[i] = Read(); for (int i = 1, x, y; i < n; i ++) x = Read(), y = Read(), adj[x].pb(y), adj[y].pb(x); a[0] = INT_MAX; Dfs(1, 0); Dfs(1); for (int i = 1; i <= n; i ++) if (f[i] || (!g[i] && a[i] > a[par[i]])) printf("%d ", i); putchar(10); return 0; }
Submission Info
Submission Time | |
---|---|
Task | F - Tree Game |
User | wxh010910 |
Language | C++14 (GCC 5.4.1) |
Score | 1600 |
Code Size | 1538 Byte |
Status | AC |
Exec Time | 2 ms |
Memory | 512 KB |
Judge Result
Set Name | Sample | All | ||||
---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 1600 / 1600 | ||||
Status |
|
|
Set Name | Test Cases |
---|---|
Sample | sample1.txt, sample2.txt, sample3.txt |
All | sample1.txt, sample2.txt, sample3.txt, in1.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in2.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, in29.txt, in3.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in4.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in46.txt, in47.txt, in48.txt, in49.txt, in5.txt, in50.txt, in6.txt, in7.txt, in8.txt, in9.txt, sample1.txt, sample2.txt, sample3.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
in1.txt | AC | 2 ms | 384 KB |
in10.txt | AC | 2 ms | 384 KB |
in11.txt | AC | 2 ms | 384 KB |
in12.txt | AC | 1 ms | 384 KB |
in13.txt | AC | 2 ms | 384 KB |
in14.txt | AC | 2 ms | 384 KB |
in15.txt | AC | 2 ms | 384 KB |
in16.txt | AC | 2 ms | 384 KB |
in17.txt | AC | 2 ms | 384 KB |
in18.txt | AC | 2 ms | 384 KB |
in19.txt | AC | 2 ms | 384 KB |
in2.txt | AC | 2 ms | 384 KB |
in20.txt | AC | 2 ms | 384 KB |
in21.txt | AC | 2 ms | 512 KB |
in22.txt | AC | 2 ms | 512 KB |
in23.txt | AC | 2 ms | 512 KB |
in24.txt | AC | 2 ms | 512 KB |
in25.txt | AC | 2 ms | 512 KB |
in26.txt | AC | 2 ms | 512 KB |
in27.txt | AC | 2 ms | 512 KB |
in28.txt | AC | 2 ms | 512 KB |
in29.txt | AC | 2 ms | 512 KB |
in3.txt | AC | 2 ms | 384 KB |
in30.txt | AC | 2 ms | 512 KB |
in31.txt | AC | 2 ms | 512 KB |
in32.txt | AC | 2 ms | 512 KB |
in33.txt | AC | 2 ms | 512 KB |
in34.txt | AC | 2 ms | 512 KB |
in35.txt | AC | 2 ms | 512 KB |
in36.txt | AC | 1 ms | 384 KB |
in37.txt | AC | 1 ms | 384 KB |
in38.txt | AC | 1 ms | 384 KB |
in39.txt | AC | 1 ms | 384 KB |
in4.txt | AC | 2 ms | 384 KB |
in40.txt | AC | 2 ms | 384 KB |
in41.txt | AC | 2 ms | 384 KB |
in42.txt | AC | 2 ms | 384 KB |
in43.txt | AC | 2 ms | 384 KB |
in44.txt | AC | 2 ms | 384 KB |
in45.txt | AC | 2 ms | 384 KB |
in46.txt | AC | 2 ms | 384 KB |
in47.txt | AC | 2 ms | 384 KB |
in48.txt | AC | 2 ms | 384 KB |
in49.txt | AC | 2 ms | 384 KB |
in5.txt | AC | 2 ms | 384 KB |
in50.txt | AC | 2 ms | 384 KB |
in6.txt | AC | 2 ms | 384 KB |
in7.txt | AC | 2 ms | 384 KB |
in8.txt | AC | 2 ms | 384 KB |
in9.txt | AC | 2 ms | 384 KB |
sample1.txt | AC | 1 ms | 384 KB |
sample2.txt | AC | 1 ms | 384 KB |
sample3.txt | AC | 1 ms | 384 KB |