CF - 219D 线段树 + dfs序

题意：

n个城市之间有n-1条边相连，其中每条边都是有向的，现在要在其中选择一个城市作为首都，选择某个城市作为首都后，要保证首都可以到达任何城市，那么就要修改原来边的指向，问选首都最少修改几条边，另外要输出所有符合这个最小修改数的首都。

思路：

一道好题。如果从点的角度出发没什么思路，可以从边的角度考虑这个问题，对于某一条边(u,v)，以哪些城市作为首都需要修改这条(u,v)边呢？思考一下不难得出，因为这个图只有n-1条边，所以(u,v)就是桥，那么如果u->v，那么所有v指出城市作为首都，都需要修改这条边。如果遍历的时候也是u->v的顺序，那么v的子树要是作为首都就都需要修改这条边，也就是说这条边对v的子树的所有节点的贡献都是1，反之，就是除了v的子树以外的点都需要修改。
如果对一个子树进行统一的修改操作呢，很容易可以想到利用线段树+dfs序。

代码：

#include <bits/stdc++.h>
using namespace std;
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
const int MAXN = 2e5 + 10;
const int INF = 0x3f3f3f3f;

struct Edge {
    int u, v;
}edge[MAXN];

int dfs_cnt;
vector <int> tree[MAXN];
int Left[MAXN], Right[MAXN], fa[MAXN];
int Min[MAXN << 2], lazy[MAXN << 2];

void dfs(int u, int pre) {
    Left[u] = ++dfs_cnt;
    int cnt = tree[u].size();
    for (int i = 0; i < cnt; i++) {
        int v = tree[u][i];
        if (v == pre) continue;
        fa[v] = u;
        dfs(v, u);
    }
    Right[u] = dfs_cnt;
}

void push_up(int rt) {
    Min[rt] = min(Min[rt << 1], Min[rt << 1 | 1]);
}

void push_down(int rt) {
    if (lazy[rt] != 0) {
        lazy[rt << 1] += lazy[rt]; lazy[rt << 1 | 1] += lazy[rt];
        Min[rt << 1] += lazy[rt]; Min[rt << 1 | 1] += lazy[rt];
        lazy[rt] = 0;
    }
}

void update(int L, int R, int add, int l, int r, int rt) {
    if (L <= l && r <= R) {
        Min[rt] += add;
        lazy[rt] += add;
        return;
    }
    push_down(rt);
    int m = (l + r) >> 1;
    if (L <= m) update(L, R, add, lson);
    if (R > m) update(L, R, add, rson);
    push_up(rt);
}

int query(int L, int R, int l, int r, int rt) {
    if (L <= l && r <= R) return Min[rt];
    push_down(rt);
    int m = (l + r) >> 1, res = INF;
    if (L <= m) res = min(res, query(L, R, lson));
    if (R > m) res = min(res, query(L, R, rson));
    return res;
}

int main() {
    //freopen("in.txt", "r", stdin);
    int n;
    scanf("%d", &n);
    for (int i = 1; i < n; i++) {
        int u, v;
        scanf("%d%d", &u, &v);
        edge[i] = (Edge){u, v};
        tree[u].push_back(v);
        tree[v].push_back(u);
    }
    dfs_cnt = 0;
    fa[1] = -1;
    dfs(1, -1);
    for (int i = 1; i < n; i++) {
        int u = edge[i].u, v = edge[i].v;
        if (fa[v] == u) {
            update(Left[v], Right[v], 1, 1, n, 1);
        }
        else {
            update(1, Left[u] - 1, 1, 1, n, 1);
            update(Right[u] + 1, n, 1, 1, n, 1);
        }
    }
    int Minans = query(1, n, 1, n, 1);
    printf("%d\n", Minans);
    vector <int> ans;
    for (int i = 1; i <= n; i++) {
        //cout << i << " " << query(Left[i], Left[i], 1, n, 1) << endl;
        if (query(Left[i], Left[i], 1, n, 1) == Minans)
            ans.push_back(i);
    }
    int cnt = ans.size();
    for (int i = 0; i < cnt; i++)
        printf("%d%c", ans[i], i == cnt - 1 ? '\n' : ' ');
    return 0;
}