标签:
/* * Author: sweat123 * Created Time: 2016/7/13 10:56:50 * File Name: main.cpp */ #include<set> #include<map> #include<queue> #include<stack> #include<cmath> #include<string> #include<vector> #include<cstdio> #include<time.h> #include<cstring> #include<iostream> #include<algorithm> #define INF 1<<30 #define MOD 1000000007 #define ll long long #define lson l,m,rt<<1 #define key_value ch[ch[root][1]][0] #define rson m+1,r,rt<<1|1 #define pi acos(-1.0) using namespace std; const int MAXN = 10010; struct node{ int to; int val; int next; }edge[MAXN*2]; int dp[MAXN*2][20],ver[MAXN*2],vis[MAXN],dfn[MAXN*2],first[MAXN],pre[MAXN]; ll dis[MAXN]; int n,m,q,tot,ind,mark[MAXN]; void add(int x,int y,int z){ edge[ind].to = y; edge[ind].val = z; edge[ind].next = pre[x]; pre[x] = ind ++; } void dfs(int rt,int dep,int flag){ vis[rt] = 1; ver[++tot] = rt; dfn[tot] = dep; first[rt] = tot; mark[rt] = flag; for(int i = pre[rt]; i != -1; i = edge[i].next){ int t = edge[i].to; if(!vis[t]){ dis[t] = dis[rt] + edge[i].val; dfs(t,dep+1,flag); ver[++tot] = rt; dfn[tot] = dep; } } } void rmq(){ for(int i = 1; i <= tot; i++){ dp[i][0] = i; } for(int i = 1; i < 20; i++){ for(int j = 1; j + (1 << i) - 1 <= tot; j++){ int x = dp[j][i-1]; int y = dp[j+(1<<(i-1))][i-1]; if(dfn[x] > dfn[y]){ dp[j][i] = y; } else{ dp[j][i] = x; } } } } int askrmq(int x,int y){ x = first[x]; y = first[y]; if(x > y)swap(x,y); int k = (int)(log(y - x + 1) * 1.0 / log(2.0)); int l = dp[x][k]; int r = dp[y - (1<<k) + 1][k]; if(dfn[l] > dfn[r])return r; else return l; } int main(){ while(~scanf("%d%d%d",&n,&m,&q)){ ind = tot = 0; memset(vis,0,sizeof(vis)); memset(pre,-1,sizeof(pre)); for(int i = 1; i <= m; i++){ int x,y,z; scanf("%d%d%d",&x,&y,&z); add(x,y,z); add(y,x,z); } int cnt = 0; memset(mark,0,sizeof(mark)); for(int i = 1; i <= n; i++){ if(!vis[i])dfs(i,1,++cnt); } rmq(); while(q--){ int x,y; scanf("%d%d",&x,&y); if(mark[x] != mark[y]){ printf("Not connected\n"); } else{ int tp = ver[askrmq(x,y)]; ll ans = dis[x] - dis[tp] + dis[y] - dis[tp]; printf("%lld\n",ans); } } } return 0; }
标签:
原文地址:http://www.cnblogs.com/sweat123/p/5666233.html