結果

問題 No.1442 I-wate Shortest Path Problem
ユーザー chocoruskchocorusk
提出日時 2021-03-26 23:11:14
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 753 ms / 3,000 ms
コード長 3,516 bytes
コンパイル時間 3,796 ms
コンパイル使用メモリ 202,620 KB
実行使用メモリ 49,220 KB
最終ジャッジ日時 2024-05-06 17:13:13
合計ジャッジ時間 16,106 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
8,192 KB
testcase_01 AC 4 ms
8,192 KB
testcase_02 AC 15 ms
9,344 KB
testcase_03 AC 185 ms
8,960 KB
testcase_04 AC 17 ms
9,088 KB
testcase_05 AC 13 ms
8,960 KB
testcase_06 AC 184 ms
8,832 KB
testcase_07 AC 13 ms
8,704 KB
testcase_08 AC 180 ms
8,448 KB
testcase_09 AC 23 ms
10,112 KB
testcase_10 AC 195 ms
9,216 KB
testcase_11 AC 189 ms
9,344 KB
testcase_12 AC 601 ms
44,404 KB
testcase_13 AC 402 ms
37,828 KB
testcase_14 AC 518 ms
41,384 KB
testcase_15 AC 486 ms
40,252 KB
testcase_16 AC 543 ms
42,648 KB
testcase_17 AC 730 ms
47,280 KB
testcase_18 AC 738 ms
47,528 KB
testcase_19 AC 615 ms
44,916 KB
testcase_20 AC 715 ms
47,296 KB
testcase_21 AC 753 ms
47,336 KB
testcase_22 AC 434 ms
41,336 KB
testcase_23 AC 663 ms
49,220 KB
testcase_24 AC 402 ms
38,752 KB
testcase_25 AC 586 ms
43,968 KB
testcase_26 AC 407 ms
40,796 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<ll, int> P;
struct LCA{
    vector<vector<int>> g;
    vector<int> d;
    vector<vector<int>> p;
    int log;
    int n;
    LCA(const vector<vector<int>> &g):n(g.size()), g(g), d(g.size()){
        log=0;
        while(1<<log<=n) log++;
        p.resize(log, vector<int>(n));
    }
    void dfs(int x, int prev){
        for(auto y:g[x]){
            if(y==prev) continue;
            d[y]=d[x]+1;
            p[0][y]=x;
            dfs(y, x);
        }
    }
    void build(){
        dfs(0, -1);
        for(int i=1; i<log; i++){
            for(int j=0; j<n; j++){
                p[i][j]=p[i-1][p[i-1][j]];
            }
        }
    }
    int lca(int a, int b){
        if(d[a]>d[b]) swap(a, b);
        int dd=d[b]-d[a], i=0;
        int a1=a, b1=b;
        while(dd){
            if(dd&1) b1=p[i][b1];
            dd>>=1;
            i++;
        }
        if(a1==b1) return a1;
        for(int j=log-1; j>=0; j--){
            if(p[j][a1]!=p[j][b1]){
                a1=p[j][a1], b1=p[j][b1];
            }
        }
        return p[0][a1];
    }
    int dist(int a, int b){
        return d[a]+d[b]-2*d[lca(a, b)];
    }
};
int a[100010], b[100010];
ll c[100010];
vector<P> g[100010];
vector<P> g1[100020];
ll d1[10][100020];
ll d[100010];
void dfs(int x, int p){
    for(auto q:g[x]){
        int y=q.first;
        if(y==p) continue;
        d[y]=d[x]+q.second;
        dfs(y, x);
    }
}
int main()
{
    int n, k;
    cin>>n>>k;
    vector<vector<int>> gt(n);
    for(int i=0; i<n-1; i++){
        cin>>a[i]>>b[i]>>c[i];
        a[i]--; b[i]--;
        gt[a[i]].push_back(b[i]);
        gt[b[i]].push_back(a[i]);
        g[a[i]].push_back({b[i], c[i]*2});
        g[b[i]].push_back({a[i], c[i]*2});
        g1[a[i]].push_back({b[i], c[i]*2});
        g1[b[i]].push_back({a[i], c[i]*2});
    }
    for(int i=0; i<k; i++){
        int m; ll p; cin>>m>>p;
        for(int j=0; j<m; j++){
            int x; cin>>x; x--;
            g1[x].push_back({i+n, p});
            g1[i+n].push_back({x, p});
        }
    }
    const ll INF=1e18;
    for(int i=0; i<k; i++){
        priority_queue<P, vector<P>, greater<P>> que;
        fill(d1[i], d1[i]+n+k, INF);
        d1[i][i+n]=0;
        que.push({0, i+n});
        while(!que.empty()){
            auto p=que.top(); que.pop();
            int x=p.second;
            if(d1[i][x]<p.first) continue;
            for(auto q:g1[x]){
                int y=q.first;
                if(d1[i][y]>d1[i][x]+q.second){
                    d1[i][y]=d1[i][x]+q.second;
                    que.push({d1[i][y], y});
                }
            }
        }
    }
    LCA lca(gt);
    lca.build();
    dfs(0, -1);
    int q; cin>>q;
    while(q--){
        int u, v; cin>>u>>v; u--; v--;
        ll ans=d[u]+d[v]-2*d[lca.lca(u, v)];
        for(int i=0; i<k; i++){
            ans=min(ans, d1[i][u]+d1[i][v]);
        }
        ans/=2;
        cout<<ans<<endl;
    }
    return 0;
}
0