codevs 1227 方格取数

本文介绍了一个使用最小费用流算法解决特定网络流量问题的C++实现。通过定义边的数据结构和采用SPFA算法来更新节点状态,实现了从源点到汇点的最大流量传输,并计算了最小费用。该实现涉及了图论、最短路径算法和网络流等关键概念。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define maxn 5550
#define maxm 25000
#define Inf 210000000
using namespace std;

struct Edge{
	int u, v, cap ,flow, cost;
	int next;
}e[maxm];

int fi[maxn], ecnt = 0;
int d[maxn], a[maxn], p[maxn];
bool inq[maxn];
int flow = 0, cost = 0;
int s, t, N;

void add_edge(int u, int v, int cap , int cost) {
	e[ecnt] = (Edge){u, v, cap, 0, cost, fi[u]};
	fi[u] = ecnt;
	ecnt++;
}

bool spfa(int s, int t) {
	memset(d, -1, sizeof(d));
	memset(inq, 0, sizeof(inq));
	d[s] = 0, inq[s] = 1, p[s] = 0, a[s] = Inf;
	int q[maxn*10], head = 0, tail = 1;
	q[head] = s;
	while(head < tail) {
		int u = q[head++];
		inq[u] = 0;
		for(int i = fi[u]; i != -1; i = e[i].next)
			if(e[i].cap > e[i].flow && d[e[i].v] < d[u] + e[i].cost) {
				d[e[i].v] = d[u] + e[i].cost;
				p[e[i].v] = i;
				a[e[i].v] = min(a[u], e[i].cap - e[i].flow);
				if(!inq[e[i].v]) {
					q[tail++] = e[i].v, inq[e[i].v] = 1;
				}
			}
	}
	if(d[t] == -1) return false;
	flow += a[t], cost += d[t]*a[t];
	int u = t;
	while(u != s) {
		e[p[u]].flow += a[t];
		e[p[u]^1].flow -= a[t];
		u = e[p[u]].u;
	}
	return true;
}

int main() {
	memset(fi, -1, sizeof(fi));
	int x, k = 2;
	int G[100][100];
	scanf("%d", &x);
	int n = x*x;
	int num = 0;
	int a, b, c; scanf("%d%d%d", &a, &b, &c);
	while(!(a == 0 && b == 0 && c == 0)) {
		G[a][b] = c;
		scanf("%d%d%d", &a, &b, &c);
	}
	for(int i = 1; i <= x; i++)
		for(int j = 1; j <= x; j++) {
			num++; 
			add_edge(num, num+n, 1, G[i][j]);
			add_edge(num+n, num, 0, -G[i][j]);
			add_edge(num, num+n, k, 0);
			add_edge(num+n, num, 0, 0);
			if(i != x)
				add_edge(num+n, num+x, Inf, 0), add_edge(num+x, num+n, 0, 0);
			if(j != x)
				add_edge(num+n, num+1, Inf, 0), add_edge(num+1, num+n, 0, 0);		
		}
	N = n*2+2;
	s = N-1, t = N;
	add_edge(s, 1, k, 0);     add_edge(1, s, 0, 0);
	add_edge(num+n, t, Inf, 0); add_edge(t, num+n, 0, 0);
	while(spfa(s, t));
	printf("%d", cost);
	return 0;
} 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值