Q 2667 단지번호 붙이기 JAVA

dfs 를 어떻게 쓰는건가 고민했었는데 구글링을 하다보니 dfs를 아래 위 양 옆 이렇게 네 방향으로 사용하는 문제인 것을 알게 되었다.

package graph;
import java.util.*;

public class Q2667 {
	static int[][]arr;
	static int[][]visit;
	static int[] di = {-1,0,1,0};
	static int[] dj = {0 ,-1,0,1};
	static int num, cnt;
	public static void main(String[] args) {
		Scanner scan =new Scanner(System.in);
		num = Integer.parseInt(scan.nextLine());
		arr = new int[num][num];
		visit = new int[num][num];
		ArrayList cnts = new ArrayList<Integer>();
		
		
		for(int i=0; i<num; i++) {
			String s = new String();
			s= scan.nextLine();
			for(int j=0; j<num; j++) {
				if(s.charAt(j)=='0')
					arr[i][j]=0;
				else
					arr[i][j]=1;
			}
		}
		
		for(int i=0; i<num; i++) {
			for(int j=0; j<num; j++) {
				if(arr[i][j]==1 && visit[i][j]==0) {
					cnt=0;
					find(i,j);
					cnts.add(cnt);
				}
			}
		}
		Collections.sort(cnts);
		System.out.println(cnts.size());
		for(int i=0; i<cnts.size(); i++) {
			System.out.println(cnts.get(i));
		}
		

	}
	//방문했다면 find =1
	public static void find(int i, int j) {
		
		visit[i][j]=1;
		cnt++;
		for(int k=0; k<4; k++) {
			int ni = i + di[k];
			int nj = j + dj[k];
			
			if(ni>=0 && nj>=0 && ni<num && nj<num && arr[ni][nj]==1 && visit[ni][nj]==0) {
				find(ni, nj);
			}
		}
		
		
		
	}
	
	
	
}