P8805 [蓝桥杯 2022 国 B] 机房

[蓝桥杯 2022 国 B] 机房

提前声明,这个解法因为评测样例在洛谷过不了,但我是在蓝桥杯官网做的,全过
因为想看看能不能DJ暴力出来
正解是Tarjan->LCA

题目描述

这天，小明在机房学习。

他发现机房里一共有 $n$ 台电脑，编号为 1 到 $n$，电脑和电脑之间有网线连接，一共有 $n-1$ 根网线将 $n$ 台电脑连接起来使得任意两台电脑都直接或者间接地相连。

小明发现每台电脑转发、发送或者接受信息需要的时间取决于这台电脑和多少台电脑直接相连，而信息在网线中的传播时间可以忽略。比如如果某台电脑用网线直接连接了另外 $d$ 台电脑, 那么任何经过这台电脑的信息都会延迟 $d$ 单位时间 (发送方和接收方也会产生这样的延迟，当然如果发送方和接收方都是 同一台电脑就只会产生一次延迟)。

小明一共产生了 $m$ 个疑问：如果电脑 $u_i$ 向电脑 $v_i$ 发送信息，那么信息从 $u_i$ 传到 $v_i$ 的最短时间是多少?

输入格式

输入共 $n+m$ 行，第一行为两个正整数 $n,m$ 。

后面 $n-1$ 行，每行两个正整数 $x,y$ 表示编号为 $x$ 和 $y$ 的两台电脑用网线 直接相连。

后面 $m$ 行，每行两个正整数 $u_i,v_i$ 表示小明的第 $i$ 个疑问。

输出格式

输出共 $m$ 行，第 $i$ 行一个正整数表示小明第 $i$ 个疑问的答案。

样例 #1

样例输入 #1

4 3
1 2
1 3
2 4
2 3
3 4
3 3

样例输出 #1

5
6
1

提示

【样例说明】

这四台电脑各自的延迟分别为 $2,2,1,1$ 。

对于第一个询问, 从 $2$ 到 $3$ 需要经过 $2,1,3$, 所以时间和为 $2+2+1=5$。对于第二个询问，从 $3$ 到 $4$ 需要经过 $3,1,2,4$，所以时间和为 $1+2+2+1=6$。

对于第三个询问，从 $3$ 到 $3$ 只会产生一次延迟, 所以时间为 1。

【评测用例规模与约定】

对于 $30\%$ 的数据，保证 $n,m\le 1000$;

对于 $100\%$ 的数据，保证 $n,m\le 10^5$。

蓝桥杯 2022 国赛 B 组 H 题。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+10;
struct edge{
	int to;ll w;
	edge(int b,ll c){to=b;w=c;}
};
struct node{
	int id;ll cost;
	node(int a,ll c){id=a;cost=c;}
	bool operator<(const node& cur)const{return cur.cost<cost;}
};
vector<edge>adj[N];
ll dis[N];
priority_queue<node>wait;
bool ismin[N];
void dj(int start){
	dis[start]=0;
	wait.push(node(start,dis[start]));
	while(!wait.empty()){
		node cur=wait.top();wait.pop();
		if(ismin[cur.id])continue;
		ismin[cur.id]=true;
		for(const edge& e:adj[cur.id]){
			if(ismin[e.to])continue;
			if(dis[e.to]>cur.cost+e.w){
				dis[e.to]=cur.cost+e.w;
				wait.push(node(e.to,dis[e.to]));
			}
		}
	}
}
int main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	int n,m;cin>>n>>m;
	int v1,v2;
	ll du[n+10];memset(du,0,sizeof(du)); 
	for(int i=1;i<=n-1;++i){
		cin>>v1>>v2;
		adj[v1].push_back(edge(v2,0));
		adj[v2].push_back(edge(v1,0));
		++du[v1];++du[v2];
	}
	for(int i=1;i<=n;++i)if(adj[i].size()>0)for(edge& e:adj[i])
		e.w=du[i];
//	for(int i=1;i<=n;++i)if(adj[i].size()>0)for(edge& e:adj[i])
//		cout<<"结点"<<i<<"到邻居结点"<<e.to<<"的边权为"<<e.w<<endl; 
	memset(dis,0x7f,sizeof(dis));
	memset(ismin,false,sizeof(ismin)); 
	// int lastdj=1e5+1;
	for(int i=0;i<m;++i){
		cin>>v1>>v2;
		if(v1==v2){cout<<du[v2]<<endl;continue;}
		// if(v1==lastdj&&ismin[v2]){
		// 	cout<<dis[v2]<<endl;
		// 	continue;
		// }	
		memset(ismin,false,sizeof(ismin));
		memset(dis,0x7f,sizeof(dis));
		dj(v1);
    // lastdj=v1;
		cout<<dis[v2]+du[v2]<<endl;
	}

	return 0;
}