ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

POJ - 3259 - Wormholes = SPFA

2019-10-23 20:54:01  阅读:239  来源: 互联网

标签:3259 int top SPFA edge POJ MAXN include dis


http://poj.org/problem?id=3259

看见500的数据量莽了一发Floyd,还T了。讲道理应该不至于的,换个快读试试。

#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<map>
#include<set>
#include<stack>
#include<string>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;

const int MAXN = 500;
int dis[MAXN + 5][MAXN + 5];

int main() {
#ifdef Yinku
    freopen("Yinku.in", "r", stdin);
#endif // Yinku
    int t;
    scanf("%d", &t);
    while(t--) {
        memset(dis, 0x3f, sizeof(dis));
        int n, m, l;
        scanf("%d%d%d", &n, &m, &l);
        for(int i = 1; i <= n; ++i) {
            dis[i][i] = 0;
        }
        for(int i = 1; i <= m; ++i) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            dis[u][v] = min(dis[u][v], w);
            dis[v][u] = min(dis[v][u], w);
        }
        for(int i = 1; i <= l; ++i) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            dis[u][v] = min(dis[u][v], -w);
        }
        bool suc = 0;
        for(int k = 1; k <= n; ++k) {
            for(int i = 1; i <= n; ++i) {
                for(int j = 1; j <= n; ++j) {
                    dis[i][j] = min(dis[i][j], dis[i][k] + dis[k][j]);
                }
                if(dis[i][i] < 0) {
                    suc = 1;
                    break;
                }
            }
            if(suc)
                break;
        }
        puts(suc ? "YES" : "NO");
    }
}

但是假如图本身连通的话,写SPFA就可以了。

#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<map>
#include<set>
#include<stack>
#include<string>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;

const int MAXN = 505;
const int MAXM = 10005;

int top;
int head[MAXN];
struct Edge {
    int v, nxt;
    int w;
} edge[MAXM];

void init() {
    top = 0;
    memset(head, -1, sizeof(head));
}

void add_edge(int u, int v, int w) {
    ++top;
    edge[top].v = v;
    edge[top].w = w;
    edge[top].nxt = head[u];
    head[u] = top;
}

bool vis[MAXN];
int cnt[MAXN];
int dis[MAXN];

queue<int>q;
bool spfa(int s, int n) {
    memset(vis, 0, sizeof(vis));
    memset(cnt, 0, sizeof(cnt));
    for(int i = 0; i < MAXN; ++i)
        dis[i] = 0x3f3f3f3f;

    while(!q.empty())
        q.pop();
    q.push(s);
    vis[s] = 1;
    cnt[s] = 1;
    dis[s] = 0;

    while(!q.empty()) {
        int u = q.front();
        q.pop();
        vis[u] = 0;
        for(int i = head[u]; i != -1; i = edge[i].nxt) {
            int v = edge[i].v;
            if(dis[v] > dis[u] + edge[i].w) {
                dis[v] = dis[u] + edge[i].w;
                if(!vis[v]) {
                    vis[v] = 1;
                    q.push(v);
                    if(++cnt[v] > n)
                        return 0;
                }
            }
        }
    }
    return 1;
}

int main() {
#ifdef Yinku
    freopen("Yinku.in", "r", stdin);
#endif // Yinku
    int t;
    scanf("%d", &t);
    while(t--) {
        int n, m, l;
        scanf("%d%d%d", &n, &m, &l);
        init();
        for(int i = 1; i <= m; ++i) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            add_edge(u, v, w);
            add_edge(v, u, w);
        }
        for(int i = 1; i <= l; ++i) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            add_edge(u, v, -w);
        }
        bool suc = !spfa(1, n);
        puts(suc ? "YES" : "NO");
    }
}

速度172ms,可能可以试试优化。

标签:3259,int,top,SPFA,edge,POJ,MAXN,include,dis
来源: https://www.cnblogs.com/Inko/p/11728777.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有