취미로 코딩하기

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

이번에 볼 문제는 백준 31589번 문제인 포도주 시음이다.
문제는 아래 링크를 확인하자.

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

편의상 모든 포도주의 맛이 다르다고 가정하자.

$K$개의 포도주를 순서대로 마실 때, 어떤 연속한 세 포도주의 맛이 내림차순이거나 오름차순이라면 중간 차례로 마신 포도주를 마시지 않더라도 느끼는 총 맛의 차이가 변하지 않음을 관찰하자. 따라서 구하는 값은 처음을 제외하고 이전에 마신 포도주보다 맛없는 포도주를 마시는 것과 맛있는 포도주를 마시는 것을 반복하는 경우에서 찾아볼 수 있다. 0번째로 맛이 0인 포도주를 마셨다고 생각하면 이를 모든 포도주에 적용해 생각할 수 있음을 관찰하자.

이 때, 모든 $K$개의 포도주는 이전보다 맛있게 소비한 포도주거나 맛없게 소비한 포도주 둘로 나눌 수 있는데, (마지막으로 소비한 포도주가 이전보다 맛없는 포도주인 경우 그 포도주를 제외하면) $K$개의 포도주를 마시며 느낀 맛의 합은 (맛있게 소비한 포도주 맛 총합) - (맛없게 소비한 포도주 맛 총합)으로 계산할 수 있다. 이 값의 최댓값은 가장 맛있는 포도주들을 최대한 (맛있게 소비한 포도주 맛 총합)에 포함되게끔, 그리고 가장 맛없는 포도주들을 최대한 (맛없게 소비한 포도주 맛 총합)에 포함되게끔 포도주를 분류할 때 찾을 수 있음을 관찰할 수 있다.

위 관찰은 맛있는 정도가 같은 포도주가 있는 경우까지 어렵지 않게 확장할 수 있다. 위 내용을 구현해 문제를 해결하자.

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

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

int N, K;
int A[300001];
int L, R;
ll ans;

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

	cin >> N >> K; K++;
	for (int i = 0; i < N; i++) cin >> A[i];
	sort(A, A + N + 1);

	L = 0, R = N;
	while (K > 1) {
		ans += A[R] - A[L];
		L++, R--, K -= 2;
	}

	cout << ans;
}