[SW Expert Academy] - (D2)1979. 어디에 단어가 들어갈 수 있을까

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV5PuPq6AaQDFAUq&categoryId=AV5PuPq6AaQDFAUq&categoryType=CODE

SW Expert Academy

코드

import java.util.Scanner;

public class Solution {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int t = scan.nextInt();
		for(int tc=1; tc<=t; tc++) {
			int N = scan.nextInt();
			int K = scan.nextInt();
			
			int[][] puzzle = new int[N][N];
			int ans = 0;
			
			for(int i=0; i<N; i++) {
				for(int j=0; j<N; j++) {
					puzzle[i][j] = scan.nextInt();
				}
			}
			
			// 가로부터 들어갈 수 있는곳 찾기(흰색: 1, 검은색: 0)
			for(int i=0; i<N; i++) {
				int colCheck = 0;
				for(int j=0; j<N; j++) {
					
					// 검은색이면 0으로 초기화, 흰색이면 +1
					if(puzzle[i][j] == 0) {
						
						// 검은색일때, 앞에 흰칸이 K개인경우 -> 단어가 올 수 있음
						if(colCheck == K) {
							ans ++;
						}
						colCheck = 0;
					} else {
						colCheck += 1;
					}
				}
				
				// 퍼즐의 가로가 길이 K와 맞을경우 +1
				if(colCheck == K) {
					ans ++;
				}
			}
			
			// 세로에 들어갈 수 있는 자리 찾기
			for(int i=0; i<N; i++) {
				int rowCheck = 0;
				for(int j=0; j<N; j++) {
					
					// 검은색이면 0으로 초기화, 흰색이면 +1
					if(puzzle[j][i] == 0) {
						
						// 검은색일때, 앞에 흰칸이 K개인경우 -> 단어가 올 수 있음
						if(rowCheck == K) {
							ans ++;
						}
						rowCheck = 0;
					} else {
						rowCheck += 1;
					}
				}
				
				// 퍼즐의 세로가 길이 K와 맞을경우 +1
				if(rowCheck == K) {
					ans ++;
				}
			}
			System.out.println("#" + tc + " " + ans);
		}
		
		scan.close();
	}

}

풀이

2차원 N*N 배열에서 검은색은 0이고, 흰색은 1이다. 

단어의 길이가 K인데, 흰색의 갯수(1의 갯수)와 단어의 길이가 동일하면 그 단어가 들어갈 수 있다.

그리고, 가로 세로 검사를 해주어야 한다.

 

한가지 조심할점은, 위 사진의 가장 아랫줄과 같이 검은색이 나왔을때, 흰색의 갯수가 K와 동일하면 정답으로 쳐주어야 한다. 위 코드는 아래와 같다.

			// 검은색이면 0으로 초기화, 흰색이면 +1
			if(puzzle[i][j] == 0) {
						
				// 검은색일때, 앞에 흰칸이 K개인경우 -> 단어가 올 수 있음
				if(colCheck == K) {
					ans ++;
				}

위 조건만 검사한 후 검은색이라면 무조건 흰색의 갯수는 0으로 초기화 시켜주고, 흰색이라면 갯수를 +1 해주면 된다.