ZOJ 3946 Highway Project【dijkstra】【贪心】

题目链接

http://icpc.moe/onlinejudge/showProblem.do?problemId=5718

思路

给你一个无向图，每条边有一个时间c和花费d，叫你选一些边，使得点0到其他所有点的时间之和最小，其次，使总花费最小。

因为要使得点0到其他所有点的时间之和最小，所以是个最短路问题，用dijkstra找最短路，为了让花费最小，更新距离的时候，如果耗时相等，但新边的花费比旧边少的话，也要更新上去。

dis[i].c记录0到i点的最短用时，dis[i].d记录这个点紧连的那条边的花费。

最后答案就是所有dis[i].c和dis[i].d的和了。

比赛的时候优先队列里直接放点编号了，然后比较函数又调用了全局的dis，导致运算过程中堆会变的不合法（因为dis会变），这个在VC++里会直接抛出invalid heap 异常。于是弄到最后代码都没成功运行…（总归还是太弱）

AC代码

#include <iostream>#include <queue>#include <vector>#include <cstring>#include <cstdio>using namespace std;typedef long long ll;const int N = 100000 + 100;const int E = 200000 + 100;const ll INF = ((ll)1) << 61;struct edge_t {    int v;    int next;    int c, d;}edge[E];int head[N], tot = 0;void add_edge(int u, int v, int d, int c){    edge[tot].d = d; edge[tot].c = c;    edge[tot].v = v; edge[tot].next = head[u]; head[u] = tot++;}struct node{    int u;    ll c,d;    node (){}    node(int uu, int cc, int dd)    {        u=uu;        c=cc;        d=dd;    }    friend bool operator < (const node &a, const node &b)    {        if(a.d==b.d)return a.c>b.c;        return a.d>b.d;    }};bool vis[N];node dis[N];priority_queue<node>q;void dijkstra(){    dis[0].c = 0;    dis[0].d = 0;    q.push(dis[0]);    while (!q.empty())    {        int u = q.top().u; q.pop();        if(vis[u])continue;        vis[u] = 1;        for (int i = head[u]; ~i; i = edge[i].next)if (!vis[edge[i].v])        {            int v = edge[i].v;            if (dis[u].d + edge[i].d < dis[v].d)            {                dis[v].d = dis[u].d + edge[i].d;                dis[v].c = edge[i].c;                q.push(dis[v]);            }            else if (dis[u].d + edge[i].d == dis[v].d&& edge[i].c < dis[v].c)            {                dis[v].d = dis[u].d + edge[i].d;                dis[v].c = edge[i].c;                q.push(dis[v]);            }        }    }}void init(int n){    for (int i = 0; i < n; ++i)    {        dis[i].u = i;        dis[i].c = INF;        dis[i].d = INF;    }    memset(vis, 0, sizeof vis);    tot = 0;    memset(head, -1, sizeof head);    memset(edge, 0, sizeof edge);}int main(){    int T;    scanf("%d", &T);    while (T--)    {        int n, m;        scanf("%d%d", &n, &m);        init(n);        for (int i = 0; i < m; ++i)        {            int u, v, d, c;            scanf("%d%d%d%d", &u, &v, &d, &c);            add_edge(u, v, d, c);            add_edge(v, u, d, c);        }        dijkstra();        ll sum_c = 0, sum_d = 0;        for(int i=0 ; i<n ; ++i)        {            sum_c+=dis[i].c;            sum_d+=dis[i].d;        }        printf("%lld %lld\n", sum_d, sum_c);    }    return 0;}