蓝桥杯 历届试题 合根植物java(dfs)

问题描述

　　w星球的一个种植园，被分成 m * n 个小格子（东西方向m行，南北方向n列）。每个格子里种了一株合根植物。
　　这种植物有个特点，它的根可能会沿着南北或东西方向伸展，从而与另一个格子的植物合成为一体。

　　如果我们告诉你哪些小格子间出现了连根现象，你能说出这个园中一共有多少株合根植物吗？

输入格式

　　第一行，两个整数m，n，用空格分开，表示格子的行数、列数（1<m,n<1000）。
　　接下来一行，一个整数k，表示下面还有k行数据(0<k<100000)
　　接下来k行，第行两个整数a，b，表示编号为a的小格子和编号为b的小格子合根了。

　　格子的编号一行一行，从上到下，从左到右编号。
　　比如：5 * 4 的小格子，编号：
　　1 2 3 4
　　5 6 7 8
　　9 10 11 12
　　13 14 15 16
　　17 18 19 20

样例输入

5 4
16
2 3
1 5
5 9
4 8
7 8
9 10
10 11
11 12
10 14
12 16
14 18
17 18
15 19
19 20
9 13
13 17

样例输出

5

样例说明

　　其合根情况参考下图

     

 

 

时间限制：2.0s  

内存限制：256.0MB

import java.beans.Visibility;
import java.util.Scanner;

public class Main
{
    static int k;
    static int [][] art = new int[1000][2];
    static int [][] maze = new int[1000][1000];
    static boolean vis[][] = new boolean[1000][1000];
    static int m ;
    static int n;
    static int count = 0;//记录合根数量
    
    public static void print()
    {
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                System.out.print(maze[i][j]+" ");
            }
            System.out.println();
        }
    }
    
    public static void dfs(int x,int y)
    {
        vis[x][y] = true;
        for(int i=0;i<k;i++)
        {
            if(art[i][0]==maze[x][y]||art[i][1]==maze[x][y])//代表该点有合根，合根值为art[i][1];
            {
                if(art[i][0]==maze[x][y])
                {
                    int x1 = (art[i][1]-1)/4;
                    int y1 = (art[i][1]+3)%4;
                    if(!vis[x1][y1])//该点没有被标记
                    {
                        vis[x1][y1] = true;//将合根标记

                        dfs(x1, y1);
                    }    
                }
                else
                {
                    int x2 = art[i][0]/4;
                    int y2 = (art[i][0]+3)%4;
                    if(!vis[x2][y2])//前面一点未标记
                    {
                        vis[x2][y2] = true;
            
                        dfs(x2, y2);
                    }
                }
                
                
            }
            
        }

    }
    
    public static void main(String[] args)
    {
        // TODO Auto-generated method stub
        Scanner scanner = new Scanner(System.in);
        m = scanner.nextInt();
        n = scanner.nextInt();
        k = scanner.nextInt();
        for(int i=0;i<k;i++)
        {
            art[i][0] = scanner.nextInt();
            art[i][1] = scanner.nextInt();
        }
        
        int sum = 1;
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                maze[i][j] = sum;
                sum++;
            }
        }
    
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(!vis[i][j])//没有合根
                {
                    count++;
                    dfs(i,j);
                }
            }
        }
        System.out.print(count);
    }

}