BZOJ 4003([JLOI2015]城池攻占-带标记可合并堆)[Template:带标记可合并堆]

4003: [JLOI2015]城池攻占

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 490   Solved: 181
[ Submit][ Status][ Discuss]

Description

小铭铭最近获得了一副新的桌游，游戏中需要用 m 个骑士攻占 n 个城池。

这 n 个城池用 1 到 n 的整数表示。除 1 号城池外，城池 i 会受到另一座城池 fi 的管辖，

其中 fi <i。也就是说，所有城池构成了一棵有根树。这 m 个骑士用 1 到 m 的整数表示，其

中第 i 个骑士的初始战斗力为 si，第一个攻击的城池为 ci。

每个城池有一个防御值 hi，如果一个骑士的战斗力大于等于城池的生命值，那么骑士就可

以占领这座城池；否则占领失败，骑士将在这座城池牺牲。占领一个城池以后，骑士的战斗力

将发生变化，然后继续攻击管辖这座城池的城池，直到占领 1 号城池，或牺牲为止。

除 1 号城池外，每个城池 i 会给出一个战斗力变化参数 ai;vi。若 ai =0，攻占城池 i 以后骑士战斗力会增加 vi；若 ai =1，攻占城池 i 以后，战斗力会乘以 vi。注意每个骑士是单独计算的。也就是说一个骑士攻击一座城池，不管结果如何，均不会影响其他骑士攻击这座城池的结果。

现在的问题是，对于每个城池，输出有多少个骑士在这里牺牲；对于每个骑士，输出他攻占的城池数量。

Input

第 1 行包含两个正整数 n;m，表示城池的数量和骑士的数量。

第 2 行包含 n 个整数，其中第 i 个数为 hi，表示城池 i 的防御值。

第 3 到 n +1 行，每行包含三个整数。其中第 i +1 行的三个数为 fi;ai;vi，分别表示管辖

这座城池的城池编号和两个战斗力变化参数。

第 n +2 到 n + m +1 行，每行包含两个整数。其中第 n + i 行的两个数为 si;ci，分别表

示初始战斗力和第一个攻击的城池。

Output

 输出 n + m 行，每行包含一个非负整数。其中前 n 行分别表示在城池 1 到 n 牺牲的骑士

数量，后 m 行分别表示骑士 1 到 m 攻占的城池数量。

Sample Input

5 5
50 20 10 10 30
1 1 2
2 0 5
2 0 -10
1 0 10
20 2
10 3
40 4
20 4
35 5

Sample Output

2
2
0
0
0
1
1
3
1
1

HINT

 对于 100% 的数据，1 <= n;m <= 300000; 1 <= fi<i; 1 <= ci <= n; -10^18 <= hi,vi,si <= 10^18;ai等于1或者2；当 ai =1 时，vi > 0；保证任何时候骑士战斗力值的绝对值不超过 10^18。

Source

可合并堆，加上标记加法，乘法，DP

注意300000点链可能爆栈，RE的请使用BFS或人工栈，或细心优化

评测是在Linux下进行的，栈会比默认的windows的栈大

使用DEV的，可以在‘编译选项’中加入‘-Wl,--stack=10240000’

#pragma comment(linker, "/STACK:102400000000,102400000000") 
#include<bits/stdc++.h> 
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define Forpiter(x) for(int &p=iter[x];p;p=next[p])  
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,127,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define MAXN (300000+10)
#define MAXM (300000+10)
typedef long long ll;

class mergeable_heap
{
public:
    ll val[MAXN];  
    int dis[MAXN],ch[MAXN][2];  
	ll addv[MAXN],mulv[MAXN];
	bool is_mul[MAXN];
	void mem()
	{
		MEM(val) MEM(dis) MEM(ch) 
		dis[0]=-1;
		MEM(addv) MEM(mulv) MEM(is_mul)
	}
	void pushdown(int x)
	{
		if (x==0) return;
		if (is_mul[x])
		{
			if (ch[x][0]) 
			{
				mulv[ch[x][0]]=is_mul[ch[x][0]] ? mulv[ch[x][0]]*mulv[x] : mulv[x];
				is_mul[ch[x][0]]=1;
				addv[ch[x][0]]*=mulv[x],val[ch[x][0]]*=mulv[x];
			}
			if (ch[x][1]) 
			{
				mulv[ch[x][1]]=is_mul[ch[x][1]] ? mulv[ch[x][1]]*mulv[x] : mulv[x];
				is_mul[ch[x][1]]=1;
				addv[ch[x][1]]*=mulv[x],val[ch[x][1]]*=mulv[x];
			}
			is_mul[x]=mulv[x]=0;
		}
		if (addv[x])
		{
			if (ch[x][0]) addv[ch[x][0]]+=addv[x],val[ch[x][0]]+=addv[x];
			if (ch[x][1]) addv[ch[x][1]]+=addv[x],val[ch[x][1]]+=addv[x];
			addv[x]=0;
		}
	}
	int merge(int k1,int k2)
	{ 
		if (k1==0) return k2;
		if (k2==0) return k1;
		if (val[k1]>val[k2]) swap(k1,k2);
		pushdown(k1);
		ch[k1][1]=merge(ch[k1][1],k2);
		if (dis[ch[k1][0]]<dis[ch[k1][1]]) swap(ch[k1][0],ch[k1][1]);
		dis[k1]=dis[ch[k1][1]]+1;
		return k1;
	}
	int DeleteMin(int x)
	{
		if (x==0) return 0;
		pushdown(x); 
		return merge(ch[x][0],ch[x][1]);
		
	}
	void add(int x,ll v)
	{
		if (x==0) return;
		pushdown(x);
		addv[x]+=v;val[x]+=v;
	}
	void mul(int x,ll v)
	{
		if (x==0) return;
		pushdown(x); 
		is_mul[x]=1;mulv[x]=v;addv[x]*=v;val[x]*=v; 
	}
}S;

int n,m,f[MAXN]={0},a[MAXN]={0},c[MAXM]={0};
ll h[MAXN]={0},v[MAXN]={0};
int root[MAXN]={0};
int ans1[MAXN]={0},ans2[MAXM]={0};
int depth[MAXN]={0};


int edge[MAXN]={0},pre[MAXN]={0},next[MAXN]={0},siz=1;
void addedge(int u,int v)
{
	edge[++siz]=v;
	next[siz]=pre[u];
	pre[u]=siz;
}
#define V edge[p]
void work(int x)
{
	Forp(x)
	{
		root[x]=S.merge(root[x],root[V]);
	}
	while (root[x]&&S.val[root[x]]<h[x])
	{
		  ans2[root[x]]=depth[c[root[x]]]-depth[x];
		  root[x]=S.DeleteMin(root[x]); 
		  ans1[x]++;
	}
	if (x>1 && root[x])
		if (a[x]==0) S.add(root[x],v[x]);
		else S.mul(root[x],v[x]);
}
#undef V

int q[MAXN],head=1,tail=0;
void bfs(){
	q[++tail]=1;
	while (head<=tail) {
		int x=q[head++];
		Forp(x) {
			int V=edge[p];
			depth[V]=depth[x]+1;
			q[++tail]=V;
		}
	}
	
	ForD(i,n) work(q[i]);	
	
}


int main()
{
//	freopen("4003/fall2.in","r",stdin);
//	freopen("bzoj4003.out","w",stdout);
	MEM(root) MEM(c) MEM(pre)
	
	S.mem();
	 
	
	scanf("%d%d",&n,&m);
	For(i,n) scanf("%lld",&h[i]);
	Fork(i,2,n) {
		scanf("%d%d%lld",&f[i],&a[i],&v[i]);
		addedge(f[i],i);
	}
	f[1]=a[1]=v[1]=0;
		
			
	For(i,m)
	{
		scanf("%lld%d",&S.val[i],&c[i]);
		root[c[i]]=S.merge(root[c[i]],i);
	}
	depth[1]=1;
	bfs();
	
		
	while (root[1])
	{
		ans2[root[1]]=depth[c[root[1]]];
		root[1]=S.DeleteMin(root[1]); 	  
	} 
	
	For(i,n) printf("%d\n",ans1[i]);
	For(i,m) printf("%d\n",ans2[i]); 
	
	return 0;
}