4003: [JLOI2015]城池攻占
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 490 Solved: 181
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; }
版权声明:本文为博主原创文章,未经博主允许不得转载。