[ An Ac a Day ^_^ ][kuangbin带你飞]专题四 最短路练习 POJ 2240 Arbitrage spfa求负环

题意就是问倒腾外币能不能升值

不用spfa 用其他的最短路算法也可以

松弛条件换成dist[v]<dist[u]*e[u][i].value

当然 貌似只有spfa有这个坑……

有A  (value>1.0) A 这种情况……我的天

用Dij Floyd都只用判断如果松弛到了自己 那么一定有环 直接跳出就行

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<ctime>
#include<map>
#include<set>
#include<vector>
#include<queue>
#include<cstdlib>
#include<cassert>
#include<sstream>
#include<stack>
#include<list>
#include<bitset>
#define cl(a,b) memset(a,b,sizeof(a))
#define debug(x) cerr<<#x<<"=="<<(x)<<endl
using namespace std;
typedef long long ll;

const int maxn=1e2+10;

struct edge
{
    int v;
    double value;
    edge(int _v=0,double _value=0):v(_v),value(_value) {}
};

map<string,int>mp;
vector<edge>e[maxn];
int n,m;
bool vis[maxn],ok;
int cnt[maxn];
double dist[maxn];

void addedge(int u,int v,double w)
{
    e[u].push_back(edge(v,w));
}

void init()
{
    ok=false;
    mp.clear();
    for(int i=0;i<maxn;i++) e[i].clear();
}

bool spfa(int start)
{
    cl(vis,false),cl(dist,0),cl(cnt,0);
    //注意dist[start]初始化是1
    vis[start]=true,dist[start]=1.0,cnt[start]=1;
    queue<int>q;
    while(!q.empty()) q.pop();
    q.push(start);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=false;
        for(int i=0; i<e[u].size(); i++)
        {
            int v=e[u][i].v;
            if(dist[v]<dist[u]*e[u][i].value)
            {
                dist[v]=dist[u]*e[u][i].value;
                if(!vis[v])
                {
                    vis[v]=true;
                    q.push(v);
                    cnt[v]++;
                    if(cnt[v]==n) return true;
                }
                else if(dist[start]>1.0) return true;//特判一下
            }
        }
    }
    return false;
}

int main()
{
//    freopen("in.txt","r",stdin);
    int cas=1;
    while(~scanf("%d",&n)&&n)
    {
        init(); //记得初始化
        for(int i=0; i<n; i++)
        {
            string place;
            cin>>place;
            mp[place]=i;
        }
        scanf("%d",&m);
        for(int i=0; i<m; i++)
        {
            string place1,place2;
            double value;
            cin>>place1>>value>>place2;
            addedge(mp[place1],mp[place2],value);
            //无限WA之后加了下面这句就AC了 T_T
            if(mp[place1]==mp[place2]&&value>1.0) ok=true;
        }
        for(int i=0;i<n&&!ok;i++) ok=spfa(i);
        printf("Case %d: %s
",cas++,ok?"Yes":"No");//注意是Yes 不是YES
    }
    return 0;
}