ZOJ 3195 LCA+RMQ+找规律

题意

给三个点，求这三个点的相连最短距离。

题解

LCA+RMQ是模板，没什么好说的。规律的话，就是最短距离为dis[a]+dis[b]+dis[c]-dis[num[x1]]-dis[num[x2]]-dis[num[x3]]。画一下图就能发现这个规律对于所有的可能情况都是满足的。

注意事项

PE好几发，ORZ。。这种输入方式还搞Between Blank Line也是很有趣。。

代码

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<bitset>
#include<stack>
#include<string>
#pragma comment(linker,"/STACK:102400000,102400000")
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define MEM(a,b) memset(a,b,sizeof(a))
#define LL long long
#define INF 0x3f3f3f3f
#define MAXN 50010
#define MOD 1000000007
#define EPS 1e-3
using namespace std;
struct Node {
    int to,val;

    Node(int to,int val):to(to),val(val) {}
};
int num[MAXN<<1],dep[MAXN<<1],id[MAXN],ds[MAXN];
int d[MAXN<<1][20];
vector<Node> vc[MAXN];
int n,m,c,k;

void dfs(int u,int p,int deep,int dis) {
    id[u]=k;
    ds[u]=dis;
    num[k]=u;
    dep[k++]=deep;
    int sz=vc[u].size();
    UP(i,0,sz) {
        Node nd=vc[u][i];
        if(nd.to==p)
            continue;
        dfs(nd.to,u,deep+1,dis+nd.val);
        num[k]=u;
        dep[k++]=deep;
    }
}

void rmq_init() {
    UP(i,0,k) {
        d[i][0]=i;
    }
    for(int j=1; (1<<j)<=k; j++) {
        for(int i=0; i+(1<<j)-1<k; i++) {
            if(dep[d[i][j-1]]>dep[d[i+(1<<(j-1))][j-1]])
                d[i][j]=d[i+(1<<(j-1))][j-1];
            else
                d[i][j]=d[i][j-1];
        }
    }
}

int rmq(int l,int r) {
    int kk=0;
    W((1<<(kk+1))<=r-l+1) {
        kk++;
    }
    if(dep[d[l][kk]]>dep[d[r-(1<<kk)+1][kk]])
        return d[r-(1<<kk)+1][kk];
    else
        return d[l][kk];
}

bool cmp(int x,int y) {
    return id[x]<id[y];
}

int main() {
    bool first=true;
    W(~scanf("%d",&n)) {
        if(first) {
            first=false;
        } else {
            puts("");
        }
        MEM(d,0);
        MEM(num,0);
        MEM(dep,0);
        MEM(id,-1);
        MEM(ds,0);
        MEM(vc,0);
        k=0;
        int a,b,w;
        UP(i,0,n-1) {
            scanf("%d%d%d",&a,&b,&w);
            vc[a].push_back(Node(b,w));
            vc[b].push_back(Node(a,w));
        }
        UP(i,0,n) {
            if(id[i]==-1)
                dfs(i,-1,0,0);
        }
        rmq_init();
        scanf("%d",&c);
        int aa[3];
        W(c--) {
            scanf("%d%d%d",&aa[0],&aa[1],&aa[2]);
            sort(aa,aa+3,cmp);
            int x1=rmq(id[aa[1]],id[aa[2]]);
            int x2=rmq(id[aa[0]],id[aa[2]]);
            int x3=rmq(id[aa[0]],id[aa[1]]);
            printf("%d\n",ds[aa[1]]+ds[aa[2]]+ds[aa[0]]-ds[num[x1]]-ds[num[x2]]-ds[num[x3]]);
        }
    }
}

/*
5 3 5
1 3 2
2 4 3
5 2 3
*/