K개의 동합 구간으로 분할
실버 III의견: 0
수열 \(A_1, A_2, \dots, A_N\)을 정확히 \(K\)개의 연속 부분 수열로 나누되, 각 부분 수열의 합이 모두 같도록 하고 싶다.
수열을 \(K\)개의 비어 있지 않은 연속 구간으로 나누는 방법이 존재하면, 각 구간의 오른쪽 끝 인덱스(1-기반)를 공백으로 구분해 출력하여라. 불가능하면 \(-1\)을 출력한다.
가능한 분할이 여러 가지이면 아무 것이나 출력한다.
- \(1 \le K \le N \le 100\,000\)
- \(-10^6 \le A_i \le 10^6\)
첫째 줄에 수열의 길이 \(N\)과 분할 개수 \(K\)가 공백으로 구분되어 주어진다.
둘째 줄에 수열 \(A_1, A_2, \dots, A_N\)이 공백으로 구분되어 주어진다.
가능하면 각 구간의 오른쪽 끝 인덱스를 오름차순으로 \(K\)개 공백으로 구분해 출력한다. 불가능하면 \(-1\)을 출력한다.
6 3
1 2 3 2 1 3
-1
합이 \(12\)이므로 목표 구간 합은 \(12 / 3 = 4\)이다. 누적합 수열은 \(1, 3, 6, 8, 9, 12\)로 \(4\)가 등장하지 않아 \(K = 3\)으로 분할할 수 없다. 출력: \(-1\).
4 2
3 3 3 3
2 4
합 \(12\), 목표 합 \(6\). 위치 \(1\)부터 누적합이 \(3, 6, 9, 12\)이므로 \(6\)에 처음 도달하는 곳은 위치 \(2\), \(12\)에 도달하는 곳은 위치 \(4\). 출력: \(2\ 4\). 각 구간 합: \([3,3]=6\), \([3,3]=6\).
4 2
1 -1 1 -1
2 4
합이 \(0\)이므로 목표 합은 \(0\). 누적합이 \(0\)이 되는 위치는 \(2\)와 \(4\). 첫 두 위치가 \(2, 4\)이므로 출력: \(2\ 4\). 구간 합: \([1,-1]=0\), \([1,-1]=0\).
막혔나요? 코인으로 단계별 힌트를 잠금 해제하세요 — 첫 힌트는 가벼운 방향 제시, 뒤로 갈수록 더 많이 알려 줍니다. 문제를 풀면 모든 힌트가 무료로 공개됩니다.
riseoj 작성
출처 Original
평가 및 의견
K개의 동합 구간으로 분할
Log in to rate problems.
아직 의견이 없습니다. 자격이 된다면 위 양식에서 가장 먼저 평가해 보세요.
풀이 제출
K개의 동합 구간으로 분할