QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#752079 | #9573. Social Media | L_wwqq# | WA | 30ms | 3660kb | C++20 | 1.9kb | 2024-11-15 21:55:26 | 2024-11-15 21:55:26 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
int T;
cin >> T;
while(T -- ) {
int n, m, k;
cin >> n >> m >> k;
vector<int> p(k + 1);
for(int i = 1; i <= k; i ++ ) p[i] = i;
auto find = [&](auto &&find, int x) -> int {
if(p[x] != x) p[x] = find(find, p[x]);
return p[x];
};
int now = -1;
for(int i = 0; i < n; i ++ ) {
int x;
cin >> x;
if(now == -1) now = x;
else {
int pnow = find(find, now), px = find(find, x);
if(pnow != px) p[px] = pnow;
}
}
// cout << "!!! " << now << ' ' << find(find, now) << '\n';
vector<int> f(k + 1, 0);
map<pair<int, int>, int> mp;
for(int i = 0; i < m; i ++ ) {
int u, v;
cin >> u >> v;
int pu = find(find, u), pv = find(find, v);
if(pu == pv) {
f[pu] ++ ;
} else {
mp[{pu, pv}] ++ ;
mp[{pv, pu}] ++ ;
}
}
int ans = 0, val1 = 0, val2 = 0, id1 = -1, id2 = -1;
for(int i = 0; i < k; i ++ ) ans = max(ans, f[i]);
for(auto [_, cnt] : mp) {
auto [u, v] = _;
int pnow = find(find, now);
int pu = find(find, u), pv = find(find, v);
if(pu != pnow && pv != pnow) {
int res = f[pu] + f[pv] + f[pnow] + mp[{pu, pv}] + mp[{pu, pnow}] + mp[{pv, pnow}];
ans = max(ans, res);
} else {
if(pu != pnow && pu != id1 && pu != id2) {
int res = f[pu] + mp[{pu, pnow}];
if(res > val1) {
val2 = val1, id2 = id1;
val1 = res, id1 = pu;
} else if(res > val2) {
val2 = res, id2 = pu;
}
} else if (pv != pnow && pv != id1 && pv != id2) {
int res = f[pv] + mp[{pv, pnow}];
if(res > val1) {
val2 = val1, id2 = id1;
val1 = res, id1 = pv;
} else if(res > val2) {
val2 = res, id2 = pv;
}
}
}
}
ans = max(ans, f[find(find, now)] + val1 + val2);
cout << ans << '\n';
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3564kb
input:
5 4 12 7 5 7 3 6 3 6 2 2 1 4 2 4 1 3 7 6 4 1 5 4 1 1 1 1 2 1 3 7 2 7 6 2 4 1 2 3 2 2 5 5 4 2 6 4 6 2 6 1 1 2 1 1 2 2 1 2 1 2 1 2 2 1 100 24 11 11 24
output:
9 5 1 1 1
result:
ok 5 number(s): "9 5 1 1 1"
Test #2:
score: -100
Wrong Answer
time: 30ms
memory: 3660kb
input:
10000 19 12 20 8 12 1 5 11 7 17 13 19 6 3 9 10 15 14 20 4 18 16 4 11 7 1 8 4 16 19 1 13 15 2 16 2 8 7 3 15 11 13 5 20 18 14 17 14 20 2 9 1 12 8 11 10 17 18 16 3 15 5 14 20 13 7 15 10 3 2 5 16 7 8 6 1 6 4 18 16 1 8 4 1 20 6 6 9 4 15 7 5 14 9 1 3 18 9 15 18 17 15 11 14 7 19 7 3 1 2 5 6 4 7 5 1 4 5 3 1...
output:
12 14 1 19 6 5 1 11 19 4 3 10 4 1 4 19 15 2 18 4 17 5 1 2 5 17 3 2 6 15 6 3 6 4 4 7 3 7 4 1 16 15 2 3 6 12 12 7 6 8 8 6 8 11 16 1 4 9 8 14 2 6 19 19 16 8 20 14 8 12 7 9 6 8 2 17 9 5 5 3 6 6 20 8 13 11 10 5 3 5 3 1 8 5 8 11 7 14 10 9 8 11 7 9 5 2 8 14 10 5 3 5 5 10 1 6 9 16 5 3 19 1 4 8 8 10 4 2 1 15...
result:
wrong answer 90th numbers differ - expected: '6', found: '5'