| 16278 | 2025-04-20 09:35:05 | Hora | Testvérvárosok | cpp17 | Runtime error 30/100 | 1.185s | 262144 KiB |
#include <bits/stdc++.h>
using namespace std;
vector<vector<array<long long, 2>>> g;
long long n, k, ans = 0;
vector<vector<long long>> a;
void dfs(long long v, long long p){
a[v][0] = 1;
for(auto [x, w] : g[v]){
if(x == p) continue;
dfs(x, v);
for(long long i = 0; i < k; i++){
ans += a[x][i] * a[v][(4 * k - i - w) % k];
}
for(long long i = 0; i < k; i++){
a[v][i] += a[x][(i - w + 4 * k) % k];
}
}
}
int main() {
cin >> n >> k;
g.resize(n + 1);
a.resize(n + 1, vector<long long>(k));
for(long long i = 0; i < n - 1; i++){
long long b, c, d;
cin >> b >> c >> d;
g[b].push_back({c, d % k});
g[c].push_back({b, d % k});
}
dfs(1, 0);
cout << ans << "\n";
/*for(auto x : a){
for(auto y : x) cout << y << " ";
cout << "\n";
}*/
}
| Subtask | Sum | Test | Verdict | Time | Memory | ||
|---|---|---|---|---|---|---|---|
| subtask1 | 0/0 | ||||||
| 1 | Accepted | 1ms | 316 KiB | ||||
| 2 | Accepted | 1ms | 316 KiB | ||||
| subtask2 | 15/15 | ||||||
| 3 | Accepted | 3ms | 564 KiB | ||||
| 4 | Accepted | 1ms | 316 KiB | ||||
| 5 | Accepted | 4ms | 1072 KiB | ||||
| 6 | Accepted | 23ms | 5516 KiB | ||||
| 7 | Accepted | 2ms | 316 KiB | ||||
| 8 | Accepted | 16ms | 3688 KiB | ||||
| 9 | Accepted | 4ms | 1332 KiB | ||||
| subtask3 | 15/15 | ||||||
| 10 | Accepted | 17ms | 2104 KiB | ||||
| 11 | Accepted | 4ms | 1104 KiB | ||||
| 12 | Accepted | 75ms | 6732 KiB | ||||
| 13 | Accepted | 67ms | 6068 KiB | ||||
| 14 | Accepted | 71ms | 6884 KiB | ||||
| subtask4 | 0/20 | ||||||
| 15 | Accepted | 6ms | 1780 KiB | ||||
| 16 | Accepted | 34ms | 8092 KiB | ||||
| 17 | Accepted | 108ms | 23092 KiB | ||||
| 18 | Accepted | 405ms | 87604 KiB | ||||
| 19 | Accepted | 273ms | 60008 KiB | ||||
| 20 | Accepted | 234ms | 50064 KiB | ||||
| 21 | Accepted | 56ms | 8660 KiB | ||||
| 22 | Runtime error | 232ms | 262144 KiB | ||||
| subtask5 | 0/50 | ||||||
| 23 | Accepted | 1.185s | 253240 KiB | ||||
| 24 | Accepted | 661ms | 148852 KiB | ||||
| 25 | Accepted | 667ms | 148020 KiB | ||||
| 26 | Accepted | 280ms | 60984 KiB | ||||
| 27 | Accepted | 474ms | 99892 KiB | ||||
| 28 | Accepted | 404ms | 80692 KiB | ||||
| 29 | Runtime error | 197ms | 262144 KiB | ||||
| 30 | Accepted | 64ms | 6456 KiB | ||||
| 31 | Runtime error | 194ms | 262144 KiB | ||||