DFS 相关例题

会顶科技   两道

第一行为一个整数N，用来表示球的个数
第二行为一个整数M，用来表示桶的个数
从第三行到第N+2行，每行有M个整数，这些整数要么为0，要么为1
输出描述：
输出为一个整数，输出1代表可以放入所有的球，输出0代表无法放入所有的球。

示例1
输入
4
6
0 1 0 1 0 0
1 0 0 1 0 0
0 0 0 1 1 0
1 1 1 0 0 1
输出
1

#include <stdio.h>
#include <stdlib.h>
#include <iostream>
//typedef int bool;
using namespace std;
bool BucketCore(int*arr, int M, int N, int m,int& ballnum, bool* visisted)
{
	if (ballnum == N)
		return true;
		
	bool result = false;
	for (int i = 0; i < N; i++)
	{
		if (m < M &&m >= 0 && !visisted[i])
		{
			int l = *(arr + i*M + m);
			if(*(arr+i*M+m))
			{
				ballnum++;
				visisted[i] = true;
				result = BucketCore(arr, M, N, m + 1, ballnum, visisted);
				if (!result)
				{
					ballnum--;
					visisted[i] = false;
				}
				else
					return result;
			}
			else if (i == N)
			{
				result = BucketCore(arr, M, N, m + 1, ballnum, visisted);
				if (!result)
				{
					ballnum--;
					visisted[i] = false;
				}
				else
					return result;
			}
		}
	}
	return false;
}
bool Bucket(int* arr,int M,int N)
{
	if (!arr || M < 1 || N < 1)
		return false;
	bool* visisted = new bool[N];
	int ballnum = 0;
	memset(visisted, 0, N);
	if (BucketCore(arr, M, N, 0, ballnum, visisted))
	{
		printf("cunzai");
		return true;
	}
}

int main()
{
	int M=6, N=4;
	int arr[4][6] = { {0,1,0,1,0,0},{1,0,0,1,0,0},{0,0,0,1,1,0},{1,1,1,0,0,1} };
	Bucket(&arr[0][0], 6, 4);
	while (1);
	return 0;
}

在一片区域上有一些大小不同的水塘，区域大小是M*N的矩阵，其中1代表水，0代表土地。规定水塘是由上下左右的水
相连组成的，对角的水不能连成水糖。请根据给出的矩阵计算水塘的个数，
输入描述：
输入第一行是矩阵大小M和N（如3 3代表3x3的矩阵），
第二行是矩阵的第一行数据（如0 1 0），以此类推。
输出描述：
输进水塘个数
示例1

输入

3 3
0 1 0
1 0 1
0 1 0

输出
4