HLJOJ1214（最短路）

来源：互联网发布：java中字符串的比较编辑：程序博客网时间：2024/06/17 02:42

1214: Around the world

Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 12 Solved: 2
[Submit][Status][Web Board]

Description

现在地图上有N个城市(你所在的城市编号为1)，但是你的汽车油箱容量只有V，你只能在到达每个城市时才能够补充满油箱。现在告诉你每两个城市之间所需要消耗的油量（两个城市之间的路可能不止一条，消耗的油量也可能不同），是否存在一种方案使你能够周游这N个城市（每个城市可以走多次）。

Input

多组样例到文件尾结束。
每组样例第一行输入N,M,V，表示城市数目，道路数，油箱容量。 2 <= N <= 10000, 1 <= M <= 50000, 1 <= V <= 1000。
接下来输入M行，每行包含Xi，Yi，Ci。表示城市Xi,Yi之间存在一条耗油量Ci的路。1<=X,Y<=N, 1<=Ci<=1000。

PS：道路是双向的。

Output

如果存在输出YES
否则输出NO

Sample Input

4 4 21 4 21 3 22 3 13 4 43 3 21 2 31 3 22 3 3

Sample Output

YES

这题点太多了，所以需要用邻接表存储，而且要开到否则会RE。。。1WA,1RE,

#include <iostream>#include <cmath>#include <cstring>#include <cstdio>#include <queue>#include <algorithm>using namespace std; const int maxn = 1000000 + 10;const int INF = 99999999;int head[maxn];int num[maxn];int d[maxn];int visit[maxn];int V,M,C;int size;struct Edge{    int to,w,next;} E[maxn]; void init(){    memset(head,-1,sizeof(head));    fill(d,d+V+1,INF);    size = 0;} void addedge(int u,int v,int x){    E[size].to = v;    E[size].w = x;    E[size].next = head[u];    head[u] = size++;} int  SPFA(int s){    queue<int> q;    fill(num,num+V+1,0);    fill(visit,visit+V+1,0);    d[s]=0;    q.push(s);    visit[s] = 1;    while(!q.empty())    {         int u = q.front();        q.pop();        for(int i=head[u]; i!=-1; i=E[i].next)        {             if(d[E[i].to] > d[u] + E[i].w && E[i].w <= C )            {                d[E[i].to] = d[u]+E[i].w;                q.push(E[i].to);            }        }     }     return 1;} int main(){#ifdef xxz    freopen("in.txt","r",stdin);#endif // xxz     while(cin>>V>>M>>C)    {        init();        for(int i = 0; i < M; i++)        {            int a,b,c;            cin>>a>>b>>c;            addedge(a,b,c);            addedge(b,a,c);        }         SPFA(1);        bool flag = true;        for(int i = 1; i <= V; i++)        {            if(d[i] >= INF)            {                flag = false;                break;            }        }        if(!flag)cout<<"NO"<<endl;        else cout<<"YES"<<endl;     }    return 0;}

还有一种方法可以用并查集解决:

#include <algorithm>#include <iostream>#include <cstdio>#include <cstring>#include <cmath>#include <cstdlib>#include <vector>#include <queue>#include <map>#define INF 0x7fffffffusing namespace std; const int MOD = 1e9 + 7;const int N = 5e4 + 10; int n, m, v;struct edge {    int x, y, c;} e[N];int fa[N]; int cmp(edge a, edge b) {    return a.c < b.c;}int findx(int x) {    if(fa[x] == x) return x;    else return fa[x] = findx(fa[x]);}int main() {    while(scanf("%d%d%d", &n, &m, &v) != EOF) {        for(int i = 1; i <= n; i++) fa[i] = i;        for(int i = 0; i < m; i++) {            scanf("%d%d%d", &e[i].x, &e[i].y, &e[i].c);        }        sort(e, e + m, cmp);        int cnt = 0;        for(int i = 0; i < m && cnt < n - 1; i++) {            if(e[i].c > v) break;            int fx = findx(e[i].x);            int fy = findx(e[i].y);            if(fx != fy) {                fa[fx] = fy;                cnt++;            }        }        if(cnt == n - 1) puts("YES");        else puts("NO");    }    return 0;}

0 0