| 16275 | 2025-04-20 09:25:17 | Hora | Testvérvárosok | cpp17 | Wrong answer 0/100 | 609ms | 174764 KiB |
#include <bits/stdc++.h>
using namespace std;
vector<vector<array<int, 2>>> g;
int n, k, ans = 0;
vector<vector<int>> a;
void dfs(int v, int p){
a[v][0] = 1;
for(auto [x, w] : g[v]){
if(x == p) continue;
dfs(x, v);
for(int i = 0; i < k; i++){
ans += a[x][i] * a[v][(2 * k - i - w) % k];
}
for(int i = 0; i < k; i++){
a[v][i] += a[x][(i + w) % k];
}
}
}
int main() {
cin >> n >> k;
g.resize(n + 1);
a.resize(n + 1, vector<int>(k));
for(int i = 0; i < n - 1; i++){
int b, c, d;
cin >> b >> c >> d;
g[b].push_back({c, d});
g[c].push_back({b, d});
}
dfs(1, 0);
cout << ans;
}
| Subtask | Sum | Test | Verdict | Time | Memory | ||
|---|---|---|---|---|---|---|---|
| subtask1 | 0/0 | ||||||
| 1 | Accepted | 1ms | 316 KiB | ||||
| 2 | Wrong answer | 1ms | 508 KiB | ||||
| subtask2 | 0/15 | ||||||
| 3 | Wrong answer | 2ms | 564 KiB | ||||
| 4 | Wrong answer | 1ms | 316 KiB | ||||
| 5 | Wrong answer | 3ms | 780 KiB | ||||
| 6 | Wrong answer | 9ms | 2868 KiB | ||||
| 7 | Accepted | 2ms | 316 KiB | ||||
| 8 | Wrong answer | 6ms | 1940 KiB | ||||
| 9 | Wrong answer | 3ms | 912 KiB | ||||
| subtask3 | 0/15 | ||||||
| 10 | Wrong answer | 17ms | 2012 KiB | ||||
| 11 | Accepted | 4ms | 820 KiB | ||||
| 12 | Wrong answer | 68ms | 6452 KiB | ||||
| 13 | Accepted | 59ms | 6128 KiB | ||||
| 14 | Wrong answer | 68ms | 6768 KiB | ||||
| subtask4 | 0/20 | ||||||
| 15 | Wrong answer | 3ms | 820 KiB | ||||
| 16 | Wrong answer | 13ms | 4148 KiB | ||||
| 17 | Wrong answer | 41ms | 12340 KiB | ||||
| 18 | Wrong answer | 164ms | 45876 KiB | ||||
| 19 | Wrong answer | 123ms | 33128 KiB | ||||
| 20 | Wrong answer | 128ms | 28984 KiB | ||||
| 21 | Accepted | 52ms | 8664 KiB | ||||
| 22 | Wrong answer | 556ms | 174764 KiB | ||||
| subtask5 | 0/50 | ||||||
| 23 | Wrong answer | 444ms | 129332 KiB | ||||
| 24 | Wrong answer | 256ms | 75828 KiB | ||||
| 25 | Wrong answer | 291ms | 76084 KiB | ||||
| 26 | Wrong answer | 108ms | 31028 KiB | ||||
| 27 | Wrong answer | 199ms | 51512 KiB | ||||
| 28 | Wrong answer | 189ms | 43572 KiB | ||||
| 29 | Wrong answer | 609ms | 172340 KiB | ||||
| 30 | Accepted | 68ms | 6196 KiB | ||||
| 31 | Wrong answer | 509ms | 150836 KiB | ||||