POJ 1860 Currency Exchange（bellman变形）

题目链接：【POJ 1860】

更改bellman的判断条件，判断是否存在正权回路

正权回路：这一回路上，顶点的权值能不断增加即能一直进行松弛

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <vector>
using namespace std;
int n, m, s;
double mon;
struct node
{
	int u, v;
	double r, c;
	node(int ui, int vi, double ri, double ci) : u(ui), v(vi), r(ri), c(ci) {}
};
vector<node>vec;
double res[110];
void addedge(int u, int v, double r, double c)
{
	vec.push_back(node(u,v,r,c));
}
bool bellman()
{
	memset(res, 0, sizeof(res));
	res[s] = mon;
	for(int i=0; i<n-1; i++)
	{
		int flag = 0;
		for(int j=0; j<vec.size(); j++)
		{
			int u=vec[j].u, v=vec[j].v;
			double r=vec[j].r, c=vec[j].c;
			if(res[v]<(res[u]-c)*r)
			{
				res[v]=(res[u]-c)*r;
				flag=1;
			}
		}
		if(!flag) return false;
	}
	for(int i=0; i<vec.size(); i++)
	{
		int u=vec[i].u, v=vec[i].v;
		double r=vec[i].r, c=vec[i].c;
		if(res[v]<(res[u]-c)*r) return true;
	}
	return false;
}
int main()
{
	int a, b;
	double r, c;
	cin>>n>>m>>s>>mon;
	for(int i=0; i<m; i++)
	{
		cin>>a>>b>>r>>c;
		addedge(a, b, r, c);
		cin>>r>>c;
		addedge(b, a, r, c);
	} 
	if(bellman()) cout<<"YES"<<endl;
	else cout<<"NO"<<endl;
	return 0;
}