比赛 2024国庆练习2 评测结果 WWAAAEEEEE
题目名称 消防演练 最终得分 30
用户昵称 徐诗畅 运行时间 1.321 s
代码语言 C++ 内存使用 3.53 MiB
提交时间 2024-10-05 17:49:08
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int N=10005;
int n,cnt,head[N];
struct edge{
	int v,next;
}e[N<<1];
void addedge(int u,int v){
	e[++cnt].v=v;
	e[cnt].next=head[u];
	head[u]=cnt;
}
struct node{
	int u,num;
	bool operator<(const node &r) const{
		return num<r.num;
	}
};
priority_queue<node> q;
int nson[N],num[N],r[N],root,tot;
void dfs(int u,int fa){
	if(u==root) num[u]=nson[u]+1;
	else num[u]=nson[u]+num[fa]-1; 
	q.push(node{u,num[u]});
	if(fa==root) r[u]=++tot;
	else r[u]=r[fa];
	for(int i =head[u];i;i=e[i].next){
		int v=e[i].v;
		if(v==fa) continue;
		dfs(v,u);
	}
}
int main(){
	freopen("drill.in","r",stdin);
	freopen("drill.out","w",stdout);
	scanf("%d",&n);
	for(int i = 1;i<=n;i++) nson[i]=-1;
	for(int i = 1;i<n;i++){
		int u,v; scanf("%d%d",&u,&v);
		addedge(u,v); addedge(v,u);
		nson[u]++; nson[v]++; 
	}
	int ans=0;
	for(int i = 1;i<=n;i++) ans=max(ans,nson[i]+1);
	for(int i = 1;i<=n;i++){
		memset(num,0,sizeof(num));
		memset(r,0,sizeof(r));
		while(!q.empty()) q.pop();
		root=i; tot=0;
		dfs(i,0); 
		while(!q.empty()&&q.top().u==root) q.pop();
		int ans1=q.top().num,tmp1=q.top().u; q.pop();
		while(!q.empty()&&r[q.top().u]==r[tmp1]) q.pop();
		while(!q.empty()&&q.top().u==root) q.pop();
		if(!q.empty()) ans=max(ans,num[tmp1]+num[q.top().u]-nson[root]-1);
	}
	printf("%d",ans);
	return 0;
}