結果
問題 | No.1124 Earthquake Safety |
ユーザー |
![]() |
提出日時 | 2020-07-23 01:24:26 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 495 ms / 3,000 ms |
コード長 | 1,242 bytes |
コンパイル時間 | 2,160 ms |
コンパイル使用メモリ | 202,136 KB |
最終ジャッジ日時 | 2025-01-12 04:04:58 |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 58 |
ソースコード
#include <bits/stdc++.h> #define _GLIBCXX_DEBUG using namespace std; using ll = long long; using V = vector<int>; using VV = vector<V>; const int MOD = int(1e9)+7; struct mint { ll x; mint(ll x=0) : x(x%MOD) {} mint& operator+=(const mint a) { if ((x += a.x) >= MOD) x -= MOD; return *this; } mint& operator*=(const mint a) { (x *= a.x) %= MOD; return *this; } mint operator*(const mint a) const { return mint(*this) *= a; } mint pow(ll t) const { if (!t) return 1; mint a = pow(t>>1); a *= a; if (t&1) a *= *this; return a; } }; using Vm = vector<mint>; VV G; Vm dfs(int k, int u, int p=-1) { Vm res = {1, 1, 0, 0}; for (int v : G[u]) if (v != p) { auto q = dfs(k, v, u); q[0] *= 2; q[k] *= 2; for (int i=3; i >= 0; --i) { res[i] *= q[0]; for (int j = 1; j <= i; ++j) res[i] += res[i-j] * q[j]; } } return res; } int main() { int n; cin >> n; G.assign(n, V()); for (int i = 0; i < n-1; ++i) { int u, v; cin >> u >> v; --u, --v; G[u].push_back(v); G[v].push_back(u); } auto dp2 = dfs(2, 0); auto dp3 = dfs(3, 0); mint ans = mint(2).pow(n-1) * n; ans += dp2[2] * 6; ans += dp3[3] * 6; cout << ans.x << '\n'; return 0; }