SPFA(判负环) + 二分(01分数规划) - Sightseeing Cows - POJ 3621

SPFA(判负环) + 二分(01分数规划) - Sightseeing Cows - POJ 3621

题意：

给定一张L个点、P条边的有向图，每个点都有一个权值f[i]，每条边都有一个权值t[i]。

求图中的一个环，使“环上各点的权值之和”除以“环上各边的权值之和”最大。

输出这个最大值。

注意： 数据保证至少存在一个环。

输入格式

第一行包含两个整数L和P。

接下来L行每行一个整数，表示f[i]。

再接下来P行，每行三个整数a，b，t[i]，表示点a和b之间存在一条边，边的权值为t[i]。

输出格式

输出一个数表示结果，保留两位小数。

数据范围

$2\le L\le 1000,2\le P\le 5000,1\le f[i],t[i]\le 1000$

输入样例：

5 7
30
10
10
5
10
1 2 3
2 3 2
3 4 5
3 5 2
4 5 5
5 1 3
5 2 2

输出样例：

6.00

---

分析：

$本题目标：对图中每个环，求max(\frac{\sum f[i]}{\sum t[i]})，1\le i\le n。$

01分数规划$通常用于求解最{高}^{\prime \prime }性价{比}^{\prime \prime }的问题。$

$我们可以二分最大值mid，接着判断是否存在某个环满足max(\frac{\sum f[i]}{\sum t[i]})>mid，$

$若存在，说明答案在区间[mid,r]之间，否则答案在[l,mid]之间。$

$答案的上界为\frac{1000}{1}=1，下界为\frac{1}{1000}。所以二分的区间取[0,1000]。$

转化：

$对不等式：\frac{\sum f[i]}{\sum t[i]}>mid，$

$等价于\sum f[i]>mid\times \sum t[i]，$

$等价于\sum f[i]-mid\times \sum t[i]>0，$

$等价于\sum (f[i]-mid\times t[i])>0。$

$我们可以将每个点的权值累加到对应的出边的权值上，$

$对于二分的每一个mid，将图中所有的边权转化为f[i]-mid\times t[i]，$

$问题转化为新的图中是否存在正环。$

$此时跑一遍spfa即可。$

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;

const int N=1010, M=5010;

int n,m,wf[N];
int e[M],ne[M],wt[M],h[N],idx;
double dis[N];
int cnt[N];
bool st[N];

void add(int a,int b,int c)
{
    e[idx]=b,ne[idx]=h[a],wt[idx]=c,h[a]=idx++;
}

bool check(double mid)
{
    memset(st,false,sizeof st);
    memset(cnt,0,sizeof cnt);
    
    queue<int> Q;
    for(int i=1;i<=n;i++)
    {
        Q.push(i);
        st[i]=true;
    }
    
    while(Q.size())
    {
        int u=Q.front();
        Q.pop();
        st[u]=false;
        
        for(int i=h[u];~i;i=ne[i])
        {
            int j=e[i];
            double w=wf[u]-mid*wt[i];  //这里必须用double确保精度
            if(dis[j]<dis[u]+w)
            {
                dis[j]=dis[u]+w;
                cnt[j]=cnt[u]+1;
                if(cnt[j]>=n) return true;
                if(!st[j])
                {
                    st[j]=true;
                    Q.push(j);
                }
            }
        }
    }
    
    return false;
}

int main()
{
    memset(h,-1,sizeof h);
    
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>wf[i];
    int a,b,c;
    while(m--)
    {
        cin>>a>>b>>c;
        add(a,b,c);
    }
    
    double l=0,r=1000;
    while(r-l>1e-4)
    {
        double mid=(l+r)/2;
        if(check(mid)) l=mid;
        else r=mid;
    }
    
    printf("%.2lf\n",r);
    
    return 0;
}