假期的宿舍(最大流Dinic)

题意:n后面第一行数据为0的是外校学生,为1的是本校学生,第二行为0则且第一行为1的为本校留校学生,第二行为1切第一行为1的为本校离校学生,每个学生可以睡自己(当学生为本校学生)和认识的人的床,问能否做到每个人(外校学生+本校留校学生)都有床睡。再下面的n * n矩阵为学生i和学生j的关系,1为朋友,0为陌生人。

分析:这是一个最大流问题,我们可以利用最大流算法计算出有多少张床能被人睡。

思路:

以2n+1为源点,2n+2为汇点,建图:

先以每张床为起点,汇点为终点建边,费用为1;

再以源点为起点,每个需要床睡觉的在校人员为终点建边,费用为1;

最后遍历n*n矩阵,若满足(xs[i][j]==1||i==j)则建一条由ij且费用为1的边。

统计在校学生数量,再对建好的图跑一遍最大流Dinic,若两者相等则所有不回家的本校学生和来看他们的其他人都有地方住,输出"^_^",反之则输出"T_T".

注意:此方法数组至少要开n*2+3.

AC代码:

/*
 author:wuzx
 */
 
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<utility>
#include<string>
#include<queue>
#include<map>
#include<set>
#include<iterator>
#include<iomanip>
#include<stack>
#include<cstdio>
#define ll long long
#define endl "\n"
using namespace std;
const int maxn=110;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
using namespace std;
int ren[maxn],xx[maxn];
int xs[maxn][maxn];
struct Edge{
	int from,to,cap,flow;
};
struct Dinic{
	int n,m,s,t;
	vector<Edge> edge;
	vector<int> g[maxn];
	bool vis[maxn];
	int d[maxn];
	int cur[maxn];
	
	void init(int n)
	{
		memset(d,0,sizeof d);
		for(int i=1;i<maxn;i++)
			g[i].clear();
		edge.clear();
	}
	
	void ClearFlow()
	{
		for(int i=0;i<edge.size();i++)
			edge[i].flow=0;
	}
	
	void add(int from,int to,int cap)
	{
		edge.push_back(Edge{from,to,cap,0});
		edge.push_back(Edge{to,from,0,0});
		m=edge.size();
		g[from].push_back(m-2);
		g[to].push_back(m-1);
	}
	
	bool bfs()
	{
		memset(vis,0,sizeof vis);
		queue<int> q;
		q.push(s);
		vis[s]=1;
		d[s]=0;
		while(!q.empty())
		{
			int x=q.front();
			q.pop();
			for(int i=0;i<g[x].size();i++)
			{
				Edge& e=edge[g[x][i]];
				if(!vis[e.to]&&e.cap>e.flow)
				{
					vis[e.to]=1;
					d[e.to]=d[x]+1;
					q.push(e.to);
				}	
			} 
		}
		return vis[t];
	}
	
	int dfs(int x,int a)
	{
		if(x==t||a==0)
			return a;
		int flow=0,f;
		for(int& i=cur[x];i<g[x].size();i++)
		{
			Edge& e=edge[g[x][i]];
			if(d[x]+1==d[e.to]&&(f=dfs(e.to,min(a,e.cap-e.flow)))>0)
			{
				e.flow+=f;
				edge[g[x][i]^1].flow-=f;
				flow+=f;
				a-=f;
				if(a==0)
					break;
			}
		}
		return flow;
	}
	
	int maxflow(int s,int t)
	{
		this->s=s;this->t=t;
		int flow=0;
		while(bfs())
		{
			memset(cur,0,sizeof cur);
			flow+=dfs(s,inf);
		}
		return flow;
	}
};
int main()
{
	ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
	int t; cin>>t;
	while(t--)
	{
		int n;
		cin>>n;
		Dinic slove;
		slove.init(n);
		int s=2*n+1,t=2*n+2;
		int ans=0;
		for(int i=1;i<=n;i++)
		{
			cin>>ren[i];
			if(ren[i]==1)
				slove.add(i+n,t,1);
			else
			{
				ans++;
				slove.add(s,i,1);
			}
		}
		for(int i=1;i<=n;i++)
		{
			cin>>xx[i];
			if(xx[i]==0&&ren[i]==1)
			{
				ans++;
				slove.add(s,i,1);
			}
		}
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				cin>>xs[i][j];
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
			{
				if(xs[i][j]==1||i==j)
					slove.add(i,j+n,1);
			}
		int l=slove.maxflow(s,t);
		cout<<(l==ans?"^_^":"T_T")<<endl;
	}
	return 0;
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值