1. 程式人生 > >luogu:3261 [JLOI2015]城池攻佔

luogu:3261 [JLOI2015]城池攻佔

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。

 

其實我感覺這題也不是特別難,就是在左偏樹上打標記就可以了233.

#include<bits/stdc++.h>
#define N 2000005
using namespace std;
struct edge{
    int v ,next;
}e[N];
int p[N], eid;
void init(){
    memset(p, -1, sizeof p);
    eid = 0;
}
void insert(int u, int v){
    e[eid].v = v;
    e[eid].next = p[u];
    p[u] = eid ++;
}
int dis[N], size[N], ch[N][2], rt[N];
long long add[N], mul[N], sum[N];
void pushdown(int rt){
    add[ch[rt][0]] = add[ch[rt][0]] * mul[rt] + add[rt];
    add[ch[rt][1]] = add[ch[rt][1]] * mul[rt] + add[rt];
    
    mul[ch[rt][0]] *= mul[rt];
    mul[ch[rt][1]] *= mul[rt];
    
    sum[ch[rt][0]] = sum[ch[rt][0]] * mul[rt] + add[rt];
    sum[ch[rt][1]] = sum[ch[rt][1]] * mul[rt] + add[rt];
    
    mul[rt] = 1; add[rt] = 0;
}
int merge(int x, int y){
    pushdown(x); pushdown(y);
    if(!x || !y) return x + y;
    if(sum[x] > sum[y]) swap(x, y);
    ch[x][1] = merge(ch[x][1], y);
    if(dis[ch[x][0]] < dis[ch[x][1]]) swap(ch[x][0], ch[x][1]);
    dis[x] = dis[ch[x][1]] + 1;
    return x;
}
int n, m, l[N], a[N], deep[N], ans1[N], ans2[N], c[N];
long long ans, vals[N], V[N], H[N], s[N];
int dfs(int u){ 
    int root = rt[u]; size[u] = 1;
    for(int i = p[u]; i + 1; i = e[i].next){
        int v = e[i].v;
        deep[v] = deep[u] + 1;
        int y = dfs(v);
        size[u] += size[v];
        root = merge(root, y);
    }    for(;pushdown(root), sum[root] < H[u] && root;) ans1[u] ++, ans2[root] = u, root = merge(ch[root][0], ch[root][1]), size[u] --;
    if(a[u]) add[root] *= V[u], mul[root] *= V[u], sum[root] *= V[u];
    else add[root] += V[u], sum[root] += V[u];
    return root;
}
int main(){
    init();
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++) scanf("%lld", &H[i]), mul[i] = 1;
    for(int i = 2; i <= n; i ++){
        int u;
        scanf("%d%d%lld", &u, &a[i], &V[i]);
        insert(u, i);
    }
    for(int i = 1; i <= m; i ++){
    	scanf("%lld%d", &sum[i], &c[i]);
    	rt[c[i]] = merge(rt[c[i]], i);
    }
    insert(0, 1);
    H[0] = 1e18 + 233;
    dfs(0);
    for(int i = 1; i <= n; i ++) printf("%d\n", ans1[i]);
    for(int i = 1; i <= m; i ++) printf("%d\n", deep[c[i]] - deep[ans2[i]]);
    return 0;
}

23333