CF 787D 線段樹套堆優化Dij
阿新 • • 發佈:2018-12-31
題目連結: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");
}