POJ3621 Sightseeing Cows 01规划/最优比率生成环+spfa最长路

这道题做的并不是很顺手,spfa找最长路判负环第一次接触,不知道有负环代表了答案过大还是过小。。

同样先贴上大佬博客Orz:https://blog.csdn.net/sdj222555/article/details/7692185

                                          https://www.cnblogs.com/luyouqi233/p/8268187.html

我一开始写的bfs版spfa,TLE了,但我不知道为什么第一篇博客的大佬代码就能过。。难道是因为STL操作比较耗时间???附上TLE的代码:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
#define ll long long
typedef pair<int,int>pp;
const double pi=acos(-1.0);
const double eps=1e-9;
const int INF=0x3f3f3f3f;
const ll MOD=1e9+7ll;

const int MAX=1005;
int n,m;
double fun[MAX];
int head[MAX];
struct Edge
{
    int v;
    double w;
};
vector<struct Edge>edge[MAX];

bool vis[MAX];
int cnt[MAX];//每个点的入队列次数
int dis[MAX];
queue<int>q;
int spfa(int start,double mid)
{
    memset(vis,false,sizeof(vis));
    memset(dis,0,sizeof(dis));//注:求最长路清0而不是INF!!!
    memset(cnt,0,sizeof(cnt));
    while(!q.empty())
        q.pop();
    vis[start]=true;
    dis[start]=0;
    cnt[start]=1;
    q.push(start);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=false;
        for(int i=0;i<edge[u].size();i++)
        {
            int v=edge[u][i].v;
            double w=fun[u]-mid*edge[u][i].w;
            if(dis[v]<dis[u]+w)//最长路
            {
                dis[v]=dis[u]+w;
                if(!vis[v])
                {
                    vis[v]=true;
                    q.push(v);
                    cnt[v]++;
                    if(cnt[v]>n)//存在负环,答案还可以变得更大
                        return false;
                }
            }
        }
    }
    return true;
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)
            scanf("%lf",&fun[i]);
        memset(head,-1,sizeof(head));
        memset(edge,0,sizeof(edge));
        for(int i=0;i<=n;i++)
            edge[i].clear();
        struct Edge e;
        int u,v;double w;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%lf",&u,&v,&w);
            e.v=v;e.w=w;
            edge[u].push_back(e);
        }
        double l=0.0,r=1000.0;
        while((r-l)>1e-6)
        {
            double mid=(l+r)/2;
            if(spfa(1,mid))
                r=mid;
            else
                l=mid;
        }
        printf("%.2f\n",l);
    }
	return 0;
}

然后就看网上说写dfs版的spfa,会快点,于是只能再改了。。模仿了第二篇博客里的dfs版,找了一天的错最后发现是因为dis数组开成了int。。应该是double。。我怎么会这么菜555555…

附上AC代码:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
#define ll long long
typedef pair<int,int>pp;
const double pi=acos(-1.0);
const double eps=1e-9;
const int INF=0x3f3f3f3f;
const ll MOD=1e9+7ll;

const int MAX=1005;
int n,m;
double fun[MAX];
int head[MAX];
struct Edge
{
    int v;
    double w;
};
vector<struct Edge>edge[MAX];

bool vis[MAX];
int cnt[MAX];//每个点的入队列次数
double dis[MAX];//注意是double!!!
double w[MAX];
bool spfa(int u,double x)
{
    vis[u]=true;
    for(int i=0;i<(int)edge[u].size();i++)
    {
        int v=edge[u][i].v;
        double w=fun[u]-x*edge[u][i].w;
        if(dis[v]<dis[u]+w)//找最长路
        {
            dis[v]=dis[u]+w;
            if(vis[v]||spfa(v,x))
            {
                vis[v]=false;
                return true;
            }
        }
    }
    vis[u]=false;
    return false;
}
bool judge(double mid)
{
    for(int i=1;i<=n;i++)
        if(spfa(i,mid))
            return true;
    return false;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%lf",&fun[i]);
    memset(head,-1,sizeof(head));
    memset(edge,0,sizeof(edge));
    for(int i=0;i<=n;i++)
        edge[i].clear();
    struct Edge e;
    int u,v;double w;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%lf",&u,&v,&w);
        e.v=v;e.w=w;
        edge[u].push_back(e);
    }
    double l=0,r=100000;
    while(r-l>eps)
    {
        double mid=(l+r)/2;
        if(judge(mid))
            l=mid;
        else
            r=mid;
    }
    printf("%.2f\n",l);
    return 0;
}

 

点赞