luogu P3261 [JLOI2015]城池攻占

背景：

补坑$\text{ing...}$

题目传送门：

https://www.luogu.org/problem/P3261

题意：

城池数为$n$的树，有$m$个骑士，每一个骑士会往根$1$走，每一次经过一个城池，若大于城池生命值会占领，且战斗力变化；否则战死于该城池。求每个骑士攻占的城池的数量以及每个城池抵挡的骑士的数量。

思路：

考虑骑士只会战死一次。
因此不妨用左偏树维护一个小根堆，对于每一个城池，暴力合并儿子的骑士的信息，若当前堆顶小于该城池的战斗力，则$\text{pop}$掉，表示该骑士战死于当前的城池。
考虑每一个战斗力的修改， 发现当前活着的骑士的相对大小不会改变（同时加或同时乘上一个正数），因此直接在堆顶$root$上打上一个$\text{lazy}$标记即可。
由于打错了$\text{lazy}$标记的一个小地方，使得我的代码调了大半天。
代码中的$roo{t}_x$表示在第$x$个城池时的堆顶（战斗力最小的骑士），你可以用并查集来维护，当然，我在合并和删除时顺便记录了堆顶，因此就直接维护了。

代码：

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#define LL long long
using namespace std;
vector<int> F[600010];
	int n,m;
	LL val[600010],lazy1[600010],lazy2[600010];//1:*,2:+
	int st[600010],ed[600010],dep[600010],dis[600010],ls[600010],rs[600010],root[600010];
	struct node{LL x;int fa,op,tot;LL y;} b[600010];
void update(int x,LL cheng,LL jia)
{
	if(!x) return;
	val[x]=val[x]*cheng+jia;
	lazy1[x]*=cheng;
	lazy2[x]=lazy2[x]*cheng+jia;
}
void pushdown(int x)
{
	update(ls[x],lazy1[x],lazy2[x]);
	update(rs[x],lazy1[x],lazy2[x]);
	lazy1[x]=1,lazy2[x]=0;
}
int Merge(int x,int y)
{
	if(!x) return y;
	if(!y) return x;
	pushdown(x),pushdown(y);
	if(val[x]>val[y]) swap(x,y);
	rs[x]=Merge(rs[x],y);
	if(dis[ls[x]]<dis[rs[x]]) swap(ls[x],rs[x]);
	dis[x]=dis[rs[x]]+1;
	return x;
}
int del(int x)
{
	int lc=ls[x],rc=rs[x];
	pushdown(x);
	val[x]=0;
	dis[x]=ls[x]=rs[x]=0;
	return Merge(lc,rc);
}
void work(int x,int y)
{
	if(!root[y]) return;
	if(!root[x]) root[x]=root[y]; else root[x]=Merge(root[x],root[y]);
	while(root[x]&&val[root[x]]<b[x].x)
		ed[root[x]]=x,root[x]=del(root[x]),b[x].tot++;
}
void dfs(int x)
{
	for(int i=0;i<F[x].size();i++)
	{
		int y=F[x][i];
		dep[y]=dep[x]+1;
		dfs(y);
		work(x,y);
	}
	if(!root[x]) return;
	if(!b[x].op) update(root[x],1,b[x].y); else update(root[x],b[x].y,0);
}
int main()
{
	int x;
	LL y;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=m;i++)
		lazy1[i]=1;
	for(int i=1;i<=n;i++)
		scanf("%lld",&b[i].x);
	for(int i=2;i<=n;i++)
	{
		scanf("%d %d %lld",&b[i].fa,&b[i].op,&b[i].y);
		F[b[i].fa].push_back(i);
		b[i].tot=0;
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%lld %d",&y,&x);
		st[i]=x;
		if(y>=b[x].x)
		{
			val[i]=y;
			if(!root[x]) root[x]=i; else root[x]=Merge(root[x],i);
		} else b[x].tot++,ed[i]=x;
	}
	dep[1]=1;
	dfs(1);
	for(int i=1;i<=n;i++)
		printf("%d\n",b[i].tot);
	for(int i=1;i<=m;i++)
		printf("%d\n",dep[st[i]]-dep[ed[i]]);
}