码迷,mamicode.com
首页 > 编程语言 > 详细

Luogu P5960 【模板】差分约束算法

时间:2020-07-11 12:24:37      阅读:72      评论:0      收藏:0      [点我收藏+]

标签:eof   pac   oge   反转   void   scanf   ons   set   lin   

gate

差分约束系统用来解决:
给出\(n\)个变量\(x_1...x_n\)\(m\)个形如\(x_i-x_j \le k\)\(k\)为任意常量)的式子,
\(x_1...x_n\)的一组可行解。

将式子变形为\(x_i\le x_j + k\),发现刚好符合三角形不等式\(dis[v]\le dis[u]+w[i]\)的形式。
对于每个式子,连边\(x_j \rightarrow x_i\),权值为\(k\)\(0\)号节点向每个节点连边,权值为\(0\)
转化为求单源最短路,\(dis[1]...dis[n]\)即为解。
由于可能出现负权边,用\(SPFA\)求解。若出现负环,则无解。

对于一些其他形式的式子,需要稍作转化:
\(x_i-x_j \ge k\) 将两边同时乘\(-1\),将不等号反转。
\(x_i-x_j = k\) 拆成两个式子\(x_i-x_j \le k\)\(x_i-x_j \ge k\)

\(code\)

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<queue>
#define MogeKo qwq
using namespace std;

const int maxn = 1e5+10;
const int INF = 0x3f3f3f3f;

int n,m,x,y,z,cnt;
int head[maxn],to[maxn],nxt[maxn],w[maxn];
int dis[maxn],num[maxn];
bool vis[maxn];

void add(int x,int y,int z) {
	to[++cnt] = y;
	nxt[cnt] = head[x];
	head[x] = cnt;
	w[cnt] = z;
}

bool SPFA(int s) {
	memset(dis,INF,sizeof(dis));
	queue <int> q;
	dis[s] = 0;
	vis[s] = true;
	q.push(s);
	while(!q.empty()) {
		int u = q.front();
		q.pop();
		vis[u] = false;
		for(int i = head[u]; i; i = nxt[i]) {
			int v = to[i];
			if(dis[v] <= dis[u] + w[i]) continue;
			dis[v] = dis[u] + w[i];
			if(!vis[v]) {
				if(++num[v] == n) return false;
				vis[v] = true;
				q.push(v);
			}
		}
	}
	return true;
}

int main() {
	scanf("%d%d",&n,&m);
	for(int i = 1; i <= m; i++) {
		scanf("%d%d%d",&x,&y,&z);
		add(y,x,z);
	}
	for(int i = 1; i <= n; i++)
		add(0,i,0);
	if(!SPFA(0)) printf("NO");
	else
		for(int i = 1; i <= n; i++)
			printf("%d ",dis[i]);
	return 0;
}

Luogu P5960 【模板】差分约束算法

标签:eof   pac   oge   反转   void   scanf   ons   set   lin   

原文地址:https://www.cnblogs.com/mogeko/p/13283007.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!