dinic80分求助
查看原帖
dinic80分求助
182130
watasky楼主2022/7/13 14:14
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int n,m,S,T;
const int N=30000,M=100010,inf=1e8;
int e[M],ne[M],f[M];
int q[N],h[N],cur[N],d[N],idx; 
void add(int a,int b,int c)
{
	e[idx]=b,f[idx]=c,ne[idx]=h[a],h[a]=idx++;
	e[idx]=a,f[idx]=0,ne[idx]=h[b],h[b]=idx++;
}
bool bfs()
{
	int tt=0,hh=0;
	memset(d,-1,sizeof d);
	q[0]=S,d[S]=0,cur[S]=h[S];
	while(hh<=tt)
	{
		int t=q[hh++];
		for(int i=h[t];~i;i=ne[i])
		{
			int ver=e[i];
			if(d[ver]==-1&&f[i])
			{
				d[ver]=d[t]+1;
				cur[ver]=h[ver];
				if(ver==T) return true;
				q[++tt]=ver;
			}
		}
	}
	return false;
}
int find(int u,int limits)
{
	if(u==T) return limits;
	int flow=0;
	for(int i=cur[u];~i&&flow<limits;i=ne[i])
	{
		int ver=e[i];
		cur[u]=i;
		if(d[ver]==d[u]+1&&f[i])
		{
			int t=find(ver,min(f[i],limits-flow));
			if(!t) d[ver]=-1;
			f[i]-=t;
			f[i^1]+=t;
			flow+=t;
		}
	}
	return flow;
}
int dinic()
{
	int ans=0,flow;
	while(bfs())
	{
		while(flow=find(S,inf)) ans+=flow;
	}
	return ans;
}
int t,is[100];
int main()
{
    cin>>t;
    S=0,T=2*n+1;
    while(t--)
    {
        cin>>n;
        memset(h,-1,sizeof h);
        for(int i=1;i<=n;i++)
        {
            cin>>is[i];
            if(is[i])//只要是学生就把自己的床向汇点建边
            {
                add(i+n,T,1);//床向汇点建边
            }
        }
        int tot=0;
        for(int i=1;i<=n;i++)
        {
            int x;
            cin>>x;
            if(is[i])
            {
                if(x==0)//是学生且不回家
                {
                    add(S,i,1);//源点向不回家学生建边
                    add(i,i+n,1);//自己可以睡自己的床
                    tot++;
                }
                
            }
            else//不是学生
            {
                add(S,i,1);//源点向外来的亲戚朋友建边
                tot++;
            }
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                int t;
                cin>>t;
                if(t)//如果认识
                {
                    if(is[i]&&!is[j]) add(j,i+n,1);//i是学生j不是,j可以睡i的床
                    else if(!is[i]&&is[j])add(i,j+n,1);//j是学生i不是,i可以睡j的床
                    else if(is[i]&&is[j]) add(i,j+n,1),add(j,i+n,1);//可以相互睡
                }
            }
        }
        if(tot>dinic())printf("T_T\n");
        else printf("^_^\n");
        
        
    }
    
    return 0;
}
2022/7/13 14:14
加载中...