QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#764024 | #2766. Unique Cities | _8_8_ | 0 | 200ms | 91568kb | C++20 | 2.8kb | 2024-11-19 23:25:25 | 2024-11-19 23:25:28 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = (1 << 20) + 12, MOD = (int)1e9 + 7;
int n, m, a[N], vis[N], timer, p[N];
vector<int> g[N], gt[N];
bool is[N];
int f(int d) {
timer++;
vis[d] = timer;
int lst = d;
queue<int> q;
q.push(d);
while(!q.empty()) {
int v = q.front();
lst = v;
q.pop();
for(int to : g[v]) {
if(vis[to] != timer) {
q.push(to);
p[to] = v;
vis[to] = timer;
}
}
}
int x = lst;
timer++;
vis[d] = timer;
while(x != d) {
vis[x] = timer;
x = p[x];
}
return lst;
}
int dist[N], col[N], dep[N], mxd[N], res[N];
void dfs(int v, int pr = -1) {
mxd[v] = dep[v];
for(int to:g[v]) if(to != pr) {
gt[v].push_back(to);
dep[to] = dep[v] + 1;
dfs(to, v);
mxd[v] = max(mxd[v], mxd[to]);
}
}
vector<int> st;
set<int> cur[N];
void go(int v) {
set<int> raz;
for(int i = 0; i < (int)st.size() && mxd[v] - dep[v] < dep[v] - dep[st[i]]; i++) {
raz.insert(a[st[i]]);
cur[v].insert(a[st[i]]);
}
res[v] += (int)raz.size();
if(gt[v].empty()) return;
int sz = (int)gt[v].size();
vector<int> p(sz), s(sz);
p[0] = mxd[gt[v][0]];
s[sz - 1] = mxd[gt[v][sz - 1]];
for(int i = 1; i < sz; i++) {
p[i] = max(p[i - 1], mxd[gt[v][i]]);
}
for(int i = sz - 1; i >= 0; i--) {
s[i] = max(s[i + 1], mxd[gt[v][i]]);
}
for(int i = 0; i < sz; i++) {
int to = gt[v][i];
int mx = dep[v];
if(i) mx = max(mx, p[i - 1]);
if(i < sz - 1) mx = max(mx, s[i + 1]);
mx -= dep[v];
vector<int> bf = st;
while(!st.empty() && dep[v] - dep[st.back()] <= mx) {
st.pop_back();
}
st.push_back(v);
go(to);
st = bf;
}
}
void solve(int root) {
for(int i = 1; i <= n; i++) {
gt[i].clear();
}
st.clear();
dep[root] = 1;
dfs(root);
go(root);
}
void test() {
cin >> n >> m;
for(int i = 1; i <= n - 1; i++) {
int a, b;
cin >> a >> b;
g[a].push_back(b);
g[b].push_back(a);
}
for(int i = 1; i <= n; i++) {
cin >> a[i];
if((int)g[i].size() == 1) {
is[i] = 1;
}
}
int d = f(1), d1 = f(d);
solve(d);solve(d1);
for(int i = 1; i <= n; i++) {
// if(is[i] && i != d && i != d1) res[i]--;
// cout << res[i] << '\n';
cout << (int)cur[i].size() << '\n';
}
cout << '\n';
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
int t = 1;
// cin >> t;
while(t--)
test();
}
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 0
Wrong Answer
Test #1:
score: 4
Accepted
time: 4ms
memory: 64200kb
input:
2 1 1 2 1 1
output:
1 1
result:
ok 2 lines
Test #2:
score: 0
Wrong Answer
time: 13ms
memory: 58128kb
input:
1842 848 740 1093 1299 922 801 1560 265 1664 1043 65 1430 427 80 233 4 1238 1623 1473 1569 274 953 1485 1259 649 1671 1409 246 542 742 1517 720 1120 1527 1328 1167 1531 1056 1130 673 1222 192 980 1393 913 446 688 135 23 1065 1787 978 1481 1765 1720 310 202 1406 1451 475 523 104 774 1531 829 169 396 ...
output:
1 1 1 0 0 0 1 0 2 0 1 1 0 3 0 0 0 2 0 0 4 1 0 2 4 0 0 1 0 0 1 1 1 1 0 0 0 3 0 1 1 0 1 1 4 0 0 1 1 3 0 2 1 0 1 1 0 0 2 1 1 2 1 1 0 0 3 0 1 1 0 0 1 0 0 0 1 1 3 1 4 2 0 2 2 0 1 3 0 2 1 0 1 0 1 1 1 1 3 1 0 1 2 1 0 1 1 4 0 2 0 0 0 4 0 1 1 1 2 2 3 0 5 1 1 5 0 1 1 0 0 2 0 2 0 1 4 0 1 0 3 0 2 0 0 1 0 1 0 2 ...
result:
wrong answer 7th lines differ - expected: '2', found: '1'
Subtask #2:
score: 0
Wrong Answer
Test #33:
score: 0
Wrong Answer
time: 93ms
memory: 78356kb
input:
115391 1 32067 50006 1710 5850 21016 66364 72998 34367 24783 10670 49950 93666 81835 81234 53480 68963 87357 43320 93905 30509 72528 92224 520 100511 54804 2917 58490 23858 93643 87530 90737 65205 60740 110812 9553 90266 70028 67222 108045 88982 35584 110286 53518 21733 108735 26404 108228 109796 92...
output:
0 1 0 0 0 1 1 0 1 1 1 1 0 1 1 0 1 0 0 0 1 0 1 0 0 0 0 0 1 1 1 0 1 0 0 0 1 1 1 0 1 1 1 0 0 0 0 1 1 1 1 1 0 1 1 0 0 0 1 1 1 1 0 0 0 0 1 0 0 0 1 0 1 1 1 0 1 0 1 1 1 0 1 1 0 1 0 0 1 0 0 0 1 0 0 1 1 0 1 1 1 1 1 0 1 1 0 0 0 1 1 0 1 1 1 0 0 0 1 1 1 1 0 1 1 0 0 0 0 1 1 1 0 1 1 0 0 1 1 0 1 0 1 0 1 1 1 1 1 0 ...
result:
wrong answer 1st lines differ - expected: '1', found: '0'
Subtask #3:
score: 0
Wrong Answer
Test #50:
score: 0
Wrong Answer
time: 200ms
memory: 91568kb
input:
157976 157976 20171 157173 44732 54119 107845 121149 109200 110309 82678 108206 89140 64200 36916 128759 3966 123760 92978 105716 43700 146126 14924 3429 107721 36095 94906 78173 97162 29552 119574 39605 25145 138492 16622 99431 60281 7949 76176 136644 75716 91518 127987 110605 77999 110960 101187 5...
output:
0 4 2 2 3 1 2 1 1 0 1 0 0 0 2 0 6 1 0 2 0 1 1 0 0 1 0 2 0 3 1 0 0 0 0 0 2 0 1 4 3 5 1 1 0 0 1 2 3 0 0 0 0 3 1 0 2 1 1 0 0 3 1 2 0 2 0 1 0 0 0 0 0 2 1 0 0 1 1 0 0 5 1 2 0 0 2 0 1 2 1 1 1 1 0 3 1 0 0 0 1 0 2 0 0 0 2 1 0 1 2 1 0 0 0 1 0 0 1 1 0 1 4 1 1 0 1 0 0 4 1 0 1 0 2 0 0 0 0 2 2 0 6 5 2 0 2 1 1 5 ...
result:
wrong answer 1st lines differ - expected: '1', found: '0'
Subtask #4:
score: 0
Skipped
Dependency #1:
0%