結果

問題 No.1600 Many Shortest Path Problems
コンテスト
ユーザー chocorusk
提出日時 2021-07-15 00:42:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 5,915 bytes
コンパイル時間 4,595 ms
コンパイル使用メモリ 202,872 KB
最終ジャッジ日時 2025-01-23 01:04:11
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19 WA * 32
権限があれば一括ダウンロードができます

ソースコード

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<int, 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)];
    }
};
struct unionfind{
    vector<int> par, sz;
    unionfind() {}
    unionfind(int n):par(n), sz(n, 1){
        for(int i=0; i<n; i++) par[i]=i;
    }
    int find(int x){
        if(par[x]==x) return x;
        return par[x]=find(par[x]);
    }
    void unite(int x, int y){
        x=find(x); y=find(y);
        if(x==y) return;
        if(sz[x]>sz[y]) swap(x, y);
        par[x]=y;
        sz[y]+=sz[x];
    }
    bool same(int x, int y){
        return find(x)==find(y);
    }
    int size(int x){
        return sz[find(x)];
    }
};
template<typename Monoid>
struct SegmentTree{
	using F=function<Monoid(Monoid, Monoid)>;
	int sz;
	vector<Monoid> seg;
	const F f;
	const Monoid e;

	SegmentTree(int n, const F f, const Monoid &e): f(f), e(e){
		sz=1;
		while(sz<n) sz<<=1;
		seg.resize(2*sz, e);
	}

	SegmentTree(int n, const F f, const Monoid &e, vector<Monoid> v): f(f), e(e){
		sz=1;
		while(sz<n) sz<<=1;
		seg.resize(2*sz, e);
		for(int i=0; i<n; i++) seg[i+sz]=v[i];
		for(int i=sz-1; i>=1; i--){
			seg[i]=f(seg[2*i], seg[2*i+1]);
		}
	}

	void update(int k, const Monoid &x){
		k+=sz;
		seg[k]=x;
		while(k>1){
			k>>=1;
			seg[k]=f(seg[2*k], seg[2*k+1]);
		}
	}

	Monoid query(int a, int b){
		a+=sz, b+=sz;
		Monoid ret=e;
		for(;a<b; a>>=1, b>>=1){
			if(b&1) ret=f(ret, seg[--b]);
			if(a&1) ret=f(ret, seg[a++]);
		}
		return ret;
	}

	/*Monoid query(int a, int b){ //演算が非可換のとき
		a+=sz, b+=sz;
		Monoid retl=e, retr=e;
		for(;a<b; a>>=1, b>>=1){
			if(b&1) retr=f(seg[--b], retr);
			if(a&1) retl=f(retl, seg[a++]);
		}
		return f(retl, retr);
	}*/

	Monoid operator[](const int &k) const{
		return seg[k+sz];
	}
};
int n, m;
int a[200020], b[200020];
int q;
int x[200020], y[200020], z[200020];
vector<vector<int>> g;
vector<vector<int>> ge;
vector<int> e;
using mint=modint1000000007;
mint d[200020];
int d0[200020];
mint p2[200020];
//mint ans[200020];
int l[200020], r[200020], rev[200020];
vector<int> v1[200020], v2[200020], vr[200020];
int mnl[200020], mnr[200020];
bool t[200020];
int main()
{
	cin>>n>>m;
	g.resize(n);
	ge.resize(n);
	for(int i=0; i<m; i++){
		cin>>a[i]>>b[i];
		a[i]--; b[i]--;
	}
	unionfind uf(n);
	for(int i=0; i<m; i++){
		if(uf.same(a[i], b[i])){
			e.push_back(i);
		}else{
			g[a[i]].push_back(b[i]);
			g[b[i]].push_back(a[i]);
			ge[a[i]].push_back(i);
			ge[b[i]].push_back(i);
			uf.unite(a[i], b[i]);
			t[i]=1;
		}
	}
	p2[0]=1;
	for(int i=0; i<n; i++) p2[i+1]=p2[i]*2;
	LCA lca(g);
	lca.build();
	int ord=0;
	auto dfs=[&](auto dfs, int x, int p)->void{
		l[x]=ord++;
		rev[l[x]]=x;
		for(int i=0; i<g[x].size(); i++){
			int y=g[x][i];
			if(y==p) continue;
			int j=ge[x][i];
			if(y==a[j]) swap(a[j], b[j]);
			d[y]=d[x]+p2[j+1];
			//d0[y]=d0[x]+1;
			dfs(dfs, y, x);
		}
		r[x]=ord;
		vr[r[x]].push_back(x);
	};
	dfs(dfs, 0, -1);
	const int INF=1e9;
	SegmentTree<int> seg(n, [](int x, int y){ return min(x, y);}, INF);
	for(auto i:e){
		if(l[a[i]]>l[b[i]]) swap(a[i], b[i]);
		v1[l[a[i]]].push_back(i);
		v2[l[b[i]]].push_back(i);
	}
	int mn1[200020];
	fill(mn1, mn1+n, INF);
	for(int i=0; i<n; i++){
		int r1=r[rev[i]];
		mnl[rev[i]]=seg.query(i, r1);
		for(auto j:v1[i]){
			int l1=l[b[j]];
			mn1[l1]=min(mn1[l1], j);
			seg.update(l1, mn1[l1]);
		}
	}
	SegmentTree<int> seg2(n, [](int x, int y){ return min(x, y);}, INF);
	fill(mn1, mn1+n, INF);
	for(int i=n; i>=1; i--){
		for(auto j:v2[i]){
			int l1=l[a[j]];
			mn1[l1]=min(mn1[l1], j);
			seg2.update(l1, mn1[l1]);
		}
		for(auto j:vr[i]){
			mnr[j]=seg2.query(l[j], r[j]);
		}
	}
	auto calc=[&](int x, int y){
		return d[x]+d[y]-2*d[lca.lca(x, y)];
	};
	cin>>q;
	for(int i=0; i<q; i++){
		cin>>x[i]>>y[i]>>z[i];
		x[i]--; y[i]--; z[i]--;
		if(!t[z[i]] || lca.dist(x[i], y[i])!=lca.dist(x[i], a[z[i]])+lca.dist(y[i], a[z[i]]) || lca.dist(x[i], y[i])!=lca.dist(x[i], b[z[i]])+lca.dist(y[i], b[z[i]])){
			mint ans=calc(x[i], y[i]);
			cout<<ans.val()<<endl;
		}else{
			int bz=b[z[i]];
			int k=min(mnl[bz], mnr[bz]);
			if(k==INF){
				cout<<-1<<endl;
				continue;
			}
			int a1=a[k], b1=b[k];
			if(lca.dist(x[i], a1)>lca.dist(x[i], b1)) swap(a1, b1);
			mint ans=calc(x[i], a1)+calc(y[i], b1)+p2[k+1];
			cout<<ans.val()<<endl;
		}
	}

    return 0;
}
0