제출 #59290

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
59290		gabrielsimoes	철인 이종 경기 (APIO18_duathlon)	C++17	0 / 100	170 ms	23828 KiB

count_triplets

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const int MAXN = 1e5+10;
const int MAXM = 2e5+10;

int n, m;
vector<int> g[MAXN];
ll sub[MAXN];

void dfs0(int cur, int pre) {
    sub[cur] = 1;
    for (int nx : g[cur]) {
        if (nx != pre) {
            dfs0(nx, cur);
            sub[cur] += sub[nx];
        }
    }
}

ll ans = 0;
void dfs1(int cur, int pre, ll sum = 0) {
    ans += 2 * sum * (sub[cur]-1);

    // printf("dfs1 %d %lld %lld: ans %lld\n", cur, sum, sub[cur], ans);

    for (int nx : g[cur]) {
        if (nx != pre) {
            ans += (sub[cur]-1-sub[nx]) * sub[nx];
            dfs1(nx, cur, sum + sub[cur] - sub[nx]);
            // printf("dfs1 %d: ans %lld\n", cur, ans);
        }
    }
}

int main() {
    scanf("%d%d", &n, &m);

    assert(m == n-1);

    for (int i = 0,a,b; i < m; i++) {
        scanf("%d%d", &a, &b);
        g[a].push_back(b);
        g[b].push_back(a);
    }

    dfs0(1, 1);
    dfs1(1, 1);

    printf("%lld\n", ans);
}

컴파일 시 표준 에러 (stderr) 메시지

count_triplets.cpp: In function 'int main()':
count_triplets.cpp:38:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &n, &m);
     ~~~~~^~~~~~~~~~~~~~~~
count_triplets.cpp:43:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d%d", &a, &b);
         ~~~~~^~~~~~~~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 11
• Subtask 3: 0 / 8
• Subtask 4: 0 / 10
• Subtask 5: 0 / 13
• Subtask 6: 0 / 15
• Subtask 7: 0 / 20
• Subtask 8: 0 / 8
• Subtask 9: 0 / 10