#line 2 "/home/shogo314/cpp_include/ou-library/io.hpp" /** * @file io.hpp * @brief 空白区切り出力、iostreamのオーバーロード */ #include #include #include #include #include namespace tuple_io { template std::basic_istream& read_tuple(std::basic_istream& is, Tuple& t) { is >> std::get(t); if constexpr (I + 1 < std::tuple_size_v) { return read_tuple(is, t); } return is; } template std::basic_ostream& write_tuple(std::basic_ostream& os, const Tuple& t) { os << std::get(t); if constexpr (I + 1 < std::tuple_size_v) { os << CharT(' '); return write_tuple(os, t); } return os; } }; template std::basic_istream& operator>>(std::basic_istream& is, std::pair& p) { is >> p.first >> p.second; return is; } template std::basic_istream& operator>>(std::basic_istream& is, std::tuple& p) { return tuple_io::read_tuple, 0>(is, p); } template std::basic_istream& operator>>(std::basic_istream& is, std::array& a) { for(auto& e : a) is >> e; return is; } template std::basic_istream& operator>>(std::basic_istream& is, std::vector& v) { for(auto& e : v) is >> e; return is; } template std::basic_ostream& operator<<(std::basic_ostream& os, const std::pair& p) { os << p.first << CharT(' ') << p.second; return os; } template std::basic_ostream& operator<<(std::basic_ostream& os, const std::tuple& p) { return tuple_io::write_tuple, 0>(os, p); } template std::basic_ostream& operator<<(std::basic_ostream& os, const std::array& a) { for(size_t i = 0; i < N; ++i) { if(i) os << CharT(' '); os << a[i]; } return os; } template std::basic_ostream& operator<<(std::basic_ostream& os, const std::vector& v) { for(size_t i = 0; i < v.size(); ++i) { if(i) os << CharT(' '); os << v[i]; } return os; } /** * @brief 空行出力 */ void print() { std::cout << '\n'; } /** * @brief 出力して改行 * * @tparam T 型 * @param x 出力する値 */ template void print(const T& x) { std::cout << x << '\n'; } /** * @brief 空白区切りで出力して改行 * * @tparam T 1つ目の要素の型 * @tparam Tail 2つ目以降の要素の型 * @param x 1つ目の要素 * @param tail 2つ目以降の要素 */ template void print(const T& x, const Tail&... tail) { std::cout << x << ' '; print(tail...); } #line 2 "/home/shogo314/cpp_include/ou-library/tree.hpp" /** * @file tree.hpp * @brief 木の汎用テンプレート */ #include #include #line 11 "/home/shogo314/cpp_include/ou-library/tree.hpp" #line 2 "/home/shogo314/cpp_include/ou-library/graph.hpp" #line 4 "/home/shogo314/cpp_include/ou-library/graph.hpp" #include #include #line 7 "/home/shogo314/cpp_include/ou-library/graph.hpp" /** * @brief グラフの汎用クラス * * @tparam Cost 辺のコストの型 */ template struct Graph { /** * @brief 有向辺の構造体 * * operator int()を定義しているので、int型にキャストすると勝手にdstになる * 例えば、 * for (auto& e : g[v]) をすると、vから出る辺が列挙されるが、 * for (int dst : g[v]) とすると、vから出る辺の行き先が列挙される */ struct Edge { int src; //!< 始点 int dst; //!< 終点 Cost cost; //!< コスト int id; //!< 辺の番号(追加された順、無向辺の場合はidが同じで方向が逆のものが2つ存在する) Edge() = default; Edge(int src, int dst, Cost cost=1, int id=-1) : src(src), dst(dst), cost(cost), id(id) {} operator int() const { return dst; } }; int n; //!< 頂点数 int m; //!< 辺数 std::vector> g; //!< グラフの隣接リスト表現 /** * @brief デフォルトコンストラクタ */ Graph() : n(0), m(0), g(0) {} /** * @brief コンストラクタ * @param n 頂点数 */ explicit Graph(int n) : n(n), m(0), g(n) {} /** * @brief 無向辺を追加する * @param u 始点 * @param v 終点 * @param w コスト 省略したら1 */ void add_edge(int u, int v, Cost w=1) { g[u].push_back({u, v, w, m}); g[v].push_back({v, u, w, m++}); } /** * @brief 有向辺を追加する * @param u 始点 * @param v 終点 * @param w コスト 省略したら1 */ void add_directed_edge(int u, int v, Cost w=1) { g[u].push_back({u, v, w, m++}); } /** * @brief 辺の情報を標準入力から受け取って追加する * @param m 辺の数 * @param padding 頂点番号を入力からいくつずらすか 省略したら-1 * @param weighted 辺の重みが入力されるか 省略したらfalseとなり、重み1で辺が追加される * @param directed 有向グラフかどうか 省略したらfalse */ void read(int m, int padding=-1, bool weighted=false, bool directed=false) { for(int i = 0; i < m; i++) { int u, v; std::cin >> u >> v; u += padding, v += padding; Cost c(1); if(weighted) std::cin >> c; if(directed) add_directed_edge(u, v, c); else add_edge(u, v, c); } } /** * @brief ある頂点から出る辺を列挙する * @param v 頂点番号 * @return std::vector& vから出る辺のリスト */ std::vector& operator[](int v) { return g[v]; } /** * @brief ある頂点から出る辺を列挙する * @param v 頂点番号 * @return const std::vector& vから出る辺のリスト */ const std::vector& operator[](int v) const { return g[v]; } /** * @brief 辺のリスト * @return std::vector 辺のリスト(idの昇順) * * 無向辺は代表して1つだけ格納される */ std::vector edges() const { std::vector res(m); for(int i = 0; i < n; i++) { for(auto& e : g[i]) { res[e.id] = e; } } return res; } /** * @brief ある頂点から各頂点への最短路 * * @param s 始点 * @param weighted 1以外のコストの辺が存在するか 省略するとtrue * @param inf コストのminの単位元 未到達の頂点への距離はinfになる 省略すると-1 * @return std::pair, std::vector> first:各頂点への最短路長 second:各頂点への最短路上の直前の辺 */ std::pair, std::vector> shortest_path(int s, bool weignted = true, Cost inf = -1) const { if(weignted) return shortest_path_dijkstra(s, inf); return shortest_path_bfs(s, inf); } std::vector topological_sort() { std::vector indeg(n), sorted; std::queue q; for (int i = 0; i < n; i++) { for (int dst : g[i]) indeg[dst]++; } for (int i = 0; i < n; i++) { if (!indeg[i]) q.push(i); } while (!q.empty()) { int cur = q.front(); q.pop(); for (int dst : g[cur]) { if (!--indeg[dst]) q.push(dst); } sorted.push_back(cur); } return sorted; } private: std::pair, std::vector> shortest_path_bfs(int s, Cost inf) const { std::vector dist(n, inf); std::vector prev(n); std::queue que; dist[s] = 0; que.push(s); while(!que.empty()) { int u = que.front(); que.pop(); for(auto& e : g[u]) { if(dist[e.dst] == inf) { dist[e.dst] = dist[e.src] + 1; prev[e.dst] = e; que.push(e.dst); } } } return {dist, prev}; } std::pair, std::vector> shortest_path_dijkstra(int s, Cost inf) const { std::vector dist(n, inf); std::vector prev(n); using Node = std::pair; std::priority_queue, std::greater> que; dist[s] = 0; que.push({0, s}); while(!que.empty()) { auto [d, u] = que.top(); que.pop(); if(d > dist[u]) continue; for(auto& e : g[u]) { if(dist[e.dst] == inf || dist[e.dst] > dist[e.src] + e.cost) { dist[e.dst] = dist[e.src] + e.cost; prev[e.dst] = e; que.push({dist[e.dst], e.dst}); } } } return {dist, prev}; } }; #line 13 "/home/shogo314/cpp_include/ou-library/tree.hpp" template struct RootedTree; template struct DoublingClimbTree; /** * @brief 根なし木 * * @tparam Cost = int 辺の重み * * Graphを継承し、すべて無向辺で表す */ template struct Tree : private Graph { // Graph* g = new Tree(); ができてしまうと、delete時にメモリリークが発生 // 回避するためprivate継承にして、メンバをすべてusing宣言 using Edge = typename Graph::Edge; using Graph::n; using Graph::m; using Graph::g; using Graph::Graph; using Graph::add_edge; using Graph::add_directed_edge; using Graph::read; using Graph::operator[]; using Graph::edges; using Graph::shortest_path; /** * @brief コンストラクタ * * @param n ノード数 */ Tree(int n = 0) : Graph(n) {} /** * @brief 辺の情報を標準入力から受け取って追加する * @param padding = -1 頂点番号を入力からいくつずらすか * @param weighted = false 辺の重みが入力されるか */ void read(int padding = -1, bool weighted = false) { Graph::read(this->n - 1, padding, weighted, false); } /** * @brief 木の直径 * * @param weighted = true 重み付きか * @return std::vector 直径を構成する辺のリスト */ std::vector diameter(bool weighted = true) const { std::vector dist = shortest_path(0, weighted).first; int u = std::max_element(dist.begin(), dist.end()) - dist.begin(); std::vector prev; std::tie(dist, prev) = shortest_path(u, weighted); int v = std::max_element(dist.begin(), dist.end()) - dist.begin(); std::vector path; while(v != u) { path.push_back(prev[v]); v = prev[v].src; } reverse(path.begin(), path.end()); return path; } /** * @brief 根付き木にする * * @param root 根 * @return RootedTree 根付き木のオブジェクト */ [[nodiscard]] RootedTree set_root(int root) const& { return RootedTree(*this, root); } /** * @brief 根付き木にして返す * * @param root 根 * @return RootedTree 根付き木のオブジェクト */ [[nodiscard]] RootedTree set_root(int root) && { return RootedTree(std::move(*this), root); } /** * @brief ダブリングLCAが使える根付き木を返す * * @param root 根 * @return DoublingClimbTree ダブリング済み根付き木のオブジェクト */ [[nodiscard]] DoublingClimbTree build_lca(int root) const& { RootedTree rooted_tree(*this, root); return DoublingClimbTree(std::move(rooted_tree)); } /** * @brief ダブリングLCAが使える根付き木を返す * * @param root 根 * @return DoublingClimbTree ダブリング済み根付き木のオブジェクト */ [[nodiscard]] DoublingClimbTree build_lca(int root) && { RootedTree rooted_tree(std::move(*this), root); return DoublingClimbTree(std::move(rooted_tree)); } }; /** * @brief 根付き木 * * @tparam Cost = int 辺のコスト */ template struct RootedTree : private Tree { using Edge = typename Tree::Edge; using Tree::n; using Tree::m; using Tree::g; using Tree::operator[]; using Tree::edges; using Tree::shortest_path; using Tree::Tree; int root; //!< 根 std::vector par; //!< 親へ向かう辺 std::vector> child; //!< 子へ向かう辺のリスト std::vector depth; //!< 深さのリスト std::vector height; //!< 部分木の高さのリスト std::vector sz; //!< 部分期の頂点数のリスト std::vector preorder; //!< 先行順巡回 std::vector postorder; //!< 後行順巡回 std::vector> euler_tour; //!< オイラーツアー DFS中にとおった頂点の(頂点番号,その頂点を通った回数)のリスト RootedTree() = delete; /** * @brief 親の頂点のリストから根が0の根付き木を構築するコンストラクタ * * @param par_ 頂点0以外の親の頂点のリスト * @param padding = -1 parの頂点番号をいくつずらすか */ RootedTree(const std::vector& par_, int padding = -1) : Tree(par_.size() + 1), root(0) { for(int i = 0; i < (int)par_.size(); i++) { this->add_edge(i + 1, par_[i] + padding); } build(); } /** * @brief Treeから根付き木を構築するコンストラクタ(コピー) * * @param tree Tree * @param root 根 */ RootedTree(const Tree& tree, int root) : Tree(tree), root(root) { build(); } /** * @brief Treeから根付き木を構築するコンストラクタ(ムーブ) * * @param tree Tree * @param root 根 */ RootedTree(Tree&& tree, int root) : Tree(std::move(tree)), root(root) { build(); } /** * @brief ダブリングLCAが使える根付き木を得る * * @return DoublingClimbTree ダブリング済み根付き木のオブジェクト */ [[nodiscard]] DoublingClimbTree build_lca() const& { return DoublingClimbTree(*this); } /** * @brief ダブリングLCAが使える根付き木を得る * * @return DoublingClimbTree ダブリング済み根付き木のオブジェクト */ [[nodiscard]] DoublingClimbTree build_lca() && { return DoublingClimbTree(std::move(*this)); } private: void build() { int n = this->n; par.resize(n); par[root] = {root, -1, 0, -1}; child.resize(n); depth.resize(n); height.resize(n); sz.resize(n); std::vector iter(n); std::stack> stk; stk.emplace(root, 0); while(!stk.empty()) { auto [u, cnt] = stk.top(); stk.pop(); euler_tour.emplace_back(u, cnt); if(iter[u] == 0) preorder.push_back(u); while(iter[u] < (int)this->g[u].size() && this->g[u][iter[u]].dst == par[u]) iter[u]++; if(iter[u] == (int)this->g[u].size()) { postorder.push_back(u); sz[u] = 1; height[u] = 0; for(auto& e : child[u]) { sz[u] += sz[e.dst]; if(height[u] < height[e.dst] + e.cost) { height[u] = height[e.dst] + e.cost; } } } else { auto& e = this->g[u][iter[u]++]; par[e.dst] = {e.dst, u, e.cost, e.id}; child[u].push_back(e); depth[e.dst] = depth[u] + e.cost; stk.emplace(u, cnt + 1); stk.emplace(e.dst, 0); } } } }; /** * @brief 親頂点ダブリング済み根付き木 * * @tparam Cost = int 辺の重み(注: climbでは根の重みは考えない) */ template struct DoublingClimbTree : private RootedTree { using Edge = typename RootedTree::Edge; using RootedTree::n; using RootedTree::m; using RootedTree::g; using RootedTree::operator[]; using RootedTree::edges; using RootedTree::shortest_path; using RootedTree::RootedTree; using RootedTree::root; using RootedTree::par; using RootedTree::child; using RootedTree::depth; using RootedTree::height; using RootedTree::sz; using RootedTree::preorder; using RootedTree::postorder; using RootedTree::euler_tour; int h; //!< ダブリングの回数 std::vector> doubling_par; //!< jから(1<& tree) : RootedTree(tree) { build(); } /** * @brief RootedTreeからダブリング済み根付き木を構築するコンストラクタ(ムーブ) * * @param tree RootedTree */ DoublingClimbTree(RootedTree&& tree) : RootedTree(std::move(tree)) { build(); } /** * @brief 頂点uからk回を根の方向に遡った頂点 * * @param u 元の頂点 * @param k 登る段数 * @return int k回登った頂点 */ int climb(int u, int k) const { for(int i = 0; i < h; i++) { if(k >> i & 1) u = doubling_par[i][u]; if(u == -1) return -1; } return u; } /** * @brief 最小共通祖先 * * @param u 頂点1 * @param v 頂点2 * @return int LCAの頂点番号 */ int lca(int u, int v) const { if(this->depth[u] > this->depth[v]) std::swap(u, v); v = climb(v, this->depth[v] - this->depth[u]); if(this->depth[u] > this->depth[v]) u = climb(u, this->depth[u] - this->depth[v]); if(u == v) return u; for(int i = h - 1; i >= 0; i--) { int nu = doubling_par[i][u]; int nv = doubling_par[i][v]; if(nu == -1) continue; if(nu != nv) { u = nu; v = nv; } } return this->par[u]; } /** * @brief 頂点間距離(重みなし) * * @param u 頂点1 * @param v 頂点2 * @return int uとv間の最短経路の辺の本数 */ int dist(int u, int v) const { return this->depth[u] + this->depth[v] - this->depth[lca(u, v)] * 2; } private: void build() { int n = this->n; h = 0; while((1 << h) < n) h++; doubling_par.assign(h, std::vector(n, -1)); for(int i = 0; i < n; i++) doubling_par[0][i] = this->par[i]; for(int i = 0; i < h - 1; i++) { for(int j = 0; j < n; j++) { if(doubling_par[i][j] != -1) { doubling_par[i+1][j] = doubling_par[i][doubling_par[i][j]]; } } } } }; #line 1 "/home/shogo314/cpp_include/sh-library/base.hpp" #include #pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") using namespace std; using ll = long long; using ull = unsigned long long; using str = string; using pl = pair; template using ml = map; using mll = map; using sl = set; using ld = long double; using pd = pair; template using vec = vector; template using vv = vector>; template using vvv = vector>>; template using vp = vector>; using vl = vec; using vvl = vv; using vvvl = vvv; using vs = vec; using vc = vec; using vi = vec; using vb = vec; using vpl = vec; using spl = set; using vd = vec; using vpd = vec; template using ary = array; template using al = array; template using aal = array, N1>; template using val = vec>; #define all(obj) (obj).begin(), (obj).end() #define reps(i, a, n) for (ll i = (a); i < ll(n); i++) #define rep(i, n) reps(i, 0, (n)) #define rrep(i, n) reps(i, 1, (n) + 1) #define repds(i, a, n) for (ll i = ((n) - 1); i >= (a); i--) #define repd(i, n) repds(i, 0, (n)) #define rrepd(i, n) repds(i, 1, (n) + 1) #define rep2(i, j, x, y) rep(i, x) rep(j, y) template inline bool chmin(T1 &a, T2 b) { if (a > b) { a = b; return true; } return false; } template inline bool chmax(T1 &a, T2 b) { if (a < b) { a = b; return true; } return false; } #define LL(x) ll x;cin >> x; #define VL(a,n) vl a(n);cin >> a; #define AL(a,k) al a;cin >> a; #define VC(a,n) vc a(n);cin >> a; #define VS(a,n) vs a(n);cin >> a; #define STR(s) str s;cin >> s; #define VPL(a,n) vpl a(n);cin >> a; #define VAL(a,n,k) val a(n);cin >> a; #define VVL(a,n,m) vvl a(n,vl(m));cin >> a; #define SL(a,n) sl a;{VL(b,n);a=sl(all(b));} template using heap_max = priority_queue, less>; template using heap_min = priority_queue, greater>; #line 4 "main.cpp" void solve() { LL(N); Tree tree(N); tree.read(); ll ans = 0; ll tmp = 0; rep(i, N) { ll s = tree[i].size(); ans += s * (s - 1) / 2; tmp += s; } ans += tmp / 2; for (auto e : tree.edges()) { ans += (tree[e.src].size() - 1) * (tree[e.dst].size() - 1); } print(ans); } int main() { cin.tie(nullptr); ios_base::sync_with_stdio(false); solve(); }