3611: [Heoi2014] 대공사

2909 단어
계속해서 수허수.
이 문제의 60S 시한은 나를 매우 놀라게 했다.
그러나 3000여 MS가 A가 되었는데 다행히 랭크 4, 나이스!
처음에는 점분치인지 DP인지 고민하다가 DP가 될 수 있으면 DP가 되지 않을까 생각했어요.
그래서 DP를 했어요.
분명히 세 개의 양을 유지해야 한다.
ans2와 ans3은 모두 말하기 쉬우니 나무의 직경을 참조하여 구법을 구하면 됩니다. 차장+최장+1로 변형하세요.
ans1의 말은 비교적 번거롭다. 손으로 그림을 그려 보자. 눈으로 측정한 것은 현재 자나무의 가장자리가 다른 자나무의 가장자리와 대응하는 것이다. 그리고 다른 자나무의 가장자리가 현재 자나무의 가장자리와 계산하고 손으로 밀면 된다.
그리고 OK.
역시 Clj의 말대로 처음에는 재미있었는데 많이 하면 심미적 피로가 온다.
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int read(){
	int x=0;char ch;
	while(ch<'0'||ch>'9')ch=getchar();
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x;
}
const int N=1000000+5;
const int inf=1e9;
struct Edge{int to,next,v;}e[N<<1];
int head[N],cnt;
int siz[N],son[N],dep[N],fa[N],top[N];
int dfn[N],dfs_clock;
void ins(int u,int v){
	if(u==v)return;
	e[++cnt]=(Edge){v,head[u],dep[v]-dep[u]};head[u]=cnt;
}
void dfs1(int u){
	siz[u]=1;son[u]=0;dfn[u]=++dfs_clock;
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;if(v==fa[u])continue;
		dep[v]=dep[u]+1;fa[v]=u;
		dfs1(v);
		siz[u]+=siz[v];
		if(siz[v]>siz[son[u]])son[u]=v;
	}
}
void dfs2(int u,int tp){
	top[u]=tp;
	if(son[u])dfs2(son[u],tp);
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(v!=son[u]&&v!=fa[u])dfs2(v,v);
	}
}
int lca(int u,int v){
	while(top[u]!=top[v]){
		if(dep[top[u]]>=dep[top[v]])u=fa[top[u]];
		else v=fa[top[v]];
	}
	return dep[u]<dep[v]?u:v;
}
ll sum[N],ans1;
int mx[N],mi[N],ans2,ans3; 
bool build[N]; 
void dp(int u){
	siz[u]=build[u];
	mx[u]=build[u]?0:-inf;
	mi[u]=build[u]?0:inf;
	sum[u]=0;
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		dp(v);
		ans1+=(sum[u]+siz[u]*e[i].v)*siz[v]+sum[v]*siz[u];
		siz[u]+=siz[v];
		sum[u]+=sum[v]+siz[v]*e[i].v;
		ans2=min(ans2,mi[u]+mi[v]+e[i].v);
		ans3=max(ans3,mx[u]+mx[v]+e[i].v);
		mi[u]=min(mi[u],mi[v]+e[i].v);
		mx[u]=max(mx[u],mx[v]+e[i].v);
	}
	head[u]=0;
}
int h[N];
int st[N],tp;
bool cmp(int i,int j){
	return dfn[i]<dfn[j];
}
void solve(){
	int k=read();
	for(int i=1;i<=k;i++){
		h[i]=read();
		build[h[i]]=true;
	}
	sort(h+1,h+1+k,cmp);
	tp=cnt=0;
	st[++tp]=1;
	for(int i=1;i<=k;i++){
		int now=h[i],f=lca(now,st[tp]);
		if(f==st[tp]){st[++tp]=now;continue;}
		while(f==lca(now,st[tp-1])){
			ins(st[tp-1],st[tp]);
			tp--;f=lca(now,st[tp]);
		}
		ins(f,st[tp]);
		st[tp]=f;st[++tp]=now;
	}
	while(--tp)ins(st[tp],st[tp+1]);
	ans1=0;ans2=inf;ans3=-inf;
	dp(1);
	printf("%lld %d %d
",ans1,ans2,ans3); for(int i=1;i<=k;i++)build[h[i]]=false; } int main(){ int n=read(); int u,v; for(int i=1;i<n;i++){ u=read();v=read(); ins(u,v);ins(v,u); } dfs1(1);dfs2(1,1); memset(head,0,sizeof(head)); int q=read(); while(q--)solve(); return 0; }

좋은 웹페이지 즐겨찾기