bzoj2325: [ZJOI2011]道馆之战

题面在这里
题目大意：
这题题目太长了就简单说一下题意。
给一棵树，每个节点分为两个区域A,B，在树上一个节点的某个区域，可以走到相邻节点的同一个区域，或同一个节点的另一个区域。每个区域内有两种状态，’.’表示冰，’#’表示障碍（障碍不能通过），给一些查询u, v，询问从u走到v能经过的最多冰的数量。同时还有一些修改操作，将某一点两个区域修改为s（s是长度为2的字符串）。

做法：
这题花了我一半的脑细胞。。tm做一道zj省选题我容易么我= =本人已死有事烧纸
树剖+线段树。
线段树里要维护8个量。。
$d[o][0..4]$，0表示从左上到右上，1表示从左上到右下，2表示从左下到右上，3表示从左下到右下 的最大步数
$f[o][0..4]$，从 0表示左上，1表示左下，2表示右上，3表示右下 开始走的最远步数
合并的时候瞎搞下就行qwq
解决查询的时候，将一条链沿着lca左右分开分别计算答案，最后再把左右两边合并起来。
具体操作见代码。看了代码应该就明白要怎么写了。

/*************************************************************
    Problem: bzoj 2325 [ZJOI2011]道馆之战
    User: fengyuan
    Language: C++
    Result: Accepted
    Time: 2512 ms
    Memory: 8060 kb 
    Submit_Time: 2017-12-06 17:14:06
*************************************************************/

#include<bits/stdc++.h>
#define rep(i, x, y) for (int i = (x); i <= (y); i ++)
#define down(i, x, y) for (int i = (x); i >= (y); i --)
#define mid ((l+r)/2)
#define lc (o<<1)
#define rc (o<<1|1)
#define pb push_back
#define PII pair<int, int>
#define F first
#define S second
#define B begin()
#define E end()
using namespace std;
typedef long long LL;
//head

const int N = 30010;
const int INF = 1e9;
int n, m, cnt, clk;
int head[N], in[N], depth[N], fa[N], son[N], sz[N], top[N], q[N];
char ch[5], opt[5], mp[N][5];
//线段树d[o][0~4]，0表示从左上到右上，1表示从左上到右下，2表示从左下到右上，3表示从左下到右下 的最大步数
//线段树f[o][0~4]，从 0表示左上，1表示左下，2表示右上，3表示右下 开始走的最远步数 
struct segTree{
    int d[5], f[5];
    segTree(){
        memset(d, 0, sizeof d);
        memset(f, 0, sizeof f);
    }
}T[N<<2];
struct Edge{
    int to, nex;
}e[N<<1];

inline void add(int x, int y)
{
    e[++ cnt].to = y;
    e[cnt].nex = head[x];
    head[x] = cnt;
}

inline void dfs(int u, int last, int s)
{
    depth[u] = s; fa[u] = last; sz[u] = 1;
    for (int i = head[u]; i; i = e[i].nex){
        int v = e[i].to; if (v == last) continue;
        dfs(v, u, s+1); sz[u] += sz[v];
        if (!son[u] || sz[v] > sz[son[u]]) son[u] = v;
    }
}

inline void dfs2(int u, int t)
{
    top[u] = t; in[u] = ++ clk; q[clk] = u;
    if (son[u]) dfs2(son[u], t);
    for (int i = head[u]; i; i = e[i].nex){
        int v = e[i].to; if (v == fa[u] || v == son[u]) continue;
        dfs2(v, v);
    }
}

inline segTree merge(segTree x, segTree y)
{
    segTree ret;
    ret.d[0] = max(x.d[0]+y.d[0], x.d[1]+y.d[2]); if (ret.d[0] < 0) ret.d[0] = -INF;
    ret.d[1] = max(x.d[0]+y.d[1], x.d[1]+y.d[3]); if (ret.d[1] < 0) ret.d[1] = -INF;
    ret.d[2] = max(x.d[2]+y.d[0], x.d[3]+y.d[2]); if (ret.d[2] < 0) ret.d[2] = -INF;
    ret.d[3] = max(x.d[2]+y.d[1], x.d[3]+y.d[3]); if (ret.d[3] < 0) ret.d[3] = -INF;
    ret.f[0] = max(x.f[0], max(x.d[0]+y.f[0], x.d[1]+y.f[1])); if (ret.f[0] < 0) ret.f[0] = -INF;
    ret.f[1] = max(x.f[1], max(x.d[2]+y.f[0], x.d[3]+y.f[1])); if (ret.f[1] < 0) ret.f[1] = -INF;
    ret.f[2] = max(y.f[2], max(y.d[0]+x.f[2], y.d[2]+x.f[3])); if (ret.f[2] < 0) ret.f[2] = -INF;
    ret.f[3] = max(y.f[3], max(y.d[1]+x.f[2], y.d[3]+x.f[3])); if (ret.f[3] < 0) ret.f[3] = -INF;
    return ret;
}

inline void update(int o, int l, int r, int x)
{
    if (l == x && r == x){
        x = q[x];
        T[o].d[0] = T[o].d[1] = T[o].d[2] = T[o].d[3] = -INF;
        T[o].f[0] = T[o].f[1] = T[o].f[2] = T[o].f[3] = -INF;
        if (mp[x][0] == '.') T[o].d[0] = T[o].f[0] = T[o].f[2] = 1;
        if (mp[x][1] == '.') T[o].d[3] = T[o].f[1] = T[o].f[3] = 1;
        if (mp[x][0] == '.' && mp[x][1] == '.'){
            rep(i, 0, 3) T[o].f[i] = 2;
            T[o].d[1] = T[o].d[2] = 2;
        }
        return;
    }
    if (x <= mid) update(lc, l, mid, x);
    else update(rc, mid+1, r, x);
    T[o] = merge(T[lc], T[rc]);
}

inline segTree query(int o, int l, int r, int x, int y)
{
    if (l == x && r == y) return T[o];
    if (y <= mid) return query(lc, l, mid, x, y);
    else if (x > mid) return query(rc, mid+1, r, x, y);
    else return merge(query(lc, l, mid, x, mid), query(rc, mid+1, r, mid+1, y));
}

inline void solve(int x, int y)
{
    if (mp[x][0] == '#' && mp[x][1] == '#') { puts("0"); return; }
    segTree ans1, ans2;
    while (top[x] != top[y]){
        if (depth[top[x]] > depth[top[y]]) ans1 = merge(query(1, 1, n, in[top[x]], in[x]), ans1), x = fa[top[x]];
        else ans2 = merge(query(1, 1, n, in[top[y]], in[y]), ans2), y = fa[top[y]];
    }
    if (depth[x] < depth[y]) ans2 = merge(query(1, 1, n, in[x], in[y]), ans2);
    else ans1 = merge(query(1, 1, n, in[y], in[x]), ans1);
    swap(ans1.d[1], ans1.d[2]); swap(ans1.f[0], ans1.f[2]); swap(ans1.f[1], ans1.f[3]);
    ans1 = merge(ans1, ans2);
    printf("%d\n", max(ans1.f[0], ans1.f[1]));
}

int main()
{
    scanf("%d%d", &n, &m);
    rep(i, 1, n-1){
        int x, y; scanf("%d%d", &x, &y);
        add(x, y); add(y, x);
    }
    dfs(1, 0, 0); dfs2(1, 1);
    rep(i, 1, n){
        scanf("%s", mp[i]);
        update(1, 1, n, in[i]);
    }
    while (m --){
        scanf("%s", opt); int x, y;
        if (opt[0] == 'Q'){
            scanf("%d%d", &x, &y);
            solve(x, y);
        } else {
            scanf("%d", &x); scanf("%s", mp[x]);
            update(1, 1, n, in[x]);
        }
    }
    return 0;
}