noi.ac #47 power set

#Description

小D梦见了一棵包含n个节点的树，这棵树包含着神秘的能量。具体来讲，对于这棵树中的一个联通块，它的能量为它拥有的节点中编号连续的最长的一段。举例来说，如果一个连通块包含了原树中编号为{1,3,4,5,7,8}的节点，那么编号连续的最长的一段为{3,4,5}。它所具有的能量值为3。

小D想要从这棵树中获得能量，但由于种种原因，小D只能从这棵树中选取一个节点数不大于k的连通块并获得它的能量值，小D想要知道他能取得的最大能量值是多少。

一棵树包含n个节点的树就是有n个点，n−1条边的连通图。

对于10%的数据，n≤20。
对于20%的数据，n≤200。
对于40%的数据，n≤2000。
对于另外15%的数据，保证给出的树是一条链。
对于另外15%的数据，保证树随机生成。
对于100%的数据，n≤100000,k≤n。
时间限制：1s1s 空间限制：512MB

#Solution

今天的T3，再给我十分钟就写出来了。还好暴力打得稳emm

考虑枚举答案中连续的区间，那么这个区间合法当且仅当区间内节点组成的连通块size不大于k
考虑用set维护这个东西，连通块的size=从任意一点出发走完全部染色节点并回到原点的最短路/2+1，这个东西就和bzoj3991一样了

感觉自己还是不太稳啊

#Code

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <set>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

const int N=200005;

struct edge {int y,next;} e[N*2];
struct data {
	int x,y;

	bool operator <(data b) const {
		return y<b.y;
	}
} ;

std:: set <data> set;

int fa[N],size[N],pos[N],bl[N],dep[N];
int ls[N],edCnt=1,tot;

bool vis[N];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void add_edge(int x,int y) {
	e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
	e[++edCnt]=(edge) {x,ls[y]}; ls[y]=edCnt;
}

void dfs1(int now) {
	size[now]=1;
	for (int i=ls[now];i;i=e[i].next) {
		if (e[i].y==fa[now]) continue;
		fa[e[i].y]=now; dep[e[i].y]=dep[now]+1;
		dfs1(e[i].y); size[now]+=size[e[i].y];
	}
}

void dfs2(int now,int up) {
	pos[now]=++pos[0]; bl[now]=up;
	int mx=0;
	for (int i=ls[now];i;i=e[i].next) {
		if (e[i].y!=fa[now]&&size[e[i].y]>size[mx]) mx=e[i].y;
	}
	if (!mx) return ;
	dfs2(mx,up);
	for (int i=ls[now];i;i=e[i].next) {
		if (e[i].y!=fa[now]&&e[i].y!=mx) dfs2(e[i].y,e[i].y);
	}
}

int get_lca(int x,int y) {
	for (;bl[x]!=bl[y];) {
		if (dep[bl[x]]<dep[bl[y]]) std:: swap(x,y);
		x=fa[bl[x]];
	}
	return (dep[x]<dep[y])?x:y;
}

int get_dis(int x,int y) {
	return dep[x]+dep[y]-dep[get_lca(x,y)]*2;
}

void ins(int r) {
	std:: set <data>:: iterator it;
	set.insert((data) {r,pos[r]}); int x,y;
	it=set.find((data) {r,pos[r]});
	if (set.size()==1) {
		return ;
	}

	x=y=0; it++;
	if (it!=set.end()) {
		x=(*it).x; tot+=get_dis(x,r);
	}
	it--;
	if (it!=set.begin()) {
		it--;
		y=(*it).x; tot+=get_dis(y,r);
		it++;
	}
	if (x&&y) tot-=get_dis(x,y);

}

void del(int l) {
	std:: set <data>:: iterator it;
	it=set.find((data) {l,pos[l]}); int x,y;
	if (set.size()==1) {
		set.erase(it);
		return ;
	}

	x=y=0; it++;
	if (it!=set.end()) {
		x=(*it).x; tot-=get_dis(x,l);
	}
	it--;
	if (it!=set.begin()) {
		it--;
		y=(*it).x; tot-=get_dis(y,l);
		it++;
	}
	if (x&&y) tot+=get_dis(x,y);
	set.erase((data) {l,pos[l]});
}

int calc() {
	if (!set.size()) return 0;
	std:: set <data>:: iterator it;
	it=set.begin(); int st=(*it).x;
	it=set.end(); it--; int ed=(*it).x;
	return get_dis(st,ed);
}

int main(void) {
	// freopen("data.in","r",stdin);
	int n=read(),m=read();
	rep(i,2,n) add_edge(read(),read());
	dfs1(dep[1]=1); dfs2(1,1); int ans=0;

	set.insert((data) {1,pos[1]});
	for (int l=1,r=1;r<=n;) { r++;
		ins(r);
		for (;(tot+calc())/2+1>m&&l<r;) {
			del(l++);
		}
		ans=std:: max(ans,r-l+1);
	}
	printf("%d\n", ans);
	return 0;
}