結果
問題 |
No.1124 Earthquake Safety
|
ユーザー |
![]() |
提出日時 | 2020-07-22 22:48:52 |
言語 | C++17(clang) (17.0.6 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,757 bytes |
コンパイル時間 | 4,339 ms |
コンパイル使用メモリ | 163,824 KB |
実行使用メモリ | 20,352 KB |
最終ジャッジ日時 | 2024-06-22 23:46:40 |
合計ジャッジ時間 | 10,436 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 6 TLE * 1 -- * 51 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define int long long #define ii pair <int, int> #define app push_back #define all(a) a.begin(), a.end() #define bp __builtin_popcountll #define ll long long #define mp make_pair #define f first #define s second #define Time (double)clock()/CLOCKS_PER_SEC #define debug(x) std::cout << #x << ": " << x << '\n'; const int N = 2007; int dist[N][N]; vector <int> g[N]; void dfs(int u, int p, int *dist) { for (int v : g[u]) { if (v != p) { dist[v] = dist[u] + 1; dfs(v, u, dist); } } } const int MOD = 1000 * 1000 * 1000 + 7; int pw[N]; int n; int get(int u, int v, int k) { vector <int> a = {u, v, k}; for (int i = 0; i < 3; ++i) { int u = a[i]; int v = a[(i + 1) % 3]; int k = a[(i + 2) % 3]; if (dist[u][v] + dist[u][k] == dist[v][k]) { return pw[n - 1 - dist[v][k]]; } } int sum = (dist[u][v] + dist[v][k] + dist[u][k]) / 2; return pw[n - 1 - sum]; } signed main() { #ifdef HOME freopen("input.txt", "r", stdin); #else #define endl '\n' ios_base::sync_with_stdio(0); cin.tie(0); #endif pw[0] = 1; for (int i = 1; i < N; ++i) pw[i] = pw[i - 1] * 2 % MOD; cin >> n; for (int i = 0; i < n - 1; ++i) { int u, v; cin >> u >> v; g[u].app(v); g[v].app(u); } for (int i = 1; i <= n; ++i) { dfs(i, i, dist[i]); } int ans = 0; for (int a = 1; a <= n; ++a) { for (int b = 1; b <= n; ++b) { for (int c = 1; c <= n; ++c) { ans += get(a, b, c); ans %= MOD; } } } cout << ans << endl; }