알고리즘/DP

[백준 1010] 다리 놓기 (Java 풀이)

파프리카. 2023. 8. 25. 14:47

https://www.acmicpc.net/problem/1010

 

1010번: 다리 놓기

입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트케이스에 대해 강의 서쪽과 동쪽에 있는 사이트의 개수 정수 N, M (0 < N ≤ M < 30)이 주어진다.

www.acmicpc.net

풀이 코드 1

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


public class Main {
	static int[][] dp = new int[30][30];
	
	// 조합 함수
	static int combi(int n, int r) {
		if (dp[n][r] > 0) {
			return dp[n][r];
		}
		if (n == r || r == 0) {
			return dp[n][r] = 1;
		}
		return dp[n][r] = combi(n - 1, r - 1) + combi(n - 1, r);
	}

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		int T = Integer.parseInt(bf.readLine());
		StringBuilder sb = new StringBuilder();
		for (int tc = 1; tc <= T; tc++) {
			String s = bf.readLine();
			StringTokenizer st = new StringTokenizer(s);
			int N = Integer.parseInt(st.nextToken());
			int M = Integer.parseInt(st.nextToken());
			// M개 중에 N개 뽑는 경우의 수 = mCn
			sb.append(combi(M,N)).append('\n');
		}
		System.out.println(sb);
	}
}

다리끼리는 서로 겹칠 수 없기 때문에 이 문제는 동쪽의 M개의 사이트 중 N개를 선택하여 서쪽의 사이트와 연결하는 경우의 수를 구하는 것과 같다. 즉 mCn를 구하면 된다.

조합 함수는 조합 성질과

재귀를 활용하여 만들었다.

	// 조합 함수
	static int combi(int n, int r) {
		if (dp[n][r] > 0) {
			return dp[n][r];
		}
		if (n == r || r == 0) {
			return dp[n][r] = 1;
		}
		return dp[n][r] = combi(n - 1, r - 1) + combi(n - 1, r);
	}

풀이 코드 2

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

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		int T = Integer.parseInt(bf.readLine());
		StringBuilder sb = new StringBuilder();
		for (int tc = 1; tc <= T; tc++) {
			String s = bf.readLine();
			StringTokenizer st = new StringTokenizer(s);
			int N = Integer.parseInt(st.nextToken());
			int M = Integer.parseInt(st.nextToken());

			int[][] dp = new int[30][30];

			for (int i = 0; i < 30; i++) {
				dp[i][0] = 1; // C(i,0) = 1
				dp[i][i] = 1; // C(i,i) = 1
			}

			for (int i = 2; i < 30; i++) {
				for (int j = 1; j < 30; j++) {
					dp[i][j] = dp[i-1][j-1]+dp[i-1][j];
				}
			}
			// M개 중에 N개 뽑는 경우의 수 = mCn
			sb.append(dp[M][N]).append('\n');
		}
		System.out.println(sb);
	}
}

DP로 푼 방법.