[Java] SWEA 1767번 : [SW Test 샘플문제] 프로세서 연결하기

1. 문제

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV4suNtaXFEDFAUf 

SW Expert Academy

 

 

 

2. 풀이 과정

테두리에 있지 않는 각 프로세서마다 4방향 or 선택X 경우 합쳐 5가지 경우의 수가 있다.

기본적인 부분집합인 선택O or 선택X에서 선택지가 5개로 늘어났다고 볼 수 있는 것이다.

 

이 문제의 핵심은 부분집합과, 재귀에서의 복구하는 과정이다.

아래 코드에서는 연결한 부분을 2로 표시한 후, 재귀가 끝나고 다시 0으로 되돌려주고 다음 case를 진행하도록 했다.

 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.StringTokenizer;

public class SWEA1767 {
	static ArrayList<int[]> chips;
	static int map[][], maxChip, minLine;
	static int N, dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
	
	//살펴본 프로세서 수, 연결된 프로세서 수, 전선 길이
	public static void go(int cnt, int chipCnt, int lineCnt) {
		if(cnt==chips.size()) {
			if(maxChip < chipCnt) {
				maxChip = chipCnt;
				minLine = lineCnt;
			}
			else if(maxChip == chipCnt)
				minLine = Math.min(minLine, lineCnt);
			return;
		}
		
		//연결O
		for(int i=0;i<4;i++) {
			if(!isAvailable(chips.get(cnt)[0], chips.get(cnt)[1], i)) continue;
			
			go(cnt+1, chipCnt+1, lineCnt + setUp(chips.get(cnt)[0], chips.get(cnt)[1], i, 2));//전선 = 2
			setUp(chips.get(cnt)[0], chips.get(cnt)[1], i, 0);//복구
		}
		//연결X
		go(cnt+1, chipCnt, lineCnt);
	}
	
	// d 방향으로 map[x][y]에 있는 프로세서를 연결할 수 있는지 확인
	public static boolean isAvailable(int x, int y, int d) {
		while(true) {
			x += dx[d];
			y += dy[d];
			if(0<=x && x<N && 0<=y && y<N) {
				if(map[x][y]>0) {
					return false;
				}
			}
			else {
				return true;
			}	
		}
	}
	
	//2 = 연결하기(전선)
	//0 = 연결해제(빈칸)
	public static int setUp(int x, int y, int d, int s) {
		int cnt = 0;
		while(true) {
			x += dx[d];
			y += dy[d];
			if(0<=x && x<N && 0<=y && y<N) {
				map[x][y] = s;
				cnt++;//연결한 전선 길이
			}
			else {
				break;
			}
		}
		return cnt;
	}

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		StringTokenizer st = null;
		int T = Integer.parseInt(br.readLine());
		for(int t=1; t<=T; t++) {
			maxChip = 0;
			minLine = 999_999_999;
			N = Integer.parseInt(br.readLine());
			map = new int[N][N];
			chips = new ArrayList<>();
			
			for(int i=0;i<N;i++) {
				st = new StringTokenizer(br.readLine());
				for(int j=0;j<N;j++) {
					map[i][j] = Integer.parseInt(st.nextToken());
					//테두리에 위치하지 않은 프로세서만 저장(테두리에 있는 건 이미 연결 되었으므로)
					if(map[i][j]==1 && i!=0 && i!=N-1 && j!=0 && j!=N-1) {
						chips.add(new int[] {i, j});
					}
				}
			}
			go(0, 0, 0);
			sb.append("#").append(t).append(" ").append(minLine).append("\n");
		}
		System.out.println(sb.toString());
	}
}