CDOJ 1437 谭松松的旅游计划 Label:倍增LCA && 最短路

谭松松的旅游计划

Time Limit: 3000/1000MS (Java/Others)     Memory Limit: 65535/65535KB (Java/Others)

Submit Status

谭松松是一个爱旅游的人，他非常的热爱旅游，即便身体被掏空也要坚持旅游。喵蛤王国由N个城市组成，由N-1条道路连通，每条道路长度为ci,使得任意两座城市都能相互到达。谭松松有M个旅游计划，每个旅游计划有一个起点ai,一个终点bi,所以谭松松想知道，对于每个旅游计划，从起点到终点的最短路长度为多少？

Input

第一行两个整数N(2≤N≤100000，)，M(1≤M≤100000)，表示城市个数，谭松松的旅游计划个数。

接下来N-1行，每行三个整数li,ri,ci（1≤ci≤10000），表示li号城市和ri号城市之间有一条长度为ci的道路。

接下来M行，每行两个整数ai,bi表示旅游计划的起点和终点。

Output

输出M行，表示每个旅游计划的最短距离。

Sample input and output

Sample Input	Sample Output
7 6 1 2 5 1 3 4 3 7 7 2 4 4 2 5 3 5 6 2 1 2 4 7 6 3 3 2 5 4 7 6	5 20 14 9 7 21

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#define ll long long
const int MAXN=200005;
using namespace std;

int N,M;
int pa[30][MAXN],depth[MAXN],vis[MAXN];
ll cost[30][MAXN];//注意long long 
vector<int> G[MAXN],c[MAXN];

void dfs(int x,int pre,int d){
    vis[x]=1;
    depth[x]=d;
    pa[0][x]=pre;
    for(int i=0;i<G[x].size();i++){
        int to=G[x][i];
        if(!vis[to]){
            cost[0][to]=c[x][i];
            dfs(to,x,d+1);
        }
    }
}

void init_lca(){
    memset(pa,-1,sizeof(pa));
    
    dfs(1,-1,0);
    for(int k=0;k<=25;k++){
        for(int i=1;i<=N;i++){
            if(pa[k][i]>0){
                pa[k+1][i]=pa[k][ pa[k][i] ];
                cost[k+1][i]=cost[k][i]+cost[k][ pa[k][i] ];
//                cout<<cost[k+1][i]<<endl;
            }
        }
    }
}

ll lca(int u,int v){//long long
    ll sum=0;
    if(depth[u]>depth[v]) swap(u,v);
//    cout<<u<<v<<endl;
    for(int k=0;k<25;k++){
        if((depth[v]-depth[u])>>k&1){
            sum+=cost[k][v];
            v=pa[k][v];//这两行不能换..... 
//            cout<<cost[k][u]<<endl;
        }
    }
    
    if(u==v) return sum;
    for(int k=25;k>=0;k--){
        if(pa[k][u]!=pa[k][v]){
            sum+=(cost[k][u]+cost[k][v]);
            u=pa[k][u];
            v=pa[k][v];
        }
    }
    sum+=(cost[0][v]+cost[0][u]);
    return sum;
}

int main(){
//    freopen("01.in","r",stdin);
    scanf("%d%d",&N,&M);
    for(int i=1;i<N;i++){
        int x,y,cc;
        scanf("%d%d%d",&x,&y,&cc);
        G[x].push_back(y);
        G[y].push_back(x);
        c[x].push_back(cc);
        c[y].push_back(cc);
    }
    init_lca();
    while(M--){
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%lld
",lca(a,b));//此处用cout会T!!! 
    }
    return 0;
}

Line50 和Line51换了会爆炸

然后83用cout会爆炸，直接T掉

输出1万次就得用printf

cout也就比printf慢几千倍吧~