洛谷P2055 [ZJOI2009] 假期的宿舍 [网络流]

  题目传送门

假期的宿舍

题目描述

学校放假了 · · · · · · 有些同学回家了,而有些同学则有以前的好朋友来探访,那么住宿就是一个问题。比如 A 和 B 都是学校的学生,A 要回家,而 C 来看B,C 与 A 不认识。我们假设每个人只能睡和自己直接认识的人的床。那么一个解决方案就是 B 睡 A 的床而 C 睡 B 的床。而实际情况可能非常复杂,有的人可能认识好多在校学生,在校学生之间也不一定都互相认识。我们已知一共有 n 个人,并且知道其中每个人是不是本校学生,也知道每个本校学生是否回家。问是否存在一个方案使得所有不回家的本校学生和来看他们的其他人都有地方住。

输入输出格式

输入格式:

 

第一行一个数 T 表示数据组数。接下来 T 组数据,每组数据第一行一个数n 表示涉及到的总人数。接下来一行 n 个数,第 i 个数表示第 i 个人是否是在校学生 (0 表示不是,1 表示是)。再接下来一行 n 个数,第 i 个数表示第 i 个人是否回家 (0 表示不回家,1 表示回家,注意如果第 i 个人不是在校学生,那么这个位置上的数是一个随机的数,你应该在读入以后忽略它)。接下来 n 行每行 n 个数,第 i 行第 j 个数表示 i 和 j 是否认识 (1 表示认识,0 表示不认识,第 i 行 i 个的值为 0,但是显然自己还是可以睡自己的床),认识的关系是相互的。

 

输出格式:

 

对于每组数据,如果存在一个方案则输出 “^_^”(不含引号) 否则输出“T_T”(不含引号)。(注意输出的都是半角字符,即三个符号的 ASCII 码分别为94,84,95)

 

输入输出样例

输入样例#1: 复制
1
3
1 1 0
0 1 0
0 1 1
1 0 0
1 0 0
输出样例#1: 复制
^_^

说明

对于 30% 的数据满足 1 ≤ n ≤ 12。

对于 100% 的数据满足 1 ≤ n ≤ 50,1 ≤ T ≤ 20。

多组数据!!!


  分析:

  其实也并不是很难,接近于网络流模板了。

  建图很并不难想到,设定超级源点和超级终点后,将所有不回去的在校学生和外来学生与源点连边,然后将所有本校学生与终点连边,再直接跑网络流即可。注意多组数据,要把数据清零(忘了清零数据导致蒟蒻卡了好久。。。真蠢)。

  Code:

 1 //It is made by HolseLee on 5th June 2018
 2 //Luogu.org P2055
 3 #include<bits/stdc++.h>
 4 using namespace std;
 5 const int N=5001;
 6 const int inf=2147483647;
 7 int stu[N],back[N],f[N][N];
 8 int n,ans,tot,head[N],cnt=1;
 9 int T,sta,endd,dfn[N],cur[N];
10 struct Node{
11   int to,val,next;
12 }edge[N<<3];
13 queue<int>team;
14 inline void add(int x,int y,int z)
15 {
16   edge[++cnt].to=y;
17   edge[cnt].val=z;
18   edge[cnt].next=head[x];
19   head[x]=cnt;
20 }
21 inline bool bfs()
22 {
23   memset(dfn,0,sizeof(dfn));
24   while(!team.empty())team.pop();
25   dfn[sta]=1;team.push(sta);
26   while(!team.empty()){
27     int u=team.front();team.pop();
28     for(int i=head[u];i!=-1;i=edge[i].next){
29       int v=edge[i].to;
30       if((dfn[v]==0)&&(edge[i].val>0)){
31     dfn[v]=dfn[u]+1;team.push(v);}
32     }
33   }
34   return (dfn[endd]>0);
35 }
36 inline int dfs(int u,int flow)
37 {
38   if( u==endd)return flow;
39   int used=0;
40   for(int i=head[u];i!=-1;i=edge[i].next){
41     int v=edge[i].to;
42     if(dfn[v]==dfn[u]+1&&edge[i].val>0){
43       int ka=dfs(v,min(flow,edge[i].val));
44       if(ka>0){used+=ka;flow-=ka;
45     edge[i].val-=ka;
46     edge[i^1].val+=ka;
47     if(flow==0)return used;
48       }
49     }
50   }
51   return used;
52 }
53 inline void work()
54 {
55   cin>>n;sta=0;endd=2*n+1;ans=0;tot=0;
56   memset(head,-1,sizeof(head));
57   for(int i=1;i<=n;i++){
58     cin>>stu[i];
59     if(stu[i]){
60       add(i+n,endd,1);add(endd,i+n,0);}}
61   for(int i=1;i<=n;i++){
62     cin>>back[i];
63     if((back[i]==0&&stu[i]==1)||(stu[i]==0)){
64       tot++;add(sta,i,1);add(i,sta,0);}}
65   for(int i=1;i<=n;i++)
66     for(int j=1;j<=n;j++){
67       cin>>f[i][j];
68       if(i==j||f[i][j]==1){
69     add(i,j+n,1);add(j+n,i,0);}}
70   while(bfs()){
71     while(int ka=dfs(sta,inf))ans+=ka;}
72   if(ans>=tot)cout<<"^_^"<<"\n";
73   else cout<<"T_T"<<"\n";
74 }
75 int main()
76 {
77   ios::sync_with_stdio(false);
78   cin>>T;while(T--)work();
79   return 0;
80 }

 

猜你喜欢

转载自www.cnblogs.com/cytus/p/9139283.html