QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#768549 | #2766. Unique Cities | _8_8_ | 0 | 128ms | 24324kb | C++20 | 3.8kb | 2024-11-21 12:09:19 | 2024-11-21 12:09:20 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
const int N = 2e5 + 12, MOD = (int)1e9 + 7;
int n, m, a[N], vis[N], timer, p[N];
vector<int> g[N], gt[N];
bool is[N];
vector<vector<int>> dist;
int f(int d) {
vector<int> D(n + 1, 0);
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) {
D[to] = D[v] + 1;
q.push(to);
vis[to] = timer;
}
}
}
dist.push_back(D);
return lst;
}
int col[N], dep[N], mxd[N], res[N], d, d1, o, cr;
int zap = 0;
bool ok[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);
if(ok[to]) ok[v] = 1;
mxd[v] = max(mxd[v], mxd[to]);
}
for(int i = 0; i < (int)gt[v].size(); i++) {
if(mxd[gt[v][i]] == mxd[v]) {
swap(gt[v][0], gt[v][i]);
}
}
if(v == d) ok[v] = 1;
}
vector<int> st;
multiset<int> cur;
void go(int v) {
if(dist[cr][v] > dist[o][v] || (dist[cr][v] == dist[o][v] && zap)) {
// set<int> raz;
int l = -1,r = (int)st.size();
while(r - l > 1) {
int mid = (l + r) >> 1;
if(mxd[v] - dep[v] < dep[v] - dep[st[mid]]) {
l = mid;
} else {
r = mid;
}
}
res[v] += (l + 1);
// 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 - 2; i >= 0; i--) {
s[i] = max(s[i + 1], mxd[gt[v][i]]);
}
vector<int> del;
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];
if(i == 1 && !st.empty() && st.back() == v) {
cur.erase(cur.find(a[v]));
st.pop_back();
}
if(i <= 1) {
while(!st.empty() && dep[v] - dep[st.back()] <= mx) {
del.push_back(st.back());
cur.erase(cur.find((a[st.back()])));
st.pop_back();
}
}
if((i <= 1 && (cur.find(a[v]) == cur.end())) && (!zap || ok[v])) {
cur.insert(a[v]);
st.push_back(v);
}
go(to);
}
if(!st.empty() && st.back() == v) {
cur.erase(cur.find(a[v]));
st.pop_back();
}
reverse(del.begin(), del.end());
for(int j:del) {
st.push_back(j);
cur.insert(a[j]);
}
}
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];
}
d = f(1), d1 = f(d);
f(d1);
o = 2;cr = 1;
solve(d);
zap = 1;
o = 1;cr = 2;
solve(d1);
for(int i = 1; i <= n; i++) {
cout << res[i] << '\n';
}
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
int t = 1;
// cin >> t;
while(t--)
test();
}
詳細信息
Subtask #1:
score: 0
Wrong Answer
Test #1:
score: 4
Accepted
time: 0ms
memory: 5592kb
input:
2 1 1 2 1 1
output:
1 1
result:
ok 2 lines
Test #2:
score: 0
Wrong Answer
time: 0ms
memory: 8096kb
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 1 0 0 0 2 0 0 1 2 0 1 1 0 0 1 1 1 1 1 1 1 0 0 1 3 0 1 2 0 1 1 4 0 1 1 1 3 0 1 1 0 1 1 0 0 2 1 1 2 1 1 1 0 3 0 1 1 0 0 1 0 0 0 1 1 3 1 1 1 1 2 2 0 1 3 0 1 1 0 1 1 1 1 1 1 3 1 0 1 1 1 0 1 1 4 1 1 0 0 0 4 0 1 1 2 2 2 1 0 5 2 1 5 0 1 1 0 0 1 0 2 0 1 4 0 1 0 3 0 1 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: 76ms
memory: 20444kb
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:
1 1 0 1 0 1 1 0 1 1 1 0 1 0 1 1 0 0 1 0 1 0 1 1 1 1 1 1 0 1 1 1 1 1 1 1 0 0 1 0 1 1 0 1 1 1 0 0 1 0 0 1 1 0 1 1 1 0 1 1 1 0 1 0 0 1 1 0 1 0 0 1 0 0 1 1 1 1 1 1 0 1 0 0 1 1 0 1 0 1 1 0 1 1 0 0 0 0 0 1 0 1 0 1 0 1 0 1 0 0 0 0 0 1 1 1 1 0 1 0 1 1 0 1 1 0 1 0 1 1 1 1 1 1 1 0 1 1 1 1 0 1 0 1 1 1 1 1 1 1 ...
result:
wrong answer 12th lines differ - expected: '1', found: '0'
Subtask #3:
score: 0
Wrong Answer
Test #50:
score: 0
Wrong Answer
time: 128ms
memory: 24324kb
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:
1 5 3 3 4 2 3 2 2 1 2 1 4 1 3 1 7 2 1 3 1 4 2 1 1 2 1 3 1 4 2 1 1 1 1 1 3 1 2 5 4 6 2 2 1 1 2 3 4 1 1 1 1 4 2 4 3 2 2 1 1 4 2 3 1 3 1 2 1 1 1 1 1 3 2 1 1 4 2 1 1 4 2 3 1 1 3 1 2 3 2 2 2 3 1 4 2 1 1 1 2 1 3 1 4 1 4 2 1 2 3 2 1 1 1 2 1 1 2 2 1 2 4 2 2 1 2 1 1 5 2 1 2 1 3 1 4 1 1 3 3 1 4 6 3 1 3 2 2 4 ...
result:
wrong answer 22nd lines differ - expected: '5', found: '4'
Subtask #4:
score: 0
Skipped
Dependency #1:
0%