cogs729. [网络流24题] 圆桌聚餐

«问题描述:
假设有来自m 个不同单位的代表参加一次国际会议。每个单位的代表数分别为
ri(i=1,2,3...m), 。会议餐厅共有n张餐桌,每张餐桌可容纳c i(i=1,2...n) 个代表就餐。
为了使代表们充分交流,希望从同一个单位来的代表不在同一个餐桌就餐。试设计一个算法,
给出满足要求的代表就餐方案。
«编程任务:
对于给定的代表数和餐桌数以及餐桌容量,编程计算满足要求的代表就餐方案。
«数据输入:
由文件roundtable.in提供输入数据。文件第1行有2 个正整数m和n,m表示单位数,n表
示餐桌数,1<=m<=150, 1<=n<=270。文件第2 行有m个正整数,分别表示每个单位的代表
数。文件第3 行有n个正整数,分别表示每个餐桌的容量。
«结果输出:
程序运行结束时,将代表就餐方案输出到文件roundtable.out中。如果问题有解,在文件第
1 行输出1,否则输出0。接下来的m行给出每个单位代表的就餐桌号。如果有多个满足要
求的方案,只要输出1 个方案。
输入文件示例 输出文件示例
roundtable.in


4 5
4 5 3 5
3 5 2 6 4
roundtable.out


1
1 2 4 5
1 2 3 4 5
2 4 5
1 2 3 4 5
题解:
建一个超级源点和一个超级汇点 源点与单位连边 容量为人数 单位与每一张桌子连边 容量为1 桌子与汇点连边容量为
桌子能坐的人数 跑最大流 如果能坐下 那么最大流就是最终能坐下的人数
输出方案:看哪一张桌子与单位之间的边流满了 就代表选了
代码:
#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
const int maxn=500+10;
const int INF=0x7fffffff;
vector<int>A[maxn];
vector<int>to,cap;
queue<int>q;
vector<int>ans[maxn];
int dis[maxn];
int S,T;
int n,m;
int v[maxn],d[maxn];
int edge=0;
inline bool bfs(){
	memset(dis,0,sizeof(dis));
	dis[S]=1;
	q.push(S);
	while(!q.empty()){
		int x=q.front();
		q.pop();
		for(int i=0;i<A[x].size();i++){
			int u=A[x][i];
			if(!dis[to[u]]&&cap[u]){
				dis[to[u]]=dis[x]+1;
				q.push(to[u]);
			}
		}
	}
	return dis[T];
}
inline int dfs(int x,int mx){
	if(x==T)
		return mx;
	int flow=0;
	for(int i=0;i<A[x].size();i++){
		int u=A[x][i];
		int d=0;
		if(cap[u]&&dis[to[u]]==dis[x]+1&&(d=dfs(to[u],min(mx,cap[u])))){
			cap[u]-=d;
			cap[u^1]+=d;
			flow+=d;
			if(flow==mx)
				return flow;
		}
	}
	if(!flow)
		dis[x]=0;
	return flow;
}
inline void addedge(int f,int t,int c){
	to.push_back(t);
	cap.push_back(c);
	A[f].push_back(edge++);
	to.push_back(f);
	cap.push_back(0);
	A[t].push_back(edge++);
}
int main(){
	freopen("roundtable.in","r",stdin);
	freopen("roundtable.out","w",stdout);
	scanf("%d %d",&n,&m);
	S=0,T=n+m+1;
	int sum=0;
	for(int i=1;i<=n;i++){
		scanf("%d",&v[i]);
		addedge(S,i,v[i]);
		sum+=v[i];
	}
	for(int i=1;i<=m;i++){
		scanf("%d",&d[i]);
		addedge(i+n,T,d[i]);
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++)
			addedge(i,j+n,1);
	}
	int Ans=0;
	while(bfs())
		Ans+=dfs(S,INF);
	if(Ans!=sum){
		puts("0");
		return 0;
	}
	puts("1");
	for(int i=1;i<=n;i++){
		for(int j=0;j<A[i].size();j++){
			int u=A[i][j];
			if(to[u]>n&&to[u]<=n+m){
				if(cap[u]==0)
					ans[i].push_back(to[u]-n);
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=0;j<ans[i].size();j++)
			printf("%d ",ans[i][j]);
		puts("");
	}
return 0;
}


以下是将代码修改为cot平滑的方法: 1. 首先,需要使用边界角的cot权重计算每个顶点的权重。 2. 然后,使用cot权重对每个顶点的邻域点进行加权计算,得到平滑后的坐标。 3. 最后,根据平滑后的坐标更新每个顶点的位置。 修改后的代码如下: float smooth() { float err = -1; cogs.clear(); v_end = mesh.vertices_end(); //cot平滑 for (v_it = mesh.vertices_begin(); v_it != v_end; ++v_it) { cog[0] = cog[1] = cog[2] = weight_sum = 0.0; for (vv_it = mesh.vv_iter(*v_it); vv_it.is_valid(); ++vv_it) { double cot_weight = 0.0; MyMesh::HalfedgeHandle heh = mesh.find_halfedge(*v_it, *vv_it); if (!mesh.is_boundary(heh)) { MyMesh::HalfedgeHandle prev_heh = mesh.prev_halfedge_handle(heh); MyMesh::HalfedgeHandle next_heh = mesh.next_halfedge_handle(heh); MyMesh::VertexHandle prev_vh = mesh.to_vertex_handle(prev_heh); MyMesh::VertexHandle next_vh = mesh.to_vertex_handle(next_heh); MyMesh::Point prev_p = mesh.point(prev_vh); MyMesh::Point curr_p = mesh.point(*v_it); MyMesh::Point next_p = mesh.point(next_vh); double cot_alpha = cot(prev_p - curr_p, next_p - curr_p); double cot_beta = cot(curr_p - prev_p, next_p - prev_p); cot_weight = cot_alpha + cot_beta; } cog += cot_weight * mesh.point(*vv_it); weight_sum += cot_weight; } cogs.push_back(cog / weight_sum); } for (v_it = mesh.vertices_begin(), cog_it = cogs.begin(); v_it != v_end; ++v_it, ++cog_it) { if (!mesh.is_boundary(*v_it)) { MyMesh::Point p = mesh.point(*v_it); err = max(err, (p - *cog_it).norm()); mesh.set_point(*v_it, *cog_it); } } return err; } 其中cot函数的定义如下: double cot(MyMesh::Point a, MyMesh::Point b) { return dot(a, b) / cross(a, b).norm(); } 注意,这里使用的是边界角的cot权重,因此在计算cot权重时需要判断当前边是否为边界。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值