zoukankan      html  css  js  c++  java
  • B

    //#include<bits/stdc++.h>
    //#pragma comment(linker, "/STACK:1024000000,1024000000") 
    #include<stdio.h>
    #include<algorithm>
    #include<queue>
    #include<string.h>
    #include<iostream>
    #include<math.h>
    #include<set>
    #include<map>
    #include<vector>
    #include<iomanip>
    using namespace std;
    #define ll long long
    #define pb push_back
    
    const int maxn=4e4+9;
    int n,m,root;
    
    struct EDGE{int v;int d;};
    vector<EDGE>G[maxn];
    
    int vs[maxn*2-1];	//每次访问到的节点
    int dep[maxn*2-1];	//节点深度
    int id[maxn];		//编号节点第一次访问的时间
    
    void addedge(int u,int v,int d){
    	G[u].pb((EDGE){v,d});
    	G[v].pb((EDGE){u,d});
    }
    
    void dfs(int u,int fa,int d,int &k){
    	id[u]=++k;
    	vs[k]=u;dep[k]=d;int sz=G[u].size();
    	for(int i=0;i<sz;i++){
    		EDGE &e=G[u][i];
    		if(e.v==fa)continue;
    		dfs(e.v,u,d+e.d,k);	
    		vs[++k]=u;dep[k]=d;
    	}
    }
    void init_dfs(){
    	int k=0;
    	dfs(root,-1,0,k);
    }
    
    int dp[maxn<<1][20];
    int mm[maxn<<1];
    void init_rmq(int n,int dep[]){
    	mm[0]=-1;
    	for(int i=1;i<=n;i++){
    		mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
    		dp[i][0]=i;
    	}	
    	for(int j=1;j<=mm[n];j++){
    		for(int i=1;i+(1<<j)-1<=n;i++){
    			if(dep[dp[i][j-1]]<dep[dp[i+(1<<(j-1))][j-1]])
    				dp[i][j]=dp[i][j-1];
    			else dp[i][j]=dp[i+(1<<(j-1))][j-1];
    		}
    	}
    }
    int rmq(int l,int r){
    	if(l>r)swap(l,r);
    	int k=mm[r-l+1];
    	return dep[dp[l][k]]<=dep[dp[r-(1<<k)+1][k]]?dp[l][k]:dp[r-(1<<k)+1][k];
    }
    
    int lca(int u,int v){
    	return vs[rmq(id[u],id[v])];
    }
    
    int main(){
    	int n,m;
    	while(~scanf("%d%d",&n,&m)){
    		for(int i=1;i<=n;i++)G[i].clear();
    		int u,v,w;char op[3];
    		for(int i=1;i<=m;i++){
    			scanf("%d%d%d%s",&u,&v,&w,op);
    			addedge(u,v,w);	
    		}
    		root=1;
    		init_dfs();
    		init_rmq(n*2-1,dep);
    		int q;
    		scanf("%d",&q);
    		while(q--){
    			int u,v;
    			scanf("%d%d",&u,&v);
    			printf("%d
    ",dep[id[u]]+dep[id[v]]-2*dep[id[lca(u,v)]]);
    		}
    	}
    }
    接口统一之后果然好多了
  • 相关阅读:
    leetcode33. Search in Rotated Sorted Array
    pycharm 设置sublime text3 monokai主题
    django class Meta
    leetcode30, Substring With Concatenation Of All Words
    Sublime text3修改tab键为缩进为四个空格,
    sublime text3 python打开图像的问题
    安装上imesupport输入法依然不跟随的解决办法,
    sublime text3 的插件冲突弃用问题,
    sublime text3 BracketHighlighter括号匹配的设置
    windows 下wget的使用
  • 原文地址:https://www.cnblogs.com/Drenight/p/8611303.html
Copyright © 2011-2022 走看看