BZOJ 4538: [Hnoi2016]网络

这个题目就是特别裸啊，很明显就是先树链剖分，然后在线段树每个节点上维护两个堆，来维护插入和删除，查询的时候就暴力查就好了。

似乎很简单啊，我竟然在luogu上1A了，结果交到bzoj上MLE，看过讨论后把找重儿子改成大于等于就A了。。

不过我感觉这样是假的啊，一条链在线段树上有\(\log^2n\)段，每个暴力插的话就是\(O(n\log^3n)\)啊，但是还跑的挺快的，这卡的也太不满了吧。。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define qmin(x,y) (x=min(x,y))
#define qmax(x,y) (y=max(x,y))
using namespace std;

typedef long long ll;
const int Maxn=410000;

int to[Maxn],nxt[Maxn],first[Maxn],tot=1;
int tl[Maxn],tr[Maxn],u[Maxn],v[Maxn],w[Maxn];
int dep[Maxn],fa[Maxn],ran[Maxn],top[Maxn],son[Maxn];
int n,m,x,y,opt,cnt;
priority_queue<int> t[Maxn][2];

struct node{
	int l,r;
}s[Maxn];

inline void add(int u,int v) {
	to[tot]=v;
	nxt[tot]=first[u];
	first[u]=tot++;
	to[tot]=u;
	nxt[tot]=first[v];
	first[v]=tot++;
}

int dfs1(int root) {
	int siz=1,sizm=0;
	for(int i=first[root];i;i=nxt[i])
		if(!dep[to[i]]) {
			dep[to[i]]=dep[root]+1;
			fa[to[i]]=root;
			int temp=dfs1(to[i]);
			siz+=temp;
			if(temp>=sizm) {
				sizm=temp;
				son[root]=to[i];
			}
		}
	return siz;
}

void dfs2(int root,int topp) {
	top[root]=topp;ran[root]=++cnt;
	if(son[root]) dfs2(son[root],topp);
	for(int i=first[root];i;i=nxt[i])
		if(!ran[to[i]]) dfs2(to[i],to[i]);
}

void build(int root,int l,int r) {
	tl[root]=l,tr[root]=r;
	int mid=l+r>>1;
	if(l==r) return;
	build(root<<1,l,mid);
	build((root<<1)|1,mid+1,r);
}

int gettop(int root) {
	while(!t[root][1].empty()&&t[root][0].top()==t[root][1].top()) t[root][1].pop(),t[root][0].pop();
	return t[root][0].empty()?-1:t[root][0].top();
}

int query(int root,int x) {
	int l=tl[root],r=tr[root];
	int mid=l+r>>1;
	int ans=gettop(root);
	if(l==r) return ans;
	if(x<=mid) return max(ans,query(root<<1,x));
	else return max(ans,query((root<<1)|1,x));
}

void add(int root,int l,int r,int opt,int x) {
	int lc=tl[root],rc=tr[root];
	int mid=lc+rc>>1;
	if(l<=lc&&r>=rc) {
		t[root][opt].push(x);
		return ;
	}
	if(l<=mid) add(root<<1,l,r,opt,x);
	if(r>mid) add((root<<1)|1,l,r,opt,x);
}

int cmp(node a,node b) {
	return a.l<b.l;
}

void change(int x,int y,int w,int opt) {
	int topp=0;
	while(top[x]!=top[y]) {
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		s[++topp]=(node){ran[top[x]],ran[x]};
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	s[++topp]=(node){ran[x],ran[y]};
	sort(s+1,s+topp+1,cmp);
	int zhy=0;
	for(int i=1;i<=topp;zhy=s[i++].r)
		if(zhy+1<s[i].l) add(1,zhy+1,s[i].l-1,opt,w);
	if(zhy<n) add(1,zhy+1,n,opt,w);
}

int main() {
//	freopen("test.in","r",stdin);
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++) {
		scanf("%d%d",&x,&y);
		add(x,y);
	}dep[1]=1;
	dfs1(1),dfs2(1,1);
	build(1,1,n);
	for(int i=1;i<=m;i++) {
		scanf("%d",&opt);
		if(opt==2) {
			scanf("%d",&x);
			printf("%d\n",query(1,ran[x]));
		}
		else if(opt==1) {
			scanf("%d",&x);
			change(u[x],v[x],w[x],opt);
		}
		else {
			scanf("%d%d%d",&u[i],&v[i],&w[i]);
			change(u[i],v[i],w[i],opt);
		}
	}
	return 0;
}