[bzoj4003][JLOI2015]城池攻占_左偏树

城池攻占 bzoj-4003 JLOI-2015

　　　　题目大意：一颗n个节点的有根数，m个有初始战斗力的骑士都站在节点上。每一个节点有一个standard，如果这个骑士的战斗力超过了这个门槛，他就会根据城池的奖励增加自己的战斗力。具体地：每一个城池有一个flag和一个val，表示成功到达这个城市的骑士的战斗力会乘上val还是加上val。每个骑士都只会向根节点进攻。输出每个骑士败北的城市编号。如果这个骑士成功到达根节点，输出0。

　　　　注释：$1\le n,m \le 3\cdot 10^5$，$-10^{18}\le standard , val , attack \le 10^{18}$。保证每一个骑士的atk不大于longlong。

　　　　　　想法：GXZlegend可并堆例题。和上两道题类似地，我们对于每一个城池维护一个小根堆，元素是是当前节点为子树里的骑士到达这里的atk，我们自底向顶修改城墙，顺便将骑士向上推，以及标记的下传。我们在merge函数中完成pushdown的操作。特别地，在维护双标记的时候我默认是先乘后加，所以在pushdown的时候如果发现有乘法标记的话需要先将加法标记扩大，再下传。

　　　　最后，附上丑陋的代码... ...

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 300010
using namespace std;
typedef long long ll;
int head[N],to[N],type[N],next[N],cnt,root[N],l[N],r[N],d[N],deep[N],from[N],kill[N],atk[N];
ll val[N],h[N],key[N],tadd[N],tmul[N];
inline void add(int x,int y,int a,ll b)
{
	to[++cnt]=y;
	type[cnt]=a;
	val[cnt]=b;
	next[cnt]=head[x];
	head[x]=cnt;
}
inline void pushdown(int x)
{
	if(!x)return;
	if(tmul[x]!=1)
	{
		key[l[x]]*=tmul[x];tadd[l[x]]*=tmul[x];tmul[l[x]]*=tmul[x];
		key[r[x]]*=tmul[x];tadd[r[x]]*=tmul[x];tmul[r[x]]*=tmul[x];
		tmul[x]=1;
	}
	if(tadd[x])
	{
		key[l[x]]+=tadd[x];tadd[l[x]]+=tadd[x];
		key[r[x]]+=tadd[x];tadd[r[x]]+=tadd[x];
		tadd[x]=0;
	}
}
int merge(int x,int y)
{
	if(!x) return y;
	if(!y) return x;
	pushdown(x),pushdown(y);
	if(key[x]>key[y]) swap(x,y);
	r[x]=merge(r[x],y);
	if(d[l[x]]<d[r[x]]) swap(l[x],r[x]);
	d[x]=d[r[x]]+1;
	return x;
}
void dfs(int x)
{
	for(int i=head[x];i;i=next[i])
	{
		deep[to[i]]=deep[x]+1;
		dfs(to[i]);
		if(type[i])
		{
			key[root[to[i]]]*=val[i];
			tadd[root[to[i]]]*=val[i];
			tmul[root[to[i]]]*=val[i];
		}
		else
		{
			key[root[to[i]]]+=val[i];
			tadd[root[to[i]]]+=val[i];
		}
		root[x]=merge(root[x],root[to[i]]);
	}
	while(root[x]&&key[root[x]]<h[x])
	{
		kill[x]++,atk[root[x]]=x,pushdown(root[x]),root[x]=merge(l[root[x]],r[root[x]]);
	}
}
int main()
{
	int n,m,x,a;
	ll b;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%lld",&h[i]);
	for(int i=2;i<=n;i++) scanf("%d%d%lld",&x,&a,&b),add(x,i,a,b);
	for(int i=1;i<=m;i++)
	{
		tmul[i]=1,scanf("%lld%d",&key[i],&from[i]),root[from[i]]=merge(root[from[i]],i);
	}
	d[0]=-1,deep[1]=1;
	dfs(1);
	for(int i=1;i<=n;i++)
	{
		printf("%d\n",kill[i]);
	}
	for(int i=1;i<=m;i++)
	{
		printf("%d\n",deep[from[i]]-deep[atk[i]]);
	}
	return 0;
}

　　　　小结：type标记存在哪里都无所谓qwq