[백준 / BOJ] C++ 29767 점수를 최대로

29767번: 점수를 최대로

 

문제

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

29767번: 점수를 최대로

 

 

풀이

목적지 i번 교실에 도착하면 1~i번 방의 점수를 얻습니다. 즉, 누적합 배열로 i번 교실을 목표로 했을 시 얻는 점수를 저장합니다. K명의 학생들은 서로 다른 목적지를 가지므로 누적합 배열을 내림차순 정렬하고 앞에서부터 K개 원소의 합을 출력하면 됩니다.

 

코드

#include <bits/stdc++.h>
using namespace std;
#define ll long long

int main() {
  ios::sync_with_stdio(0);
  cin.tie(0);
  ll n, k, ans = 0;
  cin >> n >> k;
  vector<ll> a(n + 1, 0);
  for (int i = 1; i <= n; i++) {
    cin >> a[i];
    a[i] += a[i - 1];
  }
  sort(a.begin(), a.end(), greater<>());
  for (int i = 0; i < k; i++)
    ans += a[i];
  cout << ans;
  return 0;
}