Sightseeing Cows POJ - 3621 01分数划分+SPFA求正环

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

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

输出这个最大值。

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

输入格式
第一行包含两个整数L和P。

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

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

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

数据范围
2≤L≤1000,
2≤P≤5000,
1≤f[i],t[i]≤1000

样例

Sample Input
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
Sample Output
6.00

思路

• 用f代表点权， w代表边
• 求f/w的最大值， 设f/w=mid， 可以得到： f - t * mid = 0;
• 我们二分mid， 如果 f - t * mid > 0; (存在正环) 此时 答案大于mid， l = mid ， 否则答案<=mid, r = mid
• 我们控制精度1e-4， 最后的 l 和 r 就是答案

代码

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

using namespace std;

const int N = 1010, M = 5010;

int e[M], ne[M], h[N], w[M], len;
int f[N];
bool vis[N];
int cnt[N];
double dis[N];
int n, m;

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

bool check(double mid)
{

    memset(dis, 0, sizeof dis);
    memset(cnt, 0, sizeof cnt);
    queue<int> q;
    for(int i = 1; i <= n; i++)
    {
        q.push(i);
        vis[i] = true;
    }
        
    while (q.size())
    {
        int t = q.front();
        q.pop();
        vis[t] = false;
        
        for(int i = h[t]; ~i; i = ne[i])
        {
            int j = e[i];
            if (dis[j] < dis[t] + f[t] - mid * w[i])
            {
                dis[j] = dis[t] + f[t] - mid * w[i];
                cnt[j] = cnt[t] + 1;
                if (cnt[j] >= n) return true;
                if (!vis[j])
                {
                    q.push(j);
                    vis[j] = true;
                }
                
            }
        }
    }
    return false;
}

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