| 16287 | 2025-04-20 10:37:48 | Hora | Testvérvárosok | cpp17 | Elfogadva 100/100 | 1.386s | 174132 KiB |
#include <bits/stdc++.h>
using namespace std;
vector<vector<array<int, 2>>> g;
long long 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 += (long long)a[x][i] * (long long)a[v][(4 * k - i - w) % k];
}
for(int i = 0; i < k; i++){
a[v][(i + w) % k] += a[x][i];
}
}
}
int main() {
cin >> n >> k;
g.resize(n + 1);
a.resize(n + 1, vector<int>(k, 0));
for(int i = 0; i < n - 1; i++){
int 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";
}*/
}
| Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
|---|---|---|---|---|---|---|---|
| subtask1 | 0/0 | ||||||
| 1 | Elfogadva | 1ms | 316 KiB | ||||
| 2 | Elfogadva | 1ms | 316 KiB | ||||
| subtask2 | 15/15 | ||||||
| 3 | Elfogadva | 2ms | 564 KiB | ||||
| 4 | Elfogadva | 1ms | 316 KiB | ||||
| 5 | Elfogadva | 4ms | 776 KiB | ||||
| 6 | Elfogadva | 20ms | 2988 KiB | ||||
| 7 | Elfogadva | 2ms | 316 KiB | ||||
| 8 | Elfogadva | 14ms | 2080 KiB | ||||
| 9 | Elfogadva | 4ms | 900 KiB | ||||
| subtask3 | 15/15 | ||||||
| 10 | Elfogadva | 16ms | 1848 KiB | ||||
| 11 | Elfogadva | 4ms | 820 KiB | ||||
| 12 | Elfogadva | 64ms | 5940 KiB | ||||
| 13 | Elfogadva | 61ms | 5312 KiB | ||||
| 14 | Elfogadva | 71ms | 6048 KiB | ||||
| subtask4 | 20/20 | ||||||
| 15 | Elfogadva | 6ms | 1044 KiB | ||||
| 16 | Elfogadva | 30ms | 4148 KiB | ||||
| 17 | Elfogadva | 93ms | 12132 KiB | ||||
| 18 | Elfogadva | 347ms | 45552 KiB | ||||
| 19 | Elfogadva | 248ms | 32820 KiB | ||||
| 20 | Elfogadva | 222ms | 28212 KiB | ||||
| 21 | Elfogadva | 54ms | 7988 KiB | ||||
| 22 | Elfogadva | 1.363s | 174132 KiB | ||||
| subtask5 | 50/50 | ||||||
| 23 | Elfogadva | 1.054s | 128564 KiB | ||||
| 24 | Elfogadva | 615ms | 75316 KiB | ||||
| 25 | Elfogadva | 601ms | 75596 KiB | ||||
| 26 | Elfogadva | 240ms | 31028 KiB | ||||
| 27 | Elfogadva | 405ms | 51252 KiB | ||||
| 28 | Elfogadva | 363ms | 42956 KiB | ||||
| 29 | Elfogadva | 1.386s | 171572 KiB | ||||
| 30 | Elfogadva | 61ms | 5684 KiB | ||||
| 31 | Elfogadva | 1.187s | 150068 KiB | ||||