【POJ1698】Alice's Chance-二分图多重匹配

测试地址:Alice's Chance

题目大意:演员Alice接到N个拍电影任务,一部电影在一周中只有某些日子能拍,并且每部电影都有一定的任务(至少要拍D天)和期限(在W周内拍完),而Alice一天只能去拍一部电影,求能不能制定一个排片计划,使得Alice能够接下所有拍电影的任务。

做法:从题目中可以看出,电影与每一天之间是一对多的匹配关系,因此可以用二分图模型中的多重匹配来解决,代码写起来很像匈牙利算法。

以下是本人代码:

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int T,n,limit[25],able[10],bmap[25][510],maxw;
int cx[510],cy[25][510];
bool vis[25];

bool findpath(int u)
{
  for(int i=1;i<=n;i++)
  {
    if (bmap[i][u]&&!vis[i])
	{
	  vis[i]=1;
	  if (cy[i][0]<limit[i])
	  {
	    cy[i][++cy[i][0]]=u;
		return 1;
	  }
	  else
	  {
	    for(int j=1;j<=cy[i][0];j++)
		  if (findpath(cy[i][j]))
		  {
		    cy[i][j]=u;
		    return 1;
	      }
	  }
	}
  }
  return 0;
}

bool mulmatch()
{
  int ans=0,s=0;
  memset(cx,0,sizeof(cx));
  memset(cy,0,sizeof(cy));
  for(int i=1;i<=maxw*7;i++)
  {
    memset(vis,0,sizeof(vis));
    if (!cx[i])
	{
	  if (findpath(i)) ans++;
    }
  }
  for(int i=1;i<=n;i++) s+=limit[i];
  return ans==s;
}

int main()
{
  scanf("%d",&T);
  while(T--)
  {
    memset(bmap,0,sizeof(bmap));
    scanf("%d",&n);
	maxw=0;
	for(int i=1,w;i<=n;i++)
	{
	  memset(able,0,sizeof(able));
	  for(int j=1;j<=7;j++) scanf("%d",&able[j]);
	  scanf("%d%d",&limit[i],&w);
	  maxw=max(w,maxw);
	  for(int j=1;j<=7;j++)
	    if (able[j])
		{
	      for(int k=0;k<w;k++)
		    bmap[i][k*7+j]=1;
	    }
	}
	
	if (mulmatch()) printf("Yes\n");
	else printf("No\n");
  }
  
  return 0;
}


posted @ 2017-03-07 20:10  Maxwei_wzj  阅读(81)  评论(0编辑  收藏  举报