취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 32628번 문제인 두 스택이다.
문제는 아래 링크를 확인하자.

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

위에서부터 총 $K$개의 물건을 들어 없애는 방법의 수는 한 스택에서 0, 1, $\cdots$, $K$개의 물건을 들어내고 다른 스택에서 $K$, $K-1$, $\cdots$, 1, 0개의 물건을 들어내는 방법 중 실제로 실현이 가능한 것(각 스택에서 들어내는 물건의 개수가 $N$ 이하인 것)의 수와 같다. 이는 최대 $K+1$가지이다.

따라서 prefix sum으로 각 스택의 밑에서부터의 물건의 수를 미리 전처리해 두고 최대 $K+1$가지의 모든 경우에 대하여 원빈이가 들어야하는 물건의 값을 각각 조사해 그 중 최솟값을 구하는 것으로 문제를 선형시간에 해결할 수 있다.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;
typedef long long ll;

int N, K; ll ans = 2000000000000000007LL;
ll A[100001], B[100001];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> N >> K;
	for (int i = 1; i <= N; i++) {
		cin >> A[i];
		A[i] += A[i - 1];
	}
	for (int i = 1; i <= N; i++) {
		cin >> B[i];
		B[i] += B[i - 1];
	}

	for (int i = 0; i <= N; i++) {
		int j = (N * 2 - K) - i;
		if (j < 0 || j > N) continue;
		ans = min(ans, max(A[i], B[j]));
	}
	cout << ans;
}