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

来源:互联网 发布:知乎 旅游代购 吐槽 编辑:程序博客网 时间:2024/04/29 21:23

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 200010inline 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;}
0 0
原创粉丝点击