제출 #1241896

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1241896		khanhtt	Stove (JOI18_stove)	C++20	100 / 100	11 ms	1480 KiB

stove

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int n,k,A[100005];
vector<int> gap;
main(){
    ios::sync_with_stdio(0);
    cin.tie(nullptr);
    cin >> n >> k;
    for (int i=1; i<=n; i++) cin >> A[i];
    if (k>=n){
        cout << n;
    }
    else{
        for (int i=2; i<=n; i++){
            gap.push_back(A[i]-A[i-1]);
        }
        sort(gap.begin(),gap.end(),greater<>());
        int cost=A[n]-A[1]+1;
        for (int i=0; i<k-1; i++){
            cost-=(gap[i]-1);
        }
        cout << cost;
    }
}

컴파일 시 표준 에러 (stderr) 메시지

stove.cpp:7:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    7 | main(){
      | ^~~~

• Subtask 1: 20 / 20
• Subtask 2: 30 / 30
• Subtask 3: 50 / 50