USACO-Section 3.2 Sweet Butter（最短路[Dijkstra]）

来源：互联网发布：中国网络黑市编辑：程序博客网时间：2024/05/24 00:30

此处有目录↑

描述

农夫John发现做出全威斯康辛州最甜的黄油的方法：糖。把糖放在一片牧场上，他知道N（1<=N<=500）只奶牛会过来舔它，这样就能做出能卖好价钱的超甜黄油。当然，他用额外赚来的钱给奶牛买奢侈品。

农夫John很狡猾。他知道他可以训练这些奶牛，让它们在听到铃声时去一个特定的牧场。他打算将糖放在那里然后下午发出铃声，以至他可以在晚上挤奶。

农夫John知道每只奶牛都在各自喜欢的牧场呆着（一个牧场不一定只有一头牛）。给出各头牛在的牧场和牧场间的路线，找出使所有牛到达的路程和最短的牧场（他将把糖放在那）。

格式

PROGRAM NAME: butter

INPUT FORMAT:

(file butter.in)

第一行: 三个数：奶牛数N，牧场数P（2<=P<=800），牧场间道路数C(1<=C<=1450).（奶牛的编号为1..N,牧场的编号为1..P)

第二行到第N+1行: 1到N头奶牛所在的牧场号.

第N+2行到第N+C+1行：每行有三个数：相连的牧场A、B，两牧场间距（1<=D<=255），当然,连接是双向的.

OUTPUT FORMAT:

(file butter.out)

一行输出奶牛必须行走的最小的距离和.

SAMPLE INPUT

3 4 52341 2 11 3 52 3 72 4 33 4 5

简直冇玛

样例图形

         P2  P1 @--1--@ C1    \    |\     \   | \      5  7  3       \ |   \        \|    \ C3      C2 @--5--@         P3    P4

SAMPLE OUTPUT

{说明：放在4号牧场最优.}

由于点是800个，所以只能用优化过的Dijkstra算法

复习时又看到 Johnson算法可以解决稀疏图的全源最短路，但是比较麻烦，需要重新构图

/*ID: your_id_herePROG: butterLANG: C++*/#include <cstdio>#include <cstring>#include <string>#include <queue>#include <vector>#include <algorithm>using namespace std;struct Node {    int e,w;    Node(int ee=0,int ww=0):e(ee),w(ww) {}    bool operator <(const Node& a) const {        return w>a.w;    }}u,v;int n,p,c,s,e,w,ans,tmp;int dis[805],cow[505];vector<Node> edge[805];priority_queue<Node> q;bool vis[805];void Dijkstra(int s) {    memset(vis,false,sizeof(vis));    memset(dis,0x3f,sizeof(dis));    dis[s]=0;    q.push(Node(s,0));    while(!q.empty()) {        u=q.top();        q.pop();        if(!vis[u.e]) {            vis[u.e]=true;            for(int i=0;i<edge[u.e].size();++i) {                v=edge[u.e][i];                if(!vis[v.e]&&dis[u.e]+v.w<dis[v.e]) {                    dis[v.e]=dis[u.e]+v.w;                    q.push(Node(v.e,dis[v.e]));                }            }        }    }}int main() {    freopen("butter.in","r",stdin);    freopen("butter.out","w",stdout);    scanf("%d%d%d",&n,&p,&c);    for(int i=0;i<n;++i) {        scanf("%d",&s);        cow[i]=s;    }    for(int i=0;i<c;++i) {        scanf("%d%d%d",&s,&e,&w);        edge[s].push_back(Node(e,w));        edge[e].push_back(Node(s,w));    }    ans=0x3f3f3f3f;    for(int i=1;i<=p;++i) {//枚举每个点作为放糖的点        Dijkstra(i);        tmp=0;        for(int j=0;j<n;++j)            tmp+=dis[cow[j]];        ans=min(ans,tmp);    }    printf("%d\n",ans);    return 0;}

0 0