BZOJ3924——[Zjoi2015]幻想乡战略游戏

0、题意:动态维护带权中心
1、分析:妈的,这题做了一天,mdzzzzzzzzzzzzzzzzzz…..
这个题是边权,我们首先要将边权转化成点权。。。
我们维护一个分支结构中到根的距离和,一个分支结构中到分支中心的距离和,这个分支结构中所有的军队总量,lca…,lca这个预处理出来,因为要求树上两点距离
询问到一个点的距离时(下面称query值),我们在 logn 个分治结构中询问答案
到x点的答案=x本身子树到这个点答案,然后我们向上跳
每次跳我们将答案+=father[x]的除了x这个子树的点到father[x]的值+这些点的value*father[x]到x的值。。口胡了一番。。具体看代码吧
修改的时候,也是修改 logn 个分支结构,我们依旧是随便的修改一下就可以了
然后在真正的询问怎么办呢?我们 logn 的搞一下,怎么搞呢?我们贪心
如果这个子树中的根的query值<这个根的query,那么我们就一下子跳到这个子树中的分支中心去,可以想象,我们要找的点必然在这个子树中。
我们最多会跳 logn 次,每次我们查询query值是 O(logn) 的,所以询问的时间复杂度是 O(log2n) ,修改的时间复杂度是 O(logn) ,总的复杂度 O(nlog2n)
代码改了一天QAQ

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
#define LL long long
#define M 200010

inline int read(){
    char ch = getchar(); int x = 0, f = 1;
    while(ch < '0' || ch > '9'){
        if(ch == '-') f = -1;
        ch = getchar();
    }
    while('0' <= ch && ch <= '9'){
        x = x * 10 + ch - '0';
        ch = getchar();
    }
    return x * f;
}

struct Edge{
    int u, v;
    LL w;
    int next;
} G[M];
int head[M], tot;
int size[M], mx[M], mi, root;
int ok[M];
int father[M];
LL fadis[M], dis;
LL ft[M];
LL s1[M], s2[M], val[M];
int fenzhi[M][20], ding[M];
int vis[M];
int n, Q;
int zz[M], ztot;
int ftt[M][20];
LL ff[M][20];
int height[M];
LL dep[M];
int pa[M];

inline void add(int u, int v, LL w){
    G[++ tot] = (Edge){u, v, w, head[u]};
    head[u] = tot;
}

inline void dfs(int x, int fa, int h, LL gh){
    height[x] = h;
    dep[x] = gh;
    ftt[x][0] = fa;
    for(int i = head[x]; i != -1; i = G[i].next) if(G[i].v != fa){
        dfs(G[i].v, x, h + 1, gh + G[i].w);
    }
}

inline int lca(int x, int y){
    if(height[x] < height[y]) swap(x, y);
    int t = height[x] - height[y];
    for(int i = 18; i >= 0; i --){
        if(t & (1 << i)){
            x = ftt[x][i];
        }
    }
    if(x == y) return x;
    for(int i = 18; i >= 0; i --){
        if(ftt[x][i] != ftt[y][i]){ 
            x = ftt[x][i];
            y = ftt[y][i];
        }
    }
    return ftt[x][0];
}

inline void init(){
    for(int i = 1; i <= 18; i ++){
        for(int j = 1; j <= n; j ++){
            ftt[j][i] = ftt[ftt[j][i - 1]][i - 1];
        }
    }
    for(int i = 1; i <= n; i ++){
        int o = i, cnt = 0;
        while(o){
            int LCA = lca(o, i);
            LL od = dep[i] + dep[o] - 2 * dep[LCA];
            ff[i][cnt] = od;
            o = father[o];
            cnt ++;
        } 
    }
}

inline void dfssize(int x, int fa){
    size[x] = 1; mx[x] = 0;
    for(int i = head[x]; i != -1; i = G[i].next) if(G[i].v != fa && !ok[G[i].v]){
        dfssize(G[i].v, x);
        size[x] += size[G[i].v];
        if(size[G[i].v] > mx[x]) mx[x] = size[G[i].v];
    }
}

inline int getroot(int r, int x, int fa, LL d){
    mx[x] = max(mx[x], size[r] - size[x]);
    if(mx[x] < mi) mi = mx[x], root = x, dis = d;
    for(int i = head[x]; i != -1; i = G[i].next) if(G[i].v != fa && !ok[G[i].v]){
        getroot(r, G[i].v, x, d + G[i].w);
    }

}

inline int solve(int x, int fa, LL d, int otot){
    mi = n; dfssize(x, 0);
    getroot(x, x, 0, 0);
    ok[root] = 1;
    father[root] = fa;
    int op = root;
    dis += d;
    fadis[root] = dis;
    LL tt = dis;
    for(int i = head[root]; i != -1; i = G[i].next) if(!ok[G[i].v]){
        int wt = solve(G[i].v, op, G[i].w, otot + 1);
        ft[wt] = G[i].w;
        fenzhi[G[i].v][otot] = wt;
        ding[wt] = G[i].v;
    }
    return op;
}

inline LL query(int x){
    LL ret = s1[x];
    int o = x;
    int cnt = 1;
    while(father[o]){
        ret += (s1[father[o]] - (s2[o] + ft[o] * val[o]) + (val[father[o]] - val[o]) * ff[x][cnt]);
        o = father[o];
        cnt ++;
    }
    return ret;
}

inline int Query(int rt, int fa, int otot){
    LL yy = query(rt);
    for(int i = head[rt]; i != -1; i = G[i].next) if(fenzhi[G[i].v][otot] != fa){
        LL y = query(G[i].v);
        if(yy > y) return Query(fenzhi[G[i].v][otot], rt, otot + 1);
    }
    int x = rt;
    return x;
}

inline int change(int x, int y, int rt){
    int o = father[x];
    int cnt = 1;
    val[x] += y;
    s2[x] += (ff[x][cnt] - ft[x]) * y;
    while(o){
        s1[o] += (ff[x][cnt]) * y;
        s2[o] += (ff[x][cnt + 1] - ft[o]) * y;
        val[o] += y;
        o = father[o];
        cnt ++;
    }
    int ret = Query(rt, 0, 0);
    return ret;
}

int main(){
    n = read(), Q = read();
    memset(head, -1, sizeof(head));
    for(int i = 1; i < n; i ++){
        int u = read(), v = read();
        LL w = (LL)read();
        add(u, v, w); add(v, u, w);
    }
    dfs(1, 0, 0, 0);
    int rt = solve(1, 0, 0, 0); fadis[rt] = 0;
    init();
    for(int i = 1; i <= Q; i ++){
        int u = read();
        LL e = read();
        printf("%lld\n", query(change(u, e, rt))); 
    }
    return 0;
}

你可能感兴趣的:(bzoj,树分治,ZJOI2015,幻想乡战略游戏,3924)