BZOJ4316 小C的独立集(圆方树dp)

题目链接

思路：
$\ \ \ \ $学习圆方树中。。。。代码很烂。就是类似树型dp，不过这里有环，如果是树边直接树形$dp$处理，否则求环上的最大独立集，类似基环树。更新的时候用方点去更新圆点的$dp$值。

#include<bits/stdc++.h>
typedef long long ll;
const int maxn = 1e5 + 10;
const int INF = 1e9 + 7;
using namespace std;

typedef pair<int, int> edge;
int n, m, T, kase = 1;
vector<int> G[maxn], bcc[maxn];
int dp[maxn][2], dfn[maxn], dfs_clock, tot;
int pre[maxn], bccno[maxn], bcc_cnt, stk[maxn][2], st[maxn][2];
stack<edge> S;
vector<int> et[maxn];

int dfs(int u, int fa) {
    int lowu = pre[u] = ++dfs_clock, child = 0;
    for(int i = 0; i < G[u].size(); i++) {
        int v = G[u][i];
        edge e = edge(u, v);
        if(!pre[v]) {
            S.push(e); child++;
            int lowv = dfs(v, u);
            lowu = min(lowu, lowv);
            if(lowv >= pre[u]) {
                bcc_cnt++; bcc[bcc_cnt].clear();
                while(1) {
                    edge x = S.top(); S.pop();
                    if(bccno[x.second] != bcc_cnt) { bcc[bcc_cnt].push_back(x.second); bccno[x.second] = bcc_cnt; } 
                    if(bccno[x.first] != bcc_cnt) { bcc[bcc_cnt].push_back(x.first); bccno[x.first] = bcc_cnt; }
                    if(x.first == u && x.second == v) break;
                }
                if(bcc[bcc_cnt].size() == 2) {
                    et[u].push_back(v);  //树边
                    et[v].push_back(u);
                } else {
                    //环边
                    tot++;
                    for(int i = 0; i < bcc[bcc_cnt].size(); i++) {
                        int v = bcc[bcc_cnt][i];
                        et[v].push_back(tot);
                        et[tot].push_back(v);
                    }
                }
            }
        } else if(pre[v] < pre[u] && v != fa) {
            S.push(e); lowu = min(lowu, pre[v]);
        }
    }
    return lowu;
}

void solve_dp(int x, int fa) {
    if(x <= n) { //圆点
        dp[x][0] = 0; dp[x][1] = 1;
        for(int i = 0; i < et[x].size(); i++) {
            int v = et[x][i];
            if(v == fa) continue;
            solve_dp(v, x);
            if(v <= n) dp[x][0] += max(dp[v][0], dp[v][1]), dp[x][1] += dp[v][0];
        }
    } else { //方点
        for(int i = 0; i < et[x].size(); i++) if(et[x][i] != fa) solve_dp(et[x][i], x);
        int top = 0;
        for(int i = 0; i < et[x].size(); i++) {
            int v = et[x][i];
            stk[++top][0] = dp[v][0];
            stk[top][1] = dp[v][1];
            st[top][0] = dp[v][0];
            st[top][1] = dp[v][1];
        }
        st[top][1] = -INF;
        for(int i = top - 1; i; i--) {
            stk[i][0] += max(stk[i + 1][0], stk[i + 1][1]);
            stk[i][1] += stk[i + 1][0];
            st[i][0] += max(st[i + 1][0], st[i + 1][1]);
            st[i][1] += st[i + 1][0];
        }
        dp[fa][0] = stk[1][0];
        dp[fa][1] = st[1][1];
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin >> n >> m;
    for(int i = 1; i <= m; i++) {
        int u, v; cin >> u >> v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    int ans = 0; tot = n;
    for(int i = 1; i <= n; i++) if(!pre[i]) {
        dfs(i, -1); solve_dp(i, -1);
        ans += max(dp[i][0], dp[i][1]);
    }
    cout << ans << endl;
    return 0;
}