1. 程式人生 > >CF 787D 線段樹套堆優化Dij

CF 787D 線段樹套堆優化Dij

題目連結:http://codeforces.com/contest/786/problem/B
題意:就是給了n個點,給了m個加邊的關係,有向邊,但是加邊不一定是u->v,可能是u->[l, r]就是u到[l, r]區間裡面的每一個點都加一條邊,或者是[l, r] -> u,就是[l, r]區間到u加一條邊。然後求單源最短路。

解法:顯然單源最短路用堆優化的DIJ來求。先考慮一下,如果我們把區間拆成點來加邊,那麼複雜度顯然會高達O(n^2logn)的。所以顯然不能這樣來加邊,這裡給了一堆線段?我們可以想怎麼把線段和點連線起來呢?顯然線段樹可以輕易的辦到,由於是有向邊所以這裡需要兩顆線段樹,線段樹的節點是2*n的,2顆就是4*n,j加上原來的n個點,所以節點數是5n。然後我們加邊原來最壞的O(n),現在可以變成logn了,那麼我們的複雜度可以 O(nlognlogn)了。

//CF 787D

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+10;
const int maxv = maxn*5;
int n, m, s;
vector <pair<int, int>> G[maxv];
void addedge(int u, int v, int w){
    G[u].push_back(make_pair(v, w));
}
int id[2][maxn<<2], idx;
void build(int l, int r, int
rt, int wh) { id[wh][rt] = ++idx; if(l == r){ if(wh == 0) addedge(id[wh][rt], l, 0); else addedge(l, id[wh][rt], 0); return; } int m = (l + r) / 2; build(l, m, rt*2, wh); build(m+1, r, rt*2+1, wh); if(wh == 0){ addedge(id[wh][rt], id[wh][rt*2], 0
); addedge(id[wh][rt], id[wh][rt*2+1], 0); } else{ addedge(id[wh][rt*2], id[wh][rt], 0); addedge(id[wh][rt*2+1], id[wh][rt], 0); } } vector <int> vs; void get(int L, int R, int l, int r, int rt, int wh) { if(L <= l && r <= R){ vs.push_back(id[wh][rt]); return ; } int m = (l + r) / 2; if(L <= m) get(L, R, l, m, rt*2, wh); if(m < R) get(L, R, m+1, r, rt*2+1, wh); } typedef long long LL; const LL inf = 0x3f3f3f3f3f3f3f3fLL; LL d[maxv]; bool vis[maxv]; void dij(int s) { for(int i = 1; i <= 5*n; i++) vis[i] = 0, d[i] = inf; d[s] = 0; priority_queue <pair<LL, int>> q; q.push({-0, s}); while(q.size()){ int u = q.top().second; q.pop(); if(vis[u]) continue; vis[u] = 1; for(auto &e : G[u]){ int v, c; tie(v, c) = e; if(d[v] > d[u] + c){ d[v] = d[u] + c; q.push({-d[v], v}); } } } } int main() { cin >> n >> m >> s; for(int i = 1; i <= 5*n; i++) G[i].clear(); idx = n; build(1, n, 1, 0); build(1, n, 1, 1); while(m--){ int t, u; scanf("%d%d", &t, &u); if(t == 1){ int v, c; scanf("%d%d", &v, &c); addedge(u, v, c); } else if(t == 2){ vs.clear(); int l, r, c; scanf("%d%d%d", &l, &r, &c); get(l, r, 1, n, 1, 0); for(int v : vs) addedge(u, v, c); } else{ vs.clear(); int l, r, c; scanf("%d%d%d", &l, &r, &c); get(l, r, 1, n, 1, 1); for(int v : vs) addedge(v, u, c); } } dij(s); for(int i = 1; i <= n; i++){ if(d[i] == inf) d[i] = -1; printf("%I64d ", d[i]); } printf("\n"); }