結果
問題 | No.1103 Directed Length Sum |
ユーザー | jupiro |
提出日時 | 2020-05-06 02:44:02 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 946 ms / 3,000 ms |
コード長 | 2,672 bytes |
コンパイル時間 | 1,653 ms |
コンパイル使用メモリ | 138,328 KB |
実行使用メモリ | 167,656 KB |
最終ジャッジ日時 | 2024-06-27 14:10:31 |
合計ジャッジ時間 | 11,017 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 429 ms
167,656 KB |
testcase_03 | AC | 278 ms
77,732 KB |
testcase_04 | AC | 530 ms
43,636 KB |
testcase_05 | AC | 946 ms
73,212 KB |
testcase_06 | AC | 349 ms
29,672 KB |
testcase_07 | AC | 57 ms
9,728 KB |
testcase_08 | AC | 94 ms
13,032 KB |
testcase_09 | AC | 30 ms
7,424 KB |
testcase_10 | AC | 138 ms
16,640 KB |
testcase_11 | AC | 561 ms
47,232 KB |
testcase_12 | AC | 313 ms
29,696 KB |
testcase_13 | AC | 137 ms
16,756 KB |
testcase_14 | AC | 24 ms
6,400 KB |
testcase_15 | AC | 235 ms
23,936 KB |
testcase_16 | AC | 649 ms
52,780 KB |
testcase_17 | AC | 675 ms
54,784 KB |
testcase_18 | AC | 135 ms
16,512 KB |
testcase_19 | AC | 580 ms
48,384 KB |
testcase_20 | AC | 38 ms
8,192 KB |
testcase_21 | AC | 84 ms
12,032 KB |
testcase_22 | AC | 464 ms
40,192 KB |
testcase_23 | AC | 253 ms
25,416 KB |
ソースコード
#include <cstdio> #include <iostream> #include <string> #include <sstream> #include <stack> #include <algorithm> #include <cmath> #include <queue> #include <map> #include <set> #include <cstdlib> #include <bitset> #include <tuple> #include <assert.h> #include <deque> #include <bitset> #include <iomanip> #include <limits> #include <chrono> #include <random> #include <array> #include <unordered_map> #include <functional> #include <complex> #include <numeric> template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; } template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; } constexpr long long MAX = 5100000; constexpr long long INF = 1LL << 60; constexpr int inf = 1 << 28; constexpr long long mod = 1000000007LL; //constexpr long long mod = 998244353LL; using namespace std; typedef unsigned long long ull; typedef long long ll; struct mint { long long x; mint(long long x = 0) :x((x% mod + mod) % mod) {} mint& operator+=(const mint a) { if ((x += a.x) >= mod) x -= mod; return *this; } mint& operator-=(const mint a) { if ((x += mod - a.x) >= mod) x -= mod; return *this; } mint& operator*=(const mint a) { (x *= a.x) %= mod; return *this; } mint operator+(const mint a) const { mint res(*this); return res += a; } mint operator-(const mint a) const { mint res(*this); return res -= a; } mint operator*(const mint a) const { mint res(*this); return res *= a; } mint pow(ll t) const { if (!t) return 1; mint a = pow(t >> 1); a *= a; if (t & 1) a *= *this; return a; } // for prime mod mint inv() const { return pow(mod - 2); } mint& operator/=(const mint a) { return (*this) *= a.inv(); } mint operator/(const mint a) const { mint res(*this); return res /= a; } }; int n; vector<vector<int>> g; vector<mint> dp; vector<mint> sz; void dfs(int cur, int pre) { sz[cur] = 1; for (auto next : g[cur]) { if (next == pre) continue; dfs(next, cur); sz[cur] += sz[next]; dp[cur] += dp[next]; } dp[cur] += sz[cur] - 1; } int main() { /* cin.tie(nullptr); ios::sync_with_stdio(false); */ scanf("%d", &n); if (n > 1000000) assert(false); g.resize(n); dp.resize(n), sz.resize(n); vector<bool> root_can(n, true); for (int i = 0; i < n - 1; i++) { int u, v; scanf("%d %d", &u, &v); if (!(u >= 1 and u <= n and v >= 1 and v <= n)) assert(false); u--; v--; g[u].emplace_back(v); g[v].emplace_back(u); root_can[v] = false; } int root = -1; for (int i = 0; i < n; i++) if (root_can[i]) root = i; dfs(root, -1); mint res = 0; for (int i = 0; i < n; i++) res += dp[i]; cout << res.x << "\n"; return 0; }