关于传智杯F题,小【】与落叶
  • 板块学术版
  • 楼主蒟蒻君HJT泽渡透香
  • 当前回复5
  • 已保存回复5
  • 发布时间2021/12/23 18:33
  • 上次更新2023/10/28 13:50:04
查看原帖
关于传智杯F题,小【】与落叶
131591
蒟蒻君HJT泽渡透香楼主2021/12/23 18:33

为什么离线树剖t掉了?我不能理解,泪奔

#include <bits/stdc++.h>
using namespace std;
int n,m,dep[100005],dfn[100005],rnk[100005];
bool visited[100005];
int hson[100005],siz[100005],node_top[100005],fa[100005];
int head[200005],ver[200005],nxt[200005];
int tot=0;int oto=1;
struct node{
	int l,r,v,lazy;
};
struct re{
	int op,x,d,ret;//op=1 insert x,dep[x],op=2 find x,>=d
};
node tree[400005];
re ak[200005];
int ans[100005],cnt_ans=0;
int x,y;int op;
int cnt=0;
bool cmp1(re a,re b){return a.d==b.d?a.op==1:a.d>b.d;}
void add_edge(int x,int y){
	++oto;nxt[oto]=head[x];head[x]=oto;
	ver[oto]=y;return ;
}
void dfs1(int now,int deep){
	dep[now]=deep;
	visited[now]=true;
	siz[now]=1;int maxx=0;
	for(int i=head[now];i;i=nxt[i]){
		if(visited[ver[i]]) continue;
		dfs1(ver[i],deep+1);
		siz[now]+=siz[ver[i]];
		fa[ver[i]]=now;
		if(siz[ver[i]]>maxx){
			maxx=siz[ver[i]];
			hson[now]=ver[i];
		}
	}
	return ;
}
void dfs2(int now,int now_top){
	++tot;dfn[now]=tot;rnk[tot]=now;
	node_top[now]=now_top;
	visited[now]=true;
	if(hson[now]) dfs2(hson[now],now_top);
	for(int i=head[now];i;i=nxt[i]){
		if(visited[ver[i]] || ver[i]==hson[now]) continue;
		dfs2(ver[i],ver[i]);
	}
	return ;
}
void build(int now,int l,int r){
	tree[now].l=l;
	tree[now].r=r;
	tree[now].v=0;
	tree[now].lazy=0;
	if(l==r) return ;
	int mid=(l+r)>>1;
	build(now*2,l,mid);
	build(now*2+1,mid+1,r);
	return ;
}
void add(int now,int l,int r){
	tree[now].v+=(r-l+1);
	if(tree[now].l==l && tree[now].r==r){
		tree[now].lazy+=1;
		return ;
	}
	int mid=(tree[now].l+tree[now].r)>>1;
	if(r<=mid) add(now*2,l,r);
	else if(l>=mid+1) add(now*2+1,l,r);
	else {add(now*2,l,mid);add(now*2+1,mid+1,r);}
	return ;
}
int ask(int now,int v){
	if(tree[now].l==tree[now].r) return tree[now].v;
	if(tree[now].lazy!=0){
		tree[now*2].lazy+=tree[now].lazy;
		tree[now*2+1].lazy+=tree[now].lazy;
		tree[now*2].v+=(tree[now*2].r-tree[now*2].l+1)*tree[now].lazy;
		tree[now*2+1].v+=(tree[now*2+1].r-tree[now*2+1].l+1)*tree[now].lazy;
		tree[now].lazy=0;
	}
	int mid=(tree[now].l+tree[now].r)>>1;
	if(v<=mid) return ask(now*2,v);
	else return ask(now*2+1,v);
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n-1;i++){
		scanf("%d%d",&x,&y);
		add_edge(x,y);
		add_edge(y,x);
	}
	build(1,1,n);
	memset(visited,false,sizeof(visited));
	dfs1(1,1);
	memset(visited,false,sizeof(visited));
	dfs2(1,1);
	re k;
	//for(int i=1;i<=n;i++) printf("%d:%d\n",i,node_top[i]);
	for(int i=1;i<=n;i++){
		k.op=1;k.x=i;k.d=dep[i];
		++cnt;ak[cnt]=k;
	}
	//op=1: >=x huang
	//op=2: x??
	int las=n+1;
	for(int i=1;i<=m;i++){
		scanf("%d",&op);
		if(op==1){
			scanf("%d",&x);
			las=x;
		}
		else {
			scanf("%d",&x);
			k.op=2;k.d=las;k.x=x;
			++cnt_ans;k.ret=cnt_ans;
			++cnt;ak[cnt]=k;
		}
	}
	sort(ak+1,ak+cnt+1,cmp1);
	for(int i=1;i<=cnt;i++){
		if(ak[i].op==1){
			int now=ak[i].x;//printf("%d:%d\n",i,now);
			while(node_top[now]!=1){
				add(1,dfn[node_top[now]],dfn[now]);
				now=fa[node_top[now]];
			}
			//printf("F");
			add(1,1,dfn[now]);
		}
		else {
			x=ak[i].x;
			ans[ak[i].ret]=ask(1,dfn[x]);
		}
	}
	for(int i=1;i<=cnt_ans;i++){
		printf("%d\n",ans[i]);
	}
	return 0;
}

2021/12/23 18:33
加载中...