poj3281 DINIC

2024-04-28 17:38

文章标签 poj3281 dinic

本文主要是介绍poj3281 DINIC，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

如题：http://poj.org/problem?id=3281

初学最大流.

这题重要的是图的构建。源点-》食物-》牛-》牛-》饮料-》汇点。0是源点，1-f食物，f+1-f+1+n食物到牛，f+1+n-f+1+2*n牛到牛，f+1+2*n-f+1+2*n+d牛到饮料,f+1+2*n+d+1最终汇点。

按照上面建图，用DINIC算法求最大流即可.

#include<iostream>
using namespace std;
#include<queue>
#define min(a,b)(a<b?a:b)
#define INF INT_MAX
#define N 410

int map[N][N];
int n,sink,maxflow,source,f,d;
int dis[N];
int bfs()
{
queue<int>QUE;
int k;
int i;
memset(dis,-1,sizeof(dis));
//dis[sink]=0;
//QUE.push(sink);
  dis[source]=0;
  QUE.push(source);
while(!QUE.empty())
{
  k=QUE.front();
  QUE.pop();
  for(i=0;i<n;i++)
  {
  // if(dis[i]==-1&&map[i][k])
   if(dis[i]==-1&&map[k][i])
   {
    dis[i]=dis[k]+1;
    QUE.push(i);
   }
  }
//if(k==source) return 1;
  if(k==sink) return 1;
}
return 0;
}

int dfs(int cur,int cp)
{
if(cur==sink) return cp;
int tmp=cp,t;
for(int i=0;i<n&&tmp;i++)
{
// if(dis[i]+1==dis[cur]&&map[cur][i])
  if(dis[i]==dis[cur]+1&&map[cur][i])
  {
   t=dfs(i,min(map[cur][i],tmp));
   map[cur][i]-=t;
   map[i][cur]+=t;
   tmp-=t;
  }
}
return cp-tmp;
}
void dinic()
{
maxflow=0;
while(bfs())
  maxflow+=dfs(source,INF);
}
int main()
{
int i,j;
int sum_f,sum_d,tmp;
freopen("C:\\1.txt","r",stdin);
while(~scanf("%d%d%d",&n,&f,&d))
{
  memset(map,0,sizeof(map));
  source=0,sink=2*n+f+1+d;
  for(i=1;i<=f;i++)
   map[source][i]=1;
  for(i=1;i<=d;i++)
   map[f+2*n+i][sink]=1;
  for(i=1;i<=n;i++)
   map[f+i][f+i+n]=1;
  for(i=1;i<=n;i++)
  {
   scanf("%d%d",&sum_f,&sum_d);
   for(j=1;j<=sum_f;j++)
   {
    scanf("%d",&tmp);
    map[tmp][f+i]=1;
   }
   for(j=1;j<=sum_d;j++)
   {
    scanf("%d",&tmp);
    map[f+n+i][f+2*n+tmp]=1;
   }
  }
  n=sink+1;
  dinic();
  printf("%d\n",maxflow);
}
}