Skip to content

Commit e1d1490

Browse files
committed
trying this
1 parent 9bb46da commit e1d1490

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

tests/library_checker_aizu_tests/edge_cd_asserts.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#pragma once
2-
void edge_cd_asserts(const vector<vector<int>>& adj,
3-
int cent, int split) {
2+
void edge_cd_asserts(const vector<vi>& adj, int cent,
3+
int split) {
44
assert(0 < split && split < sz(adj[cent]));
55
auto dfs = [&](auto&& self, int u, int p) -> int {
66
int siz = 1;

tests/library_checker_aizu_tests/trees/edge_cd_reroot_dp.test.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,16 @@ int main() {
1717
vector<basic_string<int>> adj(n);
1818
vector<mint> b(n - 1), c(n - 1);
1919
vector<pair<int, int>> par(n, {-1, -1});
20+
vector<vector<int>> base_adj(n);
2021
{
2122
vector<vector<pair<int, int>>> adj_with_id(n);
2223
for (int i = 0; i < n - 1; i++) {
2324
int u, v;
2425
cin >> u >> v >> b[i].x >> c[i].x;
2526
adj[u].push_back(v);
2627
adj[v].push_back(u);
28+
base_adj[u].push_back(v);
29+
base_adj[v].push_back(u);
2730
adj_with_id[u].emplace_back(v, i);
2831
adj_with_id[v].emplace_back(u, i);
2932
res[u] = res[u] + b[i] * a[v] + c[i];
@@ -42,7 +45,7 @@ int main() {
4245
assert(u_low ^ v_low);
4346
return u_low ? par[u].second : par[v].second;
4447
};
45-
//{ edge_cd(adj, edge_cd_asserts); }
48+
{ edge_cd(base_adj, edge_cd_asserts); }
4649
edge_cd(adj,
4750
[&](const vector<basic_string<int>>& cd_adj, int cent,
4851
int split) -> void {

0 commit comments

Comments
 (0)