题目描述:
有n个传送机排成一排,编号从1到n,每个传送机都可以把自己位置的东西传送到距离自己[l, r]距离的位置,并且花费c,问从1号传送机到其他传送机的最小花费是多少
思路:
从第一个点开始更新, 更新过的点从未被更新的点中除去, 因为在Dijkstra里 每次取出的都是最小的distance, 所以更新过的点 后面肯定不需要再次更新。更新一个点后加入堆中, 因为通过这个点可能更新别的点。
PS:不太懂网上并查集优化的,只知道下面这个集合来优化的。下面这种方式需要最短路的赋值方式,没到该点前,先为到该点的花费赋值好。
#include<bits/stdc++.h> using namespace std; typedef long long ll; #define pii pair<ll, int> const int maxn = 200005; int l[maxn], r[maxn]; ll cost[maxn], dis[maxn]; pii p[maxn]; set<int> st; set<pii> vis; int main() { int t, n; scanf("%d", &t); while(t--) { scanf("%d", &n); for(int i = 0; i < n; i++) scanf("%d", &l[i]); for(int i = 0; i < n; i++) scanf("%d", &r[i]); for(int i = 0; i < n; i++) scanf("%I64d", &cost[i]); st.clear(); vis.clear(); memset(dis, -1, sizeof(dis)); dis[0] = 0; vis.insert(pii(cost[0], 0)); for(int i = 1; i < n; i++) st.insert(i); set<int>::iterator it, it2; while(vis.size() > 0) { pii tmp = *vis.begin(); vis.erase(vis.begin()); int u = tmp.second; it = st.lower_bound(u - r[u]); while(it != st.end() && *it <= u - l[u]) { int id = *it; dis[id] = tmp.first; vis.insert(pii(tmp.first + cost[id], id)); it2 = it++; st.erase(it2); } it = st.lower_bound(u + l[u]); while(it != st.end() && *it <= u + r[u]) { int id = *it; dis[id] = tmp.first; vis.insert(pii(tmp.first + cost[id], id)); it2 = it++; st.erase(it2); } } for(int i = 0; i < n; i++) printf("%I64d%c", dis[i], i == n-1 ? '\n' : ' '); } return 0; }