結果
問題 | No.1103 Directed Length Sum |
ユーザー | peroon |
提出日時 | 2020-07-07 20:57:10 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 768 ms / 3,000 ms |
コード長 | 5,723 bytes |
コンパイル時間 | 2,203 ms |
コンパイル使用メモリ | 187,988 KB |
実行使用メモリ | 175,104 KB |
最終ジャッジ日時 | 2024-10-01 19:32:32 |
合計ジャッジ時間 | 9,313 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 372 ms
175,104 KB |
testcase_03 | AC | 190 ms
73,616 KB |
testcase_04 | AC | 380 ms
48,896 KB |
testcase_05 | AC | 768 ms
82,304 KB |
testcase_06 | AC | 180 ms
32,896 KB |
testcase_07 | AC | 31 ms
10,368 KB |
testcase_08 | AC | 47 ms
14,208 KB |
testcase_09 | AC | 18 ms
7,680 KB |
testcase_10 | AC | 70 ms
18,176 KB |
testcase_11 | AC | 440 ms
53,120 KB |
testcase_12 | AC | 179 ms
33,228 KB |
testcase_13 | AC | 76 ms
18,612 KB |
testcase_14 | AC | 14 ms
6,784 KB |
testcase_15 | AC | 133 ms
26,624 KB |
testcase_16 | AC | 491 ms
59,212 KB |
testcase_17 | AC | 530 ms
61,696 KB |
testcase_18 | AC | 70 ms
18,284 KB |
testcase_19 | AC | 420 ms
54,144 KB |
testcase_20 | AC | 23 ms
8,704 KB |
testcase_21 | AC | 45 ms
13,184 KB |
testcase_22 | AC | 311 ms
44,928 KB |
testcase_23 | AC | 145 ms
28,204 KB |
コンパイルメッセージ
main.cpp: In function 'int main()': main.cpp:241:8: warning: 'root' may be used uninitialized [-Wmaybe-uninitialized] 241 | ll root; | ^~~~
ソースコード
#ifdef LOCAL #define _GLIBCXX_DEBUG #define __clock__ #else #pragma GCC optimize("Ofast") #endif #include<bits/stdc++.h> using namespace std; using ll = long long; using ld = long double; using VI = vector<ll>; using VV = vector<VI>; using VS = vector<string>; using PII = pair<ll, ll>; // tourist set template <typename A, typename B> string to_string(pair<A, B> p); template <typename A, typename B, typename C> string to_string(tuple<A, B, C> p); template <typename A, typename B, typename C, typename D> string to_string(tuple<A, B, C, D> p); string to_string(const string& s) { return '"' + s + '"'; } string to_string(const char* s) { return to_string((string) s); } string to_string(bool b) { return (b ? "true" : "false"); } string to_string(vector<bool> v) { bool first = true; string res = "{"; for (int i = 0; i < static_cast<int>(v.size()); i++) { if (!first) { res += ", "; } first = false; res += to_string(v[i]); } res += "}"; return res; } template <size_t N> string to_string(bitset<N> v) { string res = ""; for (size_t i = 0; i < N; i++) { res += static_cast<char>('0' + v[i]); } return res; } template <typename A> string to_string(A v) { bool first = true; string res = "{"; for (const auto &x : v) { if (!first) { res += ", "; } first = false; res += to_string(x); } res += "}"; return res; } template <typename A, typename B> string to_string(pair<A, B> p) { return "(" + to_string(p.first) + ", " + to_string(p.second) + ")"; } template <typename A, typename B, typename C> string to_string(tuple<A, B, C> p) { return "(" + to_string(get<0>(p)) + ", " + to_string(get<1>(p)) + ", " + to_string(get<2>(p)) + ")"; } template <typename A, typename B, typename C, typename D> string to_string(tuple<A, B, C, D> p) { return "(" + to_string(get<0>(p)) + ", " + to_string(get<1>(p)) + ", " + to_string(get<2>(p)) + ", " + to_string(get<3>(p)) + ")"; } void debug_out() { cerr << '\n'; } template <typename Head, typename... Tail> void debug_out(Head H, Tail... T) { cerr << " " << to_string(H); debug_out(T...); } #ifdef LOCAL #define debug(...) cerr << "[" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__) #else #define debug(...) 42 #endif // tourist set end template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; } template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; } #define FOR(i,a,b) for(ll i=(a);i<(b);++i) #define rep(i,b) FOR(i, 0, b) #define ALL(v) (v).begin(), (v).end() #define p(s) cout<<(s)<<'\n' #define p2(s, t) cout << (s) << " " << (t) << '\n' #define br() p("") #define pn(s) cout << (#s) << " " << (s) << '\n' #define SZ(x) ((int)(x).size()) #define SORT(A) sort(ALL(A)) #define RSORT(A) sort(ALL(A), greater<ll>()) #define MP make_pair #define p_yes() p("Yes") #define p_no() p("No") ll SUM(VI& V){ return accumulate(ALL(V), 0LL); } ll MIN(VI& V){return *min_element(ALL(V));} ll MAX(VI& V){return *max_element(ALL(V));} void print_vector(VI& V){ ll n = V.size(); rep(i, n){ if(i) cout << ' '; cout << V[i]; } cout << endl; } ll gcd(ll a,ll b){ if(b == 0) return a; return gcd(b,a%b); } ll lcm(ll a,ll b){ ll g = gcd(a,b); return a / g * b; } // long double using ld = long double; #define EPS (1e-14) #define equals(a,b) (fabs((a)-(b)) < EPS) void no(){p_no(); exit(0);} void yes(){p_yes(); exit(0);} const ll mod = 1e9 + 7; const ll inf = 1e18; const double PI = acos(-1); // snuke's mint // auto mod int // https://youtu.be/L8grWxBlIZ4?t=9858 // https://youtu.be/ERZuLAxZffQ?t=4807 : optimize // https://youtu.be/8uowVvQ_-Mo?t=1329 : division // const int mod = 1000000007; struct mint { ll x; // typedef long long ll; mint(ll x=0):x((x%mod+mod)%mod){} mint operator-() const { return mint(-x);} 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 main(){ cin.tie(0); ios::sync_with_stdio(false); // input ll N; cin>>N; VI A(N-1); VI B(N-1); rep(i,N-1){ cin>>A[i]>>B[i]; A[i]--; B[i]--; } VI in(N); // 入次数 VV G(N); // graph rep(i,N-1){ ll a = A[i]; ll b = B[i]; G[a].push_back(b); in[b]++; } ll root; rep(i,N){ if(in[i]==0) root=i; } debug(root); VI d(N, -1); function<void(ll,ll)> dfs = [&](ll r, ll dist){ d[r]=dist; for(ll to : G[r]){ dfs(to, dist+1); } }; dfs(root, 0); VI C(N, -1); // child + self function<ll(ll)> dfs2 = [&](ll r){ if(C[r]!=-1) return C[r]; ll sum=0; for(ll to : G[r]){ sum += dfs2(to); } return C[r] = sum+1; }; dfs2(root); mint sum=0; rep(i,N-1){ ll a = A[i]; ll b = B[i]; sum += (d[a]+1)*C[b]; } p(sum.x); return 0; }