QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#382425 | #905. 三元环枚举 | Isrothy | WA | 15ms | 6132kb | C++23 | 1.5kb | 2024-04-08 14:03:23 | 2024-04-08 14:03:24 |
Judging History
answer
#include <cstdio>
#include <span>
#include <vector>
constexpr int mod = 998244353;
auto three_membered_rings(std::span<std::vector<int>> adj) {
auto n = adj.size();
std::vector<size_t> rank(n);
std::vector<int> vis_time(n);
std::vector<std::vector<int>> f(n);
for (int u = 0; u < n; ++u) {
rank[u] = adj[u].size() * (n + 1) + u;
}
for (int u = 0; u < n; ++u) {
for (auto v: adj[u]) {
if (rank[u] < rank[v]) {
f[u].push_back(v);
}
}
}
std::vector<std::tuple<int, int, int>> rings;
for (int u = 0; u < n; ++u) {
for (auto v: f[u]) {
vis_time[v] = u;
}
for (auto v: f[u]) {
for (auto w: f[v]) {
if (vis_time[w] == u) {
rings.emplace_back(u, v, w);
}
}
}
}
return rings;
}
int main() {
int n, m;
scanf("%d%d", &n, &m);
std::vector<std::vector<int>> adj(n);
std::vector<int> val(n);
for (int i = 0; i < n; ++i) {
scanf("%d", &val[i]);
}
for (int i = 0; i < m; ++i) {
int u, v;
scanf("%d%d", &u, &v);
adj[u].push_back(v);
adj[v].push_back(u);
}
auto rings = three_membered_rings(adj);
long long sum = 0;
for (auto [u, v, w]: rings) {
sum = (sum + (long long) val[u] * val[v] % mod * val[w]) % mod;
}
printf("%lld\n", sum);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3804kb
input:
4 5 1 2 3 4 0 3 2 0 2 1 2 3 1 3
output:
36
result:
ok "36"
Test #2:
score: -100
Wrong Answer
time: 15ms
memory: 6132kb
input:
17707 77101 528756313 434883274 318065816 264440383 659789617 608119380 648104885 725454492 696703871 543030428 663661240 890791532 108201616 428505484 322953840 119811886 691103780 306647414 549862302 176916719 909058872 455464665 307270851 584469329 722629343 875317523 629938577 244419357 78121457...
output:
881677485
result:
wrong answer 1st words differ - expected: '397965084', found: '881677485'