• POJ


    1、一颗树中,给出a,b,求最近的距离。(我没考虑不联通的情况,即不是一颗树的情况)

    2、用最近公共祖先来求,

    记下根结点到任意一点的距离dis[],这样ans = dis[u] + dis[v] - 2 * dis[lca(u, v)]

    3、

    /*
    离线算法,LCATarjan
    复杂度O(n+Q);
    */
    #include<iostream>
    #include<stdio.h>
    #include<string.h>
    using namespace std;
    
    const int MAXN=40005;
    const int MAXQ=10005;//查询数的最大值
    
    int dis[MAXN];//到根节点的距离
    
    //并查集部分
    int F[MAXN];//需要初始化为-1
    int find(int x){
        if(F[x]==-1)return x;
        return F[x]=find(F[x]);
    }
    void bing(int u,int v){
        int t1=find(u);
        int t2=find(v);
        if(t1!=t2)
            F[t1]=t2;
    }
    //***********************
    bool vis[MAXN];//访问标记
    int ancestor[MAXN];//祖先
    struct Edge{
        int to,next;
        int d;
    }edge[MAXN*2];
    int head[MAXN],tot;
    void addedge(int u,int v,int d){
        edge[tot].to=v;
        edge[tot].d=d;
        edge[tot].next=head[u];
        head[u]=tot++;
    }
    
    struct Query{
        int q,next;
        int index;//查询编号
    }query[MAXQ*2];
    int answer[MAXQ];//存储最后的查询结果,下标0 Q-1
    int h[MAXN];//注意此处为MAXN...
    int tt;
    
    void add_query(int u,int v,int index){
        query[tt].q=v;
        query[tt].next=h[u];
        query[tt].index=index;
        h[u]=tt++;
        query[tt].q=u;
        query[tt].next=h[v];
        query[tt].index=index;
        h[v]=tt++;
    }
    
    void init(){
        tot=0;
        memset(head,-1,sizeof(head));
        tt=0;
        memset(h,-1,sizeof(h));
        memset(vis,false,sizeof(vis));
        memset(F,-1,sizeof(F));
        memset(ancestor,0,sizeof(ancestor));
    }
    void LCA(int u){
        ancestor[u]=u;
        vis[u]=true;
        for(int i=head[u];i!=-1;i=edge[i].next){
            int v=edge[i].to;
            if(vis[v])continue;
            dis[v]=dis[u]+edge[i].d;
            LCA(v);
            bing(u,v);
            ancestor[find(u)]=u;
        }
        for(int i=h[u];i!=-1;i=query[i].next){
            int v=query[i].q;
            if(vis[v]){
                answer[query[i].index]=ancestor[find(v)];
            }
        }
    }
    
    int main(){
        int n,m;
        int i;
        int u,v,d;
        char str[2];
    
        init();
        scanf("%d%d",&n,&n);
        for(i=0;i<n;++i){
            scanf("%d%d%d%1s",&u,&v,&d,str);
            addedge(u,v,d);
            addedge(v,u,d);
        }
        scanf("%d",&m);
        for(i=0;i<m;++i){
            scanf("%d%d",&u,&v);
            add_query(u,v,i);
        }
        dis[1]=0;
        LCA(1);
        for(i=0;i<m;++i){
            printf("%d
    ",dis[query[i*2].q]+dis[query[i*2+1].q]-2*dis[answer[i]]);
        }
    
        return 0;
    }
    View Code
  • 相关阅读:
    洛谷P1368 均分纸牌(加强版) [2017年6月计划 数论14]
    洛谷P1621 集合 [2017年6月计划 数论13]
    NOIP模拟 6.30
    洛谷P1390 公约数的和 [2017年6月计划 数论12]
    洛谷P1573 栈的操作 [2017年6月计划 数论11]
    洛谷P2429 制杖题 [2017年6月计划 数论10]
    Vijos1212 Way Selection [2017年6月计划 二分图03]
    NOIP模拟赛 6.29
    RQNOJ PID192 梦幻大PK [2017年6月计划 二分图02]
    RQNOJ PID141 / 寻找代表元 [2017年6月计划 二分图01]
  • 原文地址:https://www.cnblogs.com/gongpixin/p/4963609.html
Copyright © 2020-2023  润新知