【BZOJ4196】【NOI2015】软件包管理器（树链剖分，线段树）

Description

Linux用户和OSX用户一定对软件包管理器不会陌生。通过软件包管理器，你可以通过一行命令安装某一个软件包，然后软件包管理器会帮助你从软件源下载软件包，同时自动解决所有的依赖（即下载安装这个软件包的安装所依赖的其它软件包），完成所有的配置。Debian/Ubuntu使用的apt-get，Fedora/CentOS使用的yum，以及OSX下可用的homebrew都是优秀的软件包管理器。

你决定设计你自己的软件包管理器。不可避免地，你要解决软件包之间的依赖问题。如果软件包A依赖软件包B，那么安装软件包A以前，必须先安装软件包B。同时，如果想要卸载软件包B，则必须卸载软件包A。现在你已经获得了所有的软件包之间的依赖关系。而且，由于你之前的工作，除0号软件包以外，在你的管理器当中的软件包都会依赖一个且仅一个软件包，而0号软件包不依赖任何一个软件包。依赖关系不存在环（若有m(m≥2)个软件包A1,A2,A3,…,Am，其中A1依赖A2，A2依赖A3，A3依赖A4，……，Am−1依赖Am，而Am依赖A1，则称这m个软件包的依赖关系构成环），当然也不会有一个软件包依赖自己。
现在你要为你的软件包管理器写一个依赖解决程序。根据反馈，用户希望在安装和卸载某个软件包时，快速地知道这个操作实际上会改变多少个软件包的安装状态（即安装操作会安装多少个未安装的软件包，或卸载操作会卸载多少个已安装的软件包），你的任务就是实现这个部分。注意，安装一个已安装的软件包，或卸载一个未安装的软件包，都不会改变任何软件包的安装状态，即在此情况下，改变安装状态的软件包数为0。

Solution

（终于把NOI2015搞完了）

树链剖分傻逼题。

安装时检查到根的路径上有多少个0，卸载时检查子树中有多少个1，用一棵资瓷区间赋值的线段树维护即可。

Code

/************************************************
 * Au: Hany01
 * Date: Sep 5th, 2018
 * Prob: BZOJ4196 NOI2015 软件包管理器
 * Email: [email protected] & [email protected]
 * Inst: Yali High School
************************************************/

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef long double LD;
typedef pair<int, int> PII;
#define rep(i, j) for (register int i = 0, i##_end_ = (j); i < i##_end_; ++ i)
#define For(i, j, k) for (register int i = (j), i##_end_ = (k); i <= i##_end_; ++ i)
#define Fordown(i, j, k) for (register int i = (j), i##_end_ = (k); i >= i##_end_; -- i)
#define Set(a, b) memset(a, b, sizeof(a))
#define Cpy(a, b) memcpy(a, b, sizeof(a))
#define x first
#define y second
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define SZ(a) ((int)(a).size())
#define ALL(a) a.begin(), a.end()
#define INF (0x3f3f3f3f)
#define INF1 (2139062143)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define y1 wozenmezhemecaia

template <typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template <typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }

inline int read() {
    static int _, __; static char c_;
    for (_ = 0, __ = 1, c_ = getchar(); c_ < '0' || c_ > '9'; c_ = getchar()) if (c_ == '-') __ = -1;
    for ( ; c_ >= '0' && c_ <= '9'; c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
    return _ * __;
}

const int maxn = 1e5 + 5;

int n, v[maxn], nex[maxn], e, beg[maxn], sz[maxn], dep[maxn], son[maxn], top[maxn], fa[maxn], dfn[maxn], efn[maxn], clk;

inline void add(int uu, int vv) { v[++ e] = vv, nex[e] = beg[uu], beg[uu] = e; }

void DFS1(int u) {
    dep[u] = dep[fa[u]] + 1, sz[u] = 1;
    for (register int i = beg[u]; i; i = nex[i])
        if (v[i] != fa[u]) {
            DFS1(v[i]), sz[u] += sz[v[i]];
            if (sz[son[u]] < sz[v[i]]) son[u] = v[i];
        }
}

void DFS2(int u) {
    dfn[u] = ++ clk;
    if (son[u]) top[son[u]] = top[u], DFS2(son[u]);
    for (register int i = beg[u]; i; i = nex[i])
        if (v[i] != fa[u] && v[i] != son[u]) top[v[i]] = v[i], DFS2(v[i]);
    efn[u] = clk;
}

struct SegmentTree {
    int val[maxn << 2], setv[maxn << 2];
#define lc (t << 1)
#define rc (lc | 1)
#define mid ((l + r) >> 1)
    inline void pushdown(int t, int l, int r) {
        if (setv[t] != -1)
            val[lc] = setv[t] * (mid - l + 1), val[rc] = setv[t] * (r - mid),
            setv[lc] = setv[rc] = setv[t], setv[t] = -1;
    }
    inline void maintain(int t) { val[t] = val[lc] + val[rc]; }
    void update(int t, int l, int r, int x, int y, int v) {
        if (x <= l && r <= y) { setv[t] = v, val[t] = v * (r - l + 1); return; }
        pushdown(t, l, r);
        if (x <= mid) update(lc, l, mid, x, y, v);
        if (y >  mid) update(rc, mid + 1, r, x, y, v);
        maintain(t);
    }
    int query(int t, int l, int r, int x, int y) {
        if (x <= l && r <= y) return val[t];
        pushdown(t, l, r);
        if (y <= mid) return query(lc, l, mid, x, y);
        if (x >  mid) return query(rc, mid + 1, r, x, y);
        return query(lc, l, mid, x, y) + query(rc, mid + 1, r, x, y);
    }
}ST;

inline int Solve1(int u) {
    static int ans, t;
    for (ans = 0, t = u; top[u] != 1; ) ans += ST.query(1, 1, n, dfn[top[u]], dfn[u]), ST.update(1, 1, n, dfn[top[u]], dfn[u], 1), u = fa[top[u]];
    ans += ST.query(1, 1, n, 1, dfn[u]), ST.update(1, 1, n, 1, dfn[u], 1);
    return dep[t] - ans;
}

inline int Solve2(int u) {
    static int ans;
    ans = ST.query(1, 1, n, dfn[u], efn[u]), ST.update(1, 1, n, dfn[u], efn[u], 0);
    return ans;
}

int main()
{
#ifdef hany01
    freopen("bzoj4196.in", "r", stdin);
    freopen("bzoj4196.out", "w", stdout);
#endif

    static char ty[11];

    For(i, 2, n = read()) fa[i] = read() + 1, add(fa[i], i);
    DFS1(1), DFS2(top[1] = 1);
    for (static int T = read(); T --; ) {
        scanf("%s", ty);
        if (ty[0] ^ 'u') printf("%d\n", Solve1(read() + 1));
        else printf("%d\n", Solve2(read() + 1));
    }

    return 0;
}