https://www.acmicpc.net/problem/32069
태그 : 그리디, 브루트 포스
일단 L값이 눈에 띄었다. 좌표로 뭘 처리할 문제는 아니라고 생각했다.
출력할 K 수가 비교적 작고, 가로등의 거리가 O(1)에 구해지니 O(K)에 구할 수 있겠다고 생각했다.
그래서 우선 가로등 갯수만큼 0을 정답 배열에 넣어주고,
1부터 k이하까지 (가로등이 한개라면 가장 큰 정답은 최대 K이므로) 밝기가
각 가로등 사이에 설치 가능한지 판별 후 정답 배열에 넣었다.
시작과 끝 가로등 사이의 경계 처리만 잘 해주면 된다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.StringTokenizer;
public class Main {
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static StringBuilder sb = new StringBuilder();
static StringTokenizer st;
static long [] arr = new long [300000];
static long L;
static int N, K;
static void solve() {
int size = 0;
for(int i = 0; i < Math.min(N, K); ++i) {
sb.append(0).append('\n');
++size;
}
for(int i = 1 ; i <= K && size < K; ++i) {
for (int j = 0; j < N; ++j) {
if (j == 0 || j == N - 1) {
if (j == 0) {
if (arr[j] - i >= 0) {
sb.append(i).append('\n');
++size;
}
}
if (j == N - 1 && size < K) {
if (arr[j] + i <= L) {
sb.append(i).append('\n');
++size;
}
}
}
if (j != N - 1 && arr[j] + i <= arr[j + 1] - i && size < K) {
if (arr[j] + i == arr[j + 1] - i) {
sb.append(i).append('\n');
++size;
}
else {
if (size < K) {
sb.append(i).append('\n');
++size;
}
if (size < K) {
sb.append(i).append('\n');
++size;
}
}
}
}
}
}
public static void main(String[] args) throws NumberFormatException, IOException {
st = new StringTokenizer(br.readLine().trim());
L = Long.parseLong(st.nextToken());
N = Integer.parseInt(st.nextToken());
K = Integer.parseInt(st.nextToken());
st = new StringTokenizer(br.readLine().trim());
for(int i = 0; i < N; ++i) {
arr[i] = Long.parseLong(st.nextToken());
}
solve();
System.out.println(sb);
}
}
정올 대비하는 학생들 정말 대단한 것 같다
초등부 2번 중고등부 1번..