結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー ぷらぷら
提出日時 2021-03-05 22:39:56
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 133 ms / 2,000 ms
コード長 2,836 bytes
コンパイル時間 1,777 ms
コンパイル使用メモリ 165,644 KB
実行使用メモリ 19,072 KB
最終ジャッジ日時 2024-04-16 10:26:04
合計ジャッジ時間 4,651 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
6,016 KB
testcase_01 AC 3 ms
5,888 KB
testcase_02 AC 4 ms
6,016 KB
testcase_03 AC 132 ms
12,852 KB
testcase_04 AC 128 ms
12,892 KB
testcase_05 AC 130 ms
13,056 KB
testcase_06 AC 133 ms
13,056 KB
testcase_07 AC 127 ms
12,928 KB
testcase_08 AC 74 ms
10,880 KB
testcase_09 AC 34 ms
8,192 KB
testcase_10 AC 36 ms
8,448 KB
testcase_11 AC 20 ms
7,168 KB
testcase_12 AC 57 ms
9,728 KB
testcase_13 AC 68 ms
10,496 KB
testcase_14 AC 72 ms
10,624 KB
testcase_15 AC 54 ms
9,600 KB
testcase_16 AC 9 ms
6,272 KB
testcase_17 AC 8 ms
6,400 KB
testcase_18 AC 117 ms
12,416 KB
testcase_19 AC 18 ms
7,168 KB
testcase_20 AC 6 ms
6,144 KB
testcase_21 AC 53 ms
9,216 KB
testcase_22 AC 46 ms
9,088 KB
testcase_23 AC 6 ms
6,272 KB
testcase_24 AC 7 ms
6,272 KB
testcase_25 AC 6 ms
6,016 KB
testcase_26 AC 6 ms
6,016 KB
testcase_27 AC 4 ms
6,016 KB
testcase_28 AC 5 ms
6,016 KB
testcase_29 AC 8 ms
6,144 KB
testcase_30 AC 7 ms
6,144 KB
testcase_31 AC 5 ms
6,016 KB
testcase_32 AC 6 ms
6,144 KB
testcase_33 AC 24 ms
8,960 KB
testcase_34 AC 121 ms
19,072 KB
testcase_35 AC 44 ms
11,520 KB
testcase_36 AC 11 ms
6,912 KB
testcase_37 AC 71 ms
12,752 KB
testcase_38 AC 65 ms
12,156 KB
testcase_39 AC 4 ms
6,016 KB
testcase_40 AC 4 ms
6,016 KB
testcase_41 AC 4 ms
6,016 KB
testcase_42 AC 4 ms
6,016 KB
testcase_43 AC 4 ms
6,144 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
using pcc = pair<char,char>;
using pii = pair<int,int>;
using pll = pair<ll,ll>;
using pdd = pair<ld,ld>;
using tuplis = array<ll,3>;
template<class T> using prique = priority_queue<T,vector<T>,greater<T>>;
#define rep(i,n) for(ll i = 0; i < n; i++)
#define rrep(i,n) for(ll i = n-1; i >= 0; i--)
#define all(a) a.begin(),a.end()
#define rall(a) a.rbegin(),a.rend()
#define Sort(a) sort(all(a))
#define Rev(a) reverse(all(a))
#define Uniq(a) sort(all(a));a.erase(unique(all(a)),a.end())
const ll LINF = 1001001001001001001;
const int INF = 1001001001;
const int MOD = 1000000007;
const int MODD = 998244353;
const ld PI = 3.1415926535897932;
const ll dx[] = {0, 1, 0, -1, 1, -1, 1, -1};
const ll dy[] = {1, 0, -1, 0, 1, 1, -1, -1};
inline ll popcnt(ll a){ return __builtin_popcountll(a); }
inline ll intpow(ll a, ll b){ ll ans = 1; while(b){ if(b & 1) ans *= a; a *= a; b /= 2; } return ans; }
inline ll modpow(ll a, ll b, ll p){ ll ans = 1; while(b){ if(b & 1) (ans *= a) %= p; (a *= a) %= p; b /= 2; } return ans; }
ll gcd(ll a,ll b){ if(a%b == 0){return b; } else{ return gcd(b,a%b); } }
ll lcm(ll a,ll b){ return a/gcd(a,b)*b; }
template<class T> bool chmin(T& a, const T& b){ if(a > b){ a = b; return 1; } return 0; }
template<class T> bool chmax(T& a, const T& b){ if(a < b){ a = b; return 1; } return 0; }
template<class T, class U> bool chmin(T& a, const U& b){ if(a > T(b)){ a = b; return 1; } return 0; }
template<class T, class U> bool chmax(T& a, const U& b){ if(a < T(b)){ a = b; return 1; } return 0; }
ll N,cnt[100005],used[100005],cnt2[100005],used2[100005],cnt3[100005];
vector<int>ki[100005];
ll dfs(int now) {
    ll sum = 0;
    used[now] = true;
    rep(i,ki[now].size()) {
        if(used[ki[now][i]]) {
            continue;
        }
        sum += dfs(ki[now][i]);
    }
    return cnt[now] = sum+1;
}
ll dfs2(int now) {
    ll sum = 0;
    used2[now] = true;
    rep(i,ki[now].size()) {
        if(used2[ki[now][i]]) {
            continue;
        }
        sum += dfs2(ki[now][i]);
    }
    return cnt2[now] = sum+cnt[now];
}
ll ans = 0;
void dfs3(int now) {
    ll sum = 0;
    rep(i,ki[now].size()) {
        if(cnt[ki[now][i]] > cnt[now]) {
            sum += cnt3[ki[now][i]]-cnt2[now]-cnt[now];
        }
        else {
            sum += cnt2[ki[now][i]];
        }
    }
    cnt3[now] = sum+N;
    ans += cnt3[now];
    rep(i,ki[now].size()) {
        if(cnt[ki[now][i]] > cnt[now]) {
            continue;
        }
        else {
            dfs3(ki[now][i]);
        }
    }
}
int main() {
    cin >> N;
    rep(i,N-1) {
        int A,B;
        cin >> A >> B;
        A--;B--;
        ki[A].push_back(B);
        ki[B].push_back(A);
    }
    dfs(0);
    dfs2(0);
    dfs3(0);
    cout << ans << endl;
}
0