CODE[VS]1033 蚯蚓的游戏问题

最新推荐文章于 2020-10-03 21:58:14 发布

ssl_lyy

最新推荐文章于 2020-10-03 21:58:14 发布

阅读量473

点赞数

分类专栏：费用流

本文链接：https://blog.csdn.net/ssl_lyy/article/details/69935897

版权

费用流专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目描述 Description

在一块梯形田地上，一群蚯蚓在做收集食物游戏。蚯蚓们把梯形田地上的食物堆积整理如下：
 a(1,1)  a(1,2)…a(1,m)
 a(2,1)  a(2,2)  a(2,3)…a(2,m)  a(2,m+1)                        
 a(3,1)  a (3,2)  a(3,3)…a(3,m+1)  a(3,m+2)
 ……  
 a(n,1)   a(n,2)   a(n,3)…           a(n,m+n-1)     

   它们把食物分成n行，第1行有m堆的食物，每堆的食物量分别是a(1,1),a(1,2),…,a(1,m)；
第2行有m+1堆食物，每堆的食物量分别是a(2,1),a(2,2),…,  a(2,m+1)；以下依次有m+2堆、m+3堆、…m+n-1堆食物。
现在蚯蚓们选择了k条蚯蚓来测试它们的合作能力（1≤ k ≤m）。测试法如下：第1条蚯蚓从第1行选择一堆食物，然后往左下或右下爬，并收集1堆食物，例如从a（1,2）只能爬向a(2,2) 或a(2,3)，而不能爬向其它地方。接下来再爬向下一行收集一堆食物，直到第n行收集一堆食物。第1条蚯蚓所收集到的食物量是它在每一行所收集的食物量之和；第2条蚯蚓也从第1行爬到第n行，每行收集一堆食物，爬的方法与第1条蚯蚓相类似，但不能碰到第1条蚯蚓所爬的轨迹；一般地，第i 条蚯蚓从第1行爬到第 n行，每行收集一堆食物，爬的方法与第1条蚯蚓类似，但不能碰到前 I-1 条蚯蚓所爬的轨迹。这k条蚯蚓应该如何合作，才能使它们所收集到的食物总量最多？收集到的食物总量可代表这k条蚯蚓的合作水平。

Ø编程任务：

   给定上述梯形m、n和k的值（1≤k≤m≤30;1≤n≤30）以及梯形中每堆食物的量（小于10的非整数），编程计算这k条蚯蚓所能收集到的食物的最多总量。

输入描述 InputDescription

   输入数据由文件名为INPUT1.*的文本文件提供，共有n+1行。每行的两个数据之间用一个空格隔开。
第1行是n、m和k的值。
接下来的n行依次是梯形的每一行的食物量a(i,1)，a(i,2)，…，a(i,m+i-1)，i=1,2,…,n。

输出描述 Output Description

程序运行结束时，在屏幕上输出k蚯蚓条所能收集到的食物的最多总量。

样例输入 Sample Input

3    2   2    
1   2
5   0   2
1   10  0  6

样例输出 Sample Output

解题思路

比较显然是费用流，费用流模板题。
● 建图
    1.另外加入源点S，汇点T，以及一个终点en（用T连en的边控制只有k条蚯蚓走）
    2.每队食物用两个点表示，其费用刚好为食物量（以确保每堆食物只被吃一次）
    3.从S向第一行的每个点的前一个连边；由每行的后一个点向下一行的能走到的点前一个点连边；最后一行的点向T连边。
    4.T向en连一条流量为k的边，其余流量都为1，反向弧为相反数。
●费用流
    每次用spfa找到到en的最长路径，并记录下来，修改这条路径上的流量，并把费用加入ans，直到没有增广路。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;

#define N 200
#define M 10000

const int INF=0x7fffffff;

struct note
{
    int x,y,ne,c,w;
};

note side[M];
int last [N*N],dis[N*N],pre[N*N],tt[N*N],a[N][N],b[N][N];
int s,t,n,m,k,l=1,en,ans;

void add(int x,int y,int w,int c)
{
    l++;
    side[l].x=x;
    side[l].y=y;
    side[l].w=w;
    side[l].c=c;
    side[l].ne=last[x];
    last[x]=l;
    //printf("%d %d %d %d %d %d\n",l,x,y,w,c,side[l].ne);
}

void addside(int x,int y,int c)
{
    add(x,y,1,c);
    add(y,x,0,-c);
}

void init()
{
    scanf("%d%d%d",&n,&m,&k);
    int p=1;
    memset(last,0,sizeof(last));
    for (int i=1;i<=n;++i)
      for (int j=1;j<=i+m-1;++j)
      {
        scanf("%d",&a[i][j]);
        b[i][j]=p;
        addside(p,p+1,a[i][j]);
        p+=2;
        //printf("%d %d %d\n",i,j,b[i][j]);
      }
    s=0;
    t=p;
    en=t+1;
    for (int i=1;i<=m;++i) addside(s,b[1][i],0);
    for (int i=1;i<=n-1;++i)
      for (int j=1;j<=i+m-1;++j)
        {
          addside(b[i][j]+1,b[i+1][j],0);
            addside(b[i][j]+1,b[i+1][j+1],0);
        }
    for (int i=1;i<=n+m-1;++i) addside(b[n][i]+1,t,0);
    add(t,en,k,0);
    add(en,t,-k,0);
}

bool spfa()
{
    for (int i=s;i<=en;++i)
    {
        dis[i]=-INF;
    pre[i]=-1;
        tt[i]=0;
    }
    queue<int> Q;
    dis[s]=0;
    tt[s]=1;
    Q.push(s);
    while (!Q.empty())
    {
        int now=Q.front();
        Q.pop();
        for (int i=last[now];i!=0;i=side[i].ne)
        {
            int j=side[i].y;
            if (side[i].w>0 && dis[j]<dis[now]+side[i].c)
            {
                dis[j]=dis[now]+side[i].c;
                pre[j]=i;
                if (tt[j]==0)
                {
                    tt[j]=1;
                    Q.push(j);
                }
            }
          //printf("%d %d %d %d %d %d\n",now,j,side[i].c,side[i].w,dis[now],dis[j]);
        }
      tt[now]=0;
    }
    if (dis[en]==-INF) return false;
      else return true;
}

void mcf()
{
    ans+=dis[en];
    for (int i=pre[en];;i=pre[side[i].x])
    {
        side[i].w-=1;
        side[i^1].w+=1;
        //printf("%d ",side[i].x);
        if (side[i].x==s) break;
  }
}

int main()
{
    init();
    ans=0;
    while (spfa()) mcf();
    printf("%d\n",ans);
    return 0;
}