結果
問題 | 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_DEBUGusing 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;}