Codeforces 787D Legacy 线段树 最短路

时间:2023-03-09 07:19:28
Codeforces 787D Legacy 线段树 最短路

题意:

有\(n(1 \leq n \leq 10^5)\)个点,\(q(1 \leq q \leq 10^5)\)条路和起点\(s\)

路有三种类型:

  • 从点\(v\)到点\(u\)需要花费\(w\)
  • 从点\(v\)到区间\([l,r]\)中的点花费为\(w\)
  • 从区间\([l,r]\)中的点到点\(v\)花费为\(w\)

求起点到各个点的最少花费

分析:

如下图,构建两颗线段树,边的花费都为\(0\)

对于第一种路直接加边即可

对于第二种路,添加从\(v\)到上面线段树对应区间中的点的边

对于第三种路,添加从下面线段树对应区间中的点到\(v\)的边

Codeforces 787D Legacy 线段树 最短路

最后直接跑最短路

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#include <set>
#include <vector>
#include <iostream>
#include <string>
#include <queue>
using namespace std;
#define REP(i, a, b) for(int i = a; i < b; i++)
#define PER(i, a, b) for(int i = b - 1; i >= a; i--)
#define SZ(a) ((int)a.size())
#define MP make_pair
#define PB push_back
#define EB emplace_back
#define ALL(a) a.begin(), a.end()
typedef long long LL;
typedef pair<LL, int> PII; const int maxn = 100000 + 10; int n, q, s; struct Edge {
int v, w, nxt;
Edge() {}
Edge(int v, int w, int nxt): v(v), w(w), nxt(nxt) {}
}; int ecnt, head[maxn * 9];
Edge edges[maxn * 36]; void AddEdge(int u, int v, int w) {
printf("%d->%d, w = %d\n", u, v, w);
edges[ecnt] = Edge(v, w, head[u]);
head[u] = ecnt++;
} int tot, T[2][maxn << 2];
void build(int t, int o, int L, int R) {
T[t][o] = ++tot;
if(L == R) {
if(0 == t) AddEdge(T[t][o], L, 0);
else AddEdge(L, T[t][o], 0);
return;
}
int M = (L + R) / 2;
build(t, o<<1, L, M);
build(t, o<<1|1, M+1, R);
if(0 == t) {
AddEdge(T[t][o], T[t][o<<1], 0);
AddEdge(T[t][o], T[t][o<<1|1], 0);
} else {
AddEdge(T[t][o<<1], T[t][o], 0);
AddEdge(T[t][o<<1|1], T[t][o], 0);
}
} int type, v, qL, qR, w; void update(int o, int L, int R) {
if(qL <= L && R <= qR) {
if(0 == type) AddEdge(v, T[type][o], w);
else AddEdge(T[type][o], v, w);
return;
}
int M = (L + R) / 2;
if(qL <= M) update(o<<1, L, M);
if(qR > M) update(o<<1|1, M+1, R);
} const LL INF = 0x3f3f3f3f3f3f3f3f;
bool vis[maxn * 9];
LL d[maxn * 9]; priority_queue<PII, vector<PII>, greater<PII> > Q;
void dijkstra() {
memset(d, 0x3f, sizeof(d));
d[s] = 0;
Q.emplace(0, s);
while(!Q.empty()) {
PII t = Q.top(); Q.pop();
int& u = t.second;
if(vis[u]) continue;
vis[u] = true;
for(int i = head[u]; ~i; i = edges[i].nxt) {
int& v = edges[i].v;
int& w = edges[i].w;
if(d[v] > d[u] + w) {
d[v] = d[u] + w;
Q.emplace(d[v], v);
}
}
}
} int main() {
scanf("%d%d%d", &n, &q, &s);
memset(head, -1, sizeof(head));
tot = n;
REP(t, 0, 2) build(t, 1, 1, n); while(q--) {
scanf("%d%d%d", &type, &v, &qL);
if(type == 1) {
scanf("%d", &w);
AddEdge(v, qL, w);
} else {
type -= 2;
scanf("%d%d", &qR, &w);
update(1, 1, n);
}
} dijkstra();
REP(i, 1, n + 1) printf("%lld ", d[i] == INF ? -1 : d[i]);
printf("\n"); return 0;
}