HDU5361In Touch (最短路+并查集)

来源:互联网 发布:营销推广案例 知乎 编辑:程序博客网 时间:2024/06/01 07:59

题目描述:

  有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;}


原创粉丝点击