BZOJ1131 [POI2008]Sta（洛谷P3478）

最新推荐文章于 2018-11-02 20:55:52 发布

forezxl

最新推荐文章于 2018-11-02 20:55:52 发布

阅读量211

点赞数 1

CC 4.0 BY-SA版权

分类专栏：洛谷 BZOJ 搜索---DFS（序）蒟蒻zxl的Blog专栏文章标签： BZOJ 洛谷 DFS

本文链接：https://blog.youkuaiyun.com/a1799342217/article/details/83348500

蒟蒻zxl的Blog专栏同时被 3 个专栏收录

460 篇文章

订阅专栏

洛谷

292 篇文章

订阅专栏

BZOJ

281 篇文章

订阅专栏

这篇博客介绍了如何使用DFS解决BZOJ1131和洛谷P3478问题。作者通过DFS遍历每个节点，计算节点的size以及根节点的答案。对于子节点，其子树到自身的距离减1，其他点的距离加1，以此来得到完整的解决方案。提供了相关题目链接及代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

DFS

BZOJ题目传送门
 洛谷题目传送门

先一遍DFS算出每个节点的 $s i z e$ 并统计根节点的答案。而对于一个子节点，当它作为根时，它子树到它的距离-1，其他点的距离+1。那么全算出来就好了。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000005
#define F inline
using namespace std;
typedef long long LL;
struct edge{ int nxt,to; }ed[N<<1];
int n,k,h[N],fa[N],sz[N],dep[N];
LL ans,s[N];
F char readc(){
	static char buf[100000],*l=buf,*r=buf;
	if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
	return l==r?EOF:*l++;
}
F int _read(){
	int x=0; char ch=readc();
	while (!isdigit(ch)) ch=readc();
	while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
	return x;
}
void dfs1(int x){
	dep[x]=dep[fa[x]]+1,sz[x]=1;
	for (int i=h[x],v;i;i=ed[i].nxt)
		if ((v=ed[i].to)!=fa[x])
			fa[v]=x,dfs1(v),sz[x]+=sz[v];
}
void dfs2(int x){
	s[x]=s[fa[x]]-sz[x]+n-sz[x];
	for (int i=h[x],v;i;i=ed[i].nxt)
		if ((v=ed[i].to)!=fa[x]) dfs2(v);
}
#define add(x,y) ed[++k]=(edge){h[x],y},h[x]=k
int main(){
	n=_read();
	for (int i=1,x,y;i<n;i++) x=_read(),y=_read(),add(x,y),add(y,x);
	dfs1(1); for (int i=1;i<=n;i++) s[1]+=dep[i]-1;
	for (int i=h[1];i;i=ed[i].nxt) dfs2(ed[i].to);
	for (int i=1;i<=n;i++) if (s[ans]<s[i]) ans=i;
	return printf("%lld\n",ans),0;
}