【CF786B】Legacy

题目大意:初始给定 N 个点,支持三种操作:两点之间连边;一个点与一个连续区间编号的点之间连边;一个连续区间内的点和一个点连边,求执行 N 次操作之后的单源最短路。

题解:学会了线段树优化建图。
发现若暴力进行连边,时间和空间都会被卡到 \(O(n^2)\),直接起飞。
发现连边的点的编号是连续的,结合线段树可以维护连续区间信息的思想,就产生了线段树优化建图的方法。
在初始的 N 个节点的基础上建立两棵线段树,分别表示入树和出树,其中入树的上的各个节点允许单个节点的连接;出树上的节点允许连接单个节点。对于入树中的每个节点来说,需要连一条边权为 0 的有向边到它的两个儿子,表示若有节点连向父节点,那么一定也连向了儿子节点;对于出树上的每个节点来说,需要连一条边权为 0 的有向边到它的父节点,表示若当前节点可以连向某个节点,那么其子节点也一定可以走到这个节点。
时空复杂度分析:空间为两棵线段树的空间减去一份叶子节点的大小,即:\(3 * n\),时间复杂度为 \(elog^2v\),其中 e 是边的条数,v 是节点的个数。

代码如下

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

struct edge {
    int to;
    LL w;
    edge(int x = -1, int y = -1) {
        to = x, w = y;
    }
};

struct segtree {
    #define ls(o) tree[o].lc
    #define rs(o) tree[o].rc
    struct node {
        int lc, rc;
    };
    vector<node> tree;
    int tot, rootin, rootout;
    segtree(int n, vector<vector<edge>> &adj) {
        tot = n;
        tree.resize(3 * n);
        buildin(rootin, 1, n, adj);
        buildout(rootout, 1, n, adj);
    }
    void buildin(int &o, int l, int r, vector<vector<edge>> &adj) {
        if (l == r) {
            o = l;
            return;
        }
        o = ++tot;
        int mid = l + r >> 1;
        buildin(ls(o), l, mid, adj);
        buildin(rs(o), mid + 1, r, adj);
        adj[o].emplace_back(ls(o), 0);
        adj[o].emplace_back(rs(o), 0);
    }
    void buildout(int &o, int l, int r, vector<vector<edge>> &adj) {
        if (l == r) {
            o = l;
            return;
        }
        o = ++tot;
        int mid = l + r >> 1;
        buildout(ls(o), l, mid, adj);
        buildout(rs(o), mid + 1, r, adj);
        adj[ls(o)].emplace_back(o, 0);
        adj[rs(o)].emplace_back(o, 0);
    }
    void link(int o, int l, int r, int x, int y, int u, int w, int opt, vector<vector<edge>> &adj) {// opt 2 -> in  3 -> out
        if (l == x && r == y) {
            opt == 2 ? adj[u].emplace_back(o, w) : adj[o].emplace_back(u, w);
            return;
        }
        int mid = l + r >> 1;
        if (y <= mid) {
            link(ls(o), l, mid, x, y, u, w, opt, adj);
        } else if (x > mid) {
            link(rs(o), mid + 1, r, x, y, u, w, opt, adj);
        } else {
            link(ls(o), l, mid, x, mid, u, w, opt, adj);
            link(rs(o), mid + 1, r, mid + 1, y, u, w, opt, adj);
        }
    }
};

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    int n, m, s;
    cin >> n >> m >> s;
    vector<vector<edge>> adj(3 * n);
    segtree t(n, adj);
    while (m--) {
        int opt;
        cin >> opt;
        if (opt == 1) {
            int u, v, w;
            cin >> u >> v >> w;
            adj[u].emplace_back(v, w);
        } else {
            int u, l, r, w;
            cin >> u >> l >> r >> w;
            t.link(opt == 2 ? t.rootin : t.rootout, 1, n, l, r, u, w, opt, adj);
        }
    }
    vector<LL> d(3 * n, 1e18);
    vector<bool> expand(3 * n);
    priority_queue<pair<LL, int>> q;
    auto dijkstra = [&]() {
        d[s] = 0, q.push(make_pair(0, s));
        while (!q.empty()) {
            int u = q.top().second;
            q.pop();
            if (expand[u] == 1) {
                continue;
            }
            expand[u] = 1;
            for (auto e : adj[u]) {
                int v = e.to, w = e.w;
                if (d[v] > d[u] + w) {
                    d[v] = d[u] + w;
                    q.push(make_pair(-d[v], v));
                }
            }
        }
    };
    dijkstra();
    for (int i = 1; i <= n; i++) {
        if (d[i] == 1e18) {
            cout << "-1" << " ";
        } else {
            cout << d[i] << " ";
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/wzj-xhjbk/p/11528900.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值