信息院赛-小红灌溉 bfs

Description

小红家有n行田地，每行田地由m块大小为1*1的小田地组成。这些田地构成了一个n*m的矩形。小红在某些小田地里种植了蔬菜。现在小红要给蔬菜灌溉。小红每次可以选择给某一行或某一列的田地灌溉。但是一块种植了蔬菜的田地只能被灌溉一次。由于种种原因，灌溉某行或某列的耗电量不同。所以小红想知道，灌溉完所有蔬菜的耗电量最少为多少？

Input

第一行输入一个数字T，表示输入数据组数。接下来每组数据：第一行输入两个数n，m（1<=n<=100,1<=m<=100）接下来输入n行，每行m个数字。第i行的第j个数为0表示第i行的第j块田地为空地，如果为1，表示第i行的第j列种植有蔬菜。接下来一行输入n个数字，第i个数字，表示灌溉第i行的耗电量。(每个数字大于0小于等于1000)接下来一行输入m个数字，第i个数字，表示灌溉第i列的耗电量。（每个数字大于0小于等于1000）

Output

输出T行，第i行输出"Case #i: y"，y为第i组数据的答案。

• Sample Input
• Raw

12 21 00 11 22 1

• Sample Output
• Raw

Case #1: 2

这道题是我在看完啦啦操之后突然灵光一下想起来怎么做的

哈哈哈哈

因为这里要么只能浇灌行，要么只能浇灌列，所以我们先将图以行列二分，然后将图用邻接表存起来

因为这里是只能选择行或者列，所以我们只需要遍历图的联通块，然后选择行或者列之和的最小值就是了

#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<vector>
#include<algorithm>
#include<queue>
#include<stack>
#define nn 1005
#define inff 0x3fffffff
#define mod 1000000007
#define eps 1e-9
using namespace std;
int tu[nn][nn];
struct node
{
    int en;
    int next;
}ma[nn];
int p[nn];
int num;
void init()
{
    memset(p,-1,sizeof(p));
    num=0;
}
void add(int st,int en)
{
    ma[num].en=en;
    ma[num].next=p[st];
    p[st]=num++;
}
int sumcol;
int sumrow;

int n,m;
int col[nn];
int row[nn];
int tot;
queue<int> que;
void bfs()
{
    int use[nn];
    memset(use,false,sizeof(use));
    int a;
    int b;
    tot=0;
    int i;
    for(i=1;i<=n;i++)
    {
        sumcol=0;
        sumrow=0;
        if(use[i]==false)
        {
          a=i;
          if(i<=100)
          {
              sumcol+=col[i];
          }
          else
            sumrow+=row[i-100];
        }
          que.push(a);
          use[a]=true;
          while(!que.empty())
          {
             b=que.front();
             que.pop();
             if(use[b]==false)
             {
                 use[b]=true;
                 if(b<=100)
                {
                  sumcol+=col[b];
                }
                else
                sumrow+=row[b-100];
             }
             for(int j=p[b];j!=-1;j=ma[j].next)
             {
                 if(use[ma[j].en]==false)
                 que.push(ma[j].en);
             }
          }
          tot+=min(sumcol,sumrow);
          sumcol=sumrow=0;
    }
}
int main()
{
    int t;
    int i,j;
    while(scanf("%d",&t)!=EOF)
    {
        for(int o=1;o<=t;o++)
        {
            scanf("%d %d",&n,&m);
            init();
            sumcol=sumrow=0;
            for(i=1;i<=n;i++)
                for(j=1;j<=m;j++)
                scanf("%d",&tu[i][j]);
            for(i=1;i<=n;i++)
                scanf("%d",&col[i]);
            for(i=1;i<=m;i++)
                scanf("%d",&row[i]);
            for(i=1;i<=n;i++)
            {
                for(j=1;j<=m;j++)
                {
                   if(tu[i][j]==1)
                   {
                       add(i,j+100);
                       add(j+100,i);
                   }
                }
            }
            bfs();
            printf("Case #%d: %d\n",o,tot);
        }
    }
}