背景:
补坑
ing...
\text{ing...}
ing...
题目传送门:
https://www.luogu.org/problem/P3261
题意:
城池数为
n
n
n的树,有
m
m
m个骑士,每一个骑士会往根
1
1
1走,每一次经过一个城池,若大于城池生命值会占领,且战斗力变化;否则战死于该城池。求每个骑士攻占的城池的数量以及每个城池抵挡的骑士的数量。
思路:
考虑骑士只会战死一次。
因此不妨用左偏树维护一个小根堆,对于每一个城池,暴力合并儿子的骑士的信息,若当前堆顶小于该城池的战斗力,则
pop
\text{pop}
pop掉,表示该骑士战死于当前的城池。
考虑每一个战斗力的修改, 发现当前活着的骑士的相对大小不会改变(同时加或同时乘上一个正数),因此直接在堆顶
r
o
o
t
root
root上打上一个
lazy
\text{lazy}
lazy标记即可。
由于打错了
lazy
\text{lazy}
lazy标记的一个小地方,使得我的代码调了大半天。
代码中的
r
o
o
t
x
root_x
rootx表示在第
x
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]]);
}