hdu 5438 Ponds(线段树+dfs)

题意:

有n个点,m条边,每个点都有权值。
然后开始删除点,删点规则是如果这个点连接其他点的数量小于2,那么这个点可以删除。
如果删除了某些点产生了其他可删除的点也将其删除。
直到删除到不能为止。问最后剩余的联通块中,点的数量是奇数的联通块中的点的权值和是多少。

解析:

先算出所有点的度数,然后把这些度数存入线段树中,每次询问线段树找到等于最小的度数的节点,把和该最小度数的节点相连的所有节点的度数-1,直到不能再删除为止。
线段树的作用就是加快查询最小度数的速度,以及加快查询等于最小度数节点的速度。
然后对剩下的节点进行dfs,找到奇数的联通块,把其权值加到答案中。
PS:看了题解,其实这题是拓扑排序,但是线段树做多了,看什么都像线段树了……

my code

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#define ls (o<<1)
#define rs (o<<1|1)
#define lson ls, L, M
#define rson rs, M+1, R
#define MID (L + R) >> 1
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const int N = 1e4 + 10;

int n, m;
vector<int> G[N];
ll a[N], minv[N << 2];
int deg[N];
bool vis[N];

void init() {
    memset(vis, false, sizeof(vis));
    memset(deg, 0, sizeof(deg));
    for(int i = 1; i <= n; i++)
        G[i].clear();
}

inline void addEdge(int u, int v) {
    G[u].push_back(v);
}

inline void pushUp(int o) {
    minv[o] = min(minv[ls], minv[rs]);
}

void build(int o, int L, int R) {
    if(L == R) {
        minv[o] = deg[L];
        return ;
    }
    int M = MID;
    build(lson);
    build(rson);
    pushUp(o);
}

int query(int o, int L, int R, int val) {
    if(L == R) {
        minv[o] = INF;
        return L;
    }
    int M = MID, ret = 0;
    if(val == minv[ls]) ret = query(lson, val);
    else ret = query(rson, val);
    pushUp(o);
    return ret;
}

void remove(int o, int L, int R, int pos) {
    if(L == R) {
        minv[o]--;
        if(minv[o] < 0) minv[o] = INF;
        return ;
    }
    int M = MID;
    if(pos <= M) remove(lson, pos);
    else remove(rson, pos);
    pushUp(o);
}

void dfs(int u, ll &sum, int &tot) {
    tot++;
    sum += a[u];
    for(int i = 0; i < G[u].size(); i++) {
        int v = G[u][i];
        if(vis[v]) continue;
        vis[v] = true;
        dfs(v, sum, tot);
    }
}

ll cal() {
    int tot;
    ll sum, ret = 0;
    for(int i = 1; i <= n; i++) {
        if(vis[i]) continue;
        tot = sum = 0;
        vis[i] = true;
        dfs(i, sum, tot);
        if(tot & 1) ret += sum;
    }
    return ret;
}

int main() {
    int u, v;
    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d", &n, &m);
        init();
        for(int i = 1; i <= n; i++) {
            scanf("%lld", &a[i]);
        }

        for(int i = 1; i <= m; i++) {
            scanf("%d%d", &u, &v);
            deg[u]++, deg[v]++;
            addEdge(u, v);
            addEdge(v, u);
        }

        build(1, 1, n);
        while(minv[1] <= 1) {
            u = query(1, 1, n, minv[1]);
            vis[u] = true;
            for(int i = 0; i < G[u].size(); i++) {
                int v = G[u][i];
                if(vis[v]) continue;
                remove(1, 1, n, v);
            }
        }

        printf("%lld\n", cal());
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值