提出 #4353019


ソースコード 拡げる

Copy
#define _USE_MATH_DEFINES
#include <cstdio>
#include <iostream>
#include <sstream>
#include <fstream>
#include <iomanip>
#include <algorithm>
#include <cmath>
#include <complex>
#include <string>
#include <vector>
#include <array>
#include <list>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <bitset>
#include <numeric>
#include <limits>
#include <climits>
#include <cfloat>
#include <functional>
#include <iterator>
using namespace std;

int solve1(const vector<vector<int> >& edges, int curr, int prev, vector<int>& ans)
{
    for(int next : edges[curr]){
        if(next == prev)
            continue;
        int len = solve1(edges, next, curr, ans);
        ans[curr] = max(ans[curr], len + 1);
    }
    return ans[curr];
}

void solve2(const vector<vector<int> >& edges, int curr, int prev, int len, vector<int>& ans)
{
    ans[curr] = max(ans[curr], len + 1);

    multiset<int> ms;
    for(int next : edges[curr]){
        if(next == prev)
            continue;
        ms.insert(ans[next]);
    }

    for(int next : edges[curr]){
        if(next == prev)
            continue;
        int tmp = ans[next];
        ms.erase(ms.find(tmp));
        int len2 = len + 1;
        if(!ms.empty())
            len2 = max(len2, *ms.rbegin() + 1);
        solve2(edges, next, curr, len2, ans);
        ms.insert(tmp);
    }
}

int main()
{
    vector<int> n(2);
    vector<vector<vector<int> > > edges(2);
    vector<vector<int> > len(2);
    for(int i=0; i<2; ++i){
        cin >> n[i];
        edges[i].assign(n[i], vector<int>());
        for(int j=0; j<n[i]-1; ++j){
            int p, q;
            cin >> p >> q;
            -- p;
            -- q;
            edges[i][p].push_back(q);
            edges[i][q].push_back(p);
        }
        len[i].assign(n[i], 0);
        solve1(edges[i], 0, -1, len[i]);
        solve2(edges[i], 0, -1, -1, len[i]);
        sort(len[i].begin(), len[i].end());
    }

    int maxLen = max(*max_element(len[0].begin(), len[0].end()),
                     *max_element(len[1].begin(), len[1].end()));
    long long ans = 0;
    long long sum = accumulate(len[1].begin(), len[1].end(), 0LL);
    int j = 0;
    for(int i=n[0]-1; i>=0; --i){
        while(j < n[1] && len[0][i] + len[1][j] < maxLen){
            sum -= len[1][j];
            ++ j;
        }
        ans += sum + (len[0][i] + 1) * (long long)(n[1] - j) + maxLen * (long long)j;
    }
    cout << ans << endl;

    return 0;
}

提出情報

提出日時
問題 B - Bonsai Grafting
ユーザ mamekin
言語 C++14 (GCC 5.4.1)
得点 700
コード長 2560 Byte
結果
実行時間 207 ms
メモリ 23808 KB

ジャッジ結果

セット名 得点 / 配点 テストケース
Sample 0 / 0 01.txt, 02.txt
All 700 / 700 01.txt, 02.txt, 11.txt, 12.txt, 13.txt, 14.txt, 15.txt, 16.txt, 17.txt, 18.txt, 19.txt, 20.txt, 21.txt, 22.txt, 23.txt, 24.txt, 25.txt, 26.txt, 27.txt, 28.txt, 29.txt, 30.txt, 31.txt, 32.txt, 33.txt, 34.txt, 35.txt, 36.txt, 37.txt, 38.txt, 39.txt, 40.txt, 41.txt, 42.txt, 43.txt, 44.txt, 45.txt, 46.txt, 47.txt, 48.txt, 49.txt, 50.txt
ケース名 結果 実行時間 メモリ
01.txt 1 ms 256 KB
02.txt 1 ms 256 KB
11.txt 1 ms 256 KB
12.txt 1 ms 256 KB
13.txt 1 ms 256 KB
14.txt 1 ms 256 KB
15.txt 1 ms 256 KB
16.txt 1 ms 256 KB
17.txt 1 ms 256 KB
18.txt 1 ms 256 KB
19.txt 1 ms 256 KB
20.txt 1 ms 256 KB
21.txt 191 ms 12160 KB
22.txt 197 ms 13952 KB
23.txt 191 ms 12160 KB
24.txt 196 ms 21376 KB
25.txt 195 ms 14464 KB
26.txt 191 ms 12416 KB
27.txt 202 ms 21376 KB
28.txt 195 ms 12032 KB
29.txt 204 ms 22640 KB
30.txt 207 ms 23424 KB
31.txt 194 ms 12160 KB
32.txt 194 ms 14080 KB
33.txt 193 ms 12160 KB
34.txt 203 ms 23808 KB
35.txt 195 ms 14080 KB
36.txt 194 ms 12416 KB
37.txt 200 ms 18688 KB
38.txt 193 ms 12032 KB
39.txt 202 ms 20224 KB
40.txt 203 ms 20864 KB
41.txt 94 ms 6272 KB
42.txt 96 ms 6272 KB
43.txt 98 ms 6144 KB
44.txt 97 ms 6272 KB
45.txt 99 ms 7808 KB
46.txt 98 ms 7552 KB
47.txt 105 ms 17408 KB
48.txt 97 ms 6144 KB
49.txt 102 ms 12544 KB
50.txt 99 ms 12544 KB