QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#768563 | #2766. Unique Cities | _8_8_ | 0 | 123ms | 25156kb | C++20 | 3.2kb | 2024-11-21 12:19:07 | 2024-11-21 12:19:08 |
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;
void go(int v) {
if(dist[cr][v] > dist[o][v] || (dist[cr][v] == dist[o][v] && zap)) {
for(int i = 0; i < (int)st.size() && mxd[v] - dep[v] < dep[v] - dep[st[i]]; i++) {
res[v]++;
}
}
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.pop_back();
if(i <= 1) {
while(!st.empty() && dep[v] - dep[st.back()] <= mx) {
del.push_back(st.back());
st.pop_back();
}
}
if(i <= 1) st.push_back(v);
go(to);
}
st.pop_back();
reverse(del.begin(), del.end());
for(int j:del) {
st.push_back(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: 2ms
memory: 7664kb
input:
2 1 1 2 1 1
output:
1 1
result:
ok 2 lines
Test #2:
score: 0
Wrong Answer
time: 3ms
memory: 7892kb
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 2 0 3 0 2 1 0 3 0 0 0 2 0 0 4 2 0 2 4 0 0 1 1 1 2 1 1 1 0 0 1 3 0 1 2 0 1 1 4 0 1 1 1 3 0 2 1 0 1 1 0 0 2 1 1 2 1 1 1 0 3 0 1 2 0 0 1 0 0 0 1 1 3 1 4 3 1 2 2 0 1 3 0 2 1 0 1 1 1 1 1 2 3 1 0 1 2 1 0 1 2 4 1 3 0 0 0 4 0 1 1 2 2 2 3 0 5 2 1 5 0 1 1 0 0 2 0 2 0 2 4 0 1 0 3 0 3 0 0 2 0 1 0 2 ...
result:
wrong answer 766th lines differ - expected: '5', found: '6'
Subtask #2:
score: 0
Wrong Answer
Test #33:
score: 0
Wrong Answer
time: 84ms
memory: 19524kb
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 3 0 1 0 3 2 0 3 3 4 5 1 1 3 1 1 0 1 0 4 0 2 1 1 1 1 1 3 2 2 1 2 1 1 1 1 1 5 0 7 2 2 1 1 1 0 1 4 2 2 4 1 1 3 1 1 0 2 2 2 1 1 0 0 1 2 0 1 0 1 1 1 2 2 1 3 1 2 2 1 1 1 2 1 2 0 1 2 1 1 0 3 1 0 3 2 0 3 3 2 5 1 1 4 3 0 1 0 1 1 0 2 2 2 1 1 0 2 1 3 3 0 2 2 0 1 0 1 5 4 2 1 2 5 0 1 2 5 1 1 1 1 1 2 2 5 2 5 1 ...
result:
wrong answer 2nd lines differ - expected: '1', found: '3'
Subtask #3:
score: 0
Time Limit Exceeded
Test #50:
score: 32
Accepted
time: 123ms
memory: 25156kb
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 5 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 5 2 1 1 9 2 3 1 1 3 1 2 3 2 2 2 3 1 4 2 1 1 1 2 1 3 1 4 1 6 2 1 2 3 2 1 1 1 2 1 1 2 2 1 2 8 2 2 1 2 1 1 5 2 1 2 1 3 1 4 1 1 3 3 1 10 6 3 1 3 2 2 9...
result:
ok 157976 lines
Test #51:
score: 0
Time Limit Exceeded
input:
191205 191205 92326 99995 188142 63029 71973 50899 68069 72878 159479 103397 143191 93854 27574 146981 63427 186167 104978 177629 22355 188155 184293 15184 101908 143833 111535 144687 112209 61685 59401 128258 97717 87496 168079 130237 49319 190230 119632 98368 76193 184597 22790 136093 1080 80811 1...
output:
result:
Subtask #4:
score: 0
Skipped
Dependency #1:
0%