문제(출처: https://www.acmicpc.net/problem/16938)
< 캠프 준비 >
문제 풀이
문제를 골랐을 때와 고르지 않았을 때 모든 경우를 구한다.
my solution (Java)
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;
public class _16938_ { // 캠프 준비
static int arr[], n, l, r, x, result;
public static void main(String[] args) throws IOException {
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(bf.readLine());
n = Integer.parseInt(st.nextToken());
l = Integer.parseInt(st.nextToken());
r = Integer.parseInt(st.nextToken());
x = Integer.parseInt(st.nextToken());
arr = new int[n];
result = 0;
st = new StringTokenizer(bf.readLine());
for (int i = 0; i < n; i++) {
arr[i] = Integer.parseInt(st.nextToken());
}
Arrays.sort(arr);
search(0, 0, arr[n-1], -1);
System.out.println(result);
}
private static void search(int sum, int idx, int min, int max) {
if (idx == n || sum > r) {
if (sum >= l && sum <= r && max - min >= x) {
result += 1;
}
return;
}
search(sum + arr[idx], idx + 1, Math.min(min, arr[idx]), Math.max(max, arr[idx]));
search(sum, idx + 1, min, max);
}
}
변수)
n, l, r, x : 문제 수, 난이도의 합 조건, 난이도 차 조건
arr : 문제 난이도
result : 문제를 고르는 방법의 수
문제 수와 문제 난이도의 합 조건, 난이도 차 조건을 입력받는다. 문제 수만큼 문제 난이도를 입력받은 후 오름차순으로 정렬한다. search 함수를 통해 캠프에 사용할 문제를 고르는 방법의 수를 구해 출력한다.
search ( 문제 난이도 합, 인덱스, 가장 쉬운 문제의 난이도, 가장 어려운 문제의 난이도)
문제를 다 골랐고 문제 난이도의 합이 l이상 r이하이며 난이도의 차가 x이상이라면 result +1 후 return, 문제 난이도 합이 r보다 크다면 다음 계산을 할 필요 없으므로 return 한다.
문제를 포함시켰을 경우와 문제를 포함시키지 않았을 경우 각각 계산을 위해 search 함수를 호출한다.
'🌞Algorithm > 🔥Baekjoon' 카테고리의 다른 글
[Baekjoon] 1083_소트 (0) | 2024.03.19 |
---|---|
[Baekjoon] 12931_두 배 더하기 (0) | 2024.03.18 |
[Baekjoon] 5549_행성 탐사 (0) | 2024.03.14 |
[Baekjoon] 1990_소수인팰린드롬 (0) | 2024.03.13 |
[Baekjoon] 1456_거의 소수 (0) | 2024.03.12 |