結果

問題 No.901 K-ary εxtrεεmε
ユーザー catuppercatupper
提出日時 2019-10-05 02:21:24
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 334 ms / 3,000 ms
コード長 2,414 bytes
コンパイル時間 903 ms
コンパイル使用メモリ 92,080 KB
実行使用メモリ 35,040 KB
最終ジャッジ日時 2024-10-04 07:20:53
合計ジャッジ時間 8,519 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 152 ms
35,040 KB
testcase_01 AC 3 ms
10,376 KB
testcase_02 AC 4 ms
12,076 KB
testcase_03 AC 5 ms
8,808 KB
testcase_04 AC 5 ms
11,712 KB
testcase_05 AC 5 ms
10,416 KB
testcase_06 AC 5 ms
10,384 KB
testcase_07 AC 211 ms
30,576 KB
testcase_08 AC 205 ms
31,696 KB
testcase_09 AC 219 ms
30,576 KB
testcase_10 AC 209 ms
31,688 KB
testcase_11 AC 211 ms
31,632 KB
testcase_12 AC 190 ms
31,408 KB
testcase_13 AC 206 ms
31,628 KB
testcase_14 AC 203 ms
30,932 KB
testcase_15 AC 199 ms
31,752 KB
testcase_16 AC 214 ms
31,536 KB
testcase_17 AC 281 ms
30,664 KB
testcase_18 AC 278 ms
31,584 KB
testcase_19 AC 278 ms
30,960 KB
testcase_20 AC 286 ms
30,820 KB
testcase_21 AC 291 ms
30,720 KB
testcase_22 AC 210 ms
30,980 KB
testcase_23 AC 203 ms
30,996 KB
testcase_24 AC 209 ms
30,728 KB
testcase_25 AC 224 ms
31,000 KB
testcase_26 AC 204 ms
31,596 KB
testcase_27 AC 334 ms
30,724 KB
testcase_28 AC 329 ms
30,720 KB
testcase_29 AC 323 ms
31,884 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <vector>
#include <map>
#include <set>
#include <string>
#include <queue>
#include <stack>
using namespace std;
#define MOD 1000000007
#define INF (1 << 29)
#define LINF (1LL << 60)
#define EPS (1e-10)
typedef long long Int;
typedef pair<Int, Int> P;

Int ppp;
Int come[110000];
Int p[110000][20];
Int depth[110000];
Int node[110000];
Int weight[110000];
vector<P> edge[110000];

void dfs(Int x, Int last = -1, Int d = 0)
{
    come[x] = ppp;
    node[ppp] = x;
    depth[x] = d;
    ppp++;

    if (last != -1)
    {
        for (Int i = 0; i < 20; i++)
        {
            if (i == 0)
                p[x][i] = last;
            else
                p[x][i] = p[p[x][i - 1]][i - 1];
        }
    }
    for (auto p : edge[x])
    {
        Int to = p.first, w = p.second;
        if (to == last)
            continue;
        weight[to] = weight[x] + w;
        dfs(to, x, d + 1);
    }
}

Int la(Int x, Int d, Int pp = 19)
{
    if (d == 0)
        return x;
    while (d < (1 << pp))
        pp--;
    x = p[x][pp];
    return la(x, d - (1 << pp), pp);
}

Int lca(Int x, Int y)
{
    if (depth[x] < depth[y])
        swap(x, y);
    if (depth[x] > depth[y])
        x = la(x, depth[x] - depth[y]);
    if (x == y)
        return x;
    if (depth[x] != depth[y])
        exit(1);
    Int up = 19;
    while (p[x][0] != p[y][0])
    {
        while (p[x][up] == p[y][up])
            up--;
        x = p[x][up];
        y = p[y][up];
    }
    return p[x][0];
}

void solve()
{
    Int k, ans = 0;
    cin >> k;
    vector<Int> nodes;
    for (Int i = 0; i < k; i++)
    {
        Int x;
        cin >> x;
        nodes.push_back(come[x]);
    }
    sort(nodes.begin(), nodes.end());
    for (Int i = 0; i < k; i++)
    {
        Int x = node[nodes[i]];
        Int y = node[nodes[(i + 1) % k]];
        Int l = lca(x, y);
        //cout << x << ":" << weight[x] << " "  << y << ":" << weight[y] << " "  << l << ":" << weight[l] << endl;
        ans += weight[x] - weight[l];
        ans += weight[y] - weight[l];
    }
    cout << ans / 2 << endl;
}

int main()
{
    int n, u, v, w, q;
    cin >> n;
    for (Int i = 0; i < n - 1; i++)
    {
        cin >> u >> v >> w;
        edge[u].push_back(P(v, w));
        edge[v].push_back(P(u, w));
    }
    dfs(0);
    cin >> q;
    while (q--)
        solve();
    return 0;
}
0