Submission #307177

#	Time	Username	Problem	Language	Result	Execution time	Memory
307177		radaiosm7	Stove (JOI18_stove)	C++98	100 / 100	53 ms	7152 KiB

stove

#include <bits/stdc++.h>
using namespace std;
int n, k, i, a, b, ans;
vector<pair<int,int>> ranges;
unordered_set<int> check;
vector<int> diff;

int main() {
  //freopen("test.in", "r", stdin);

  scanf("%d %d", &n, &k);
  check.rehash(n);
  ans = n;
  k--;

  for (i=0; i < n; i++) {
    scanf("%d", &a);
    check.insert(a);

    if (check.find(a-1) != check.end()) {
      ranges[ranges.size()-1].second = a+1;
    }

    else {
      ranges.push_back(make_pair(a, a+1));
    }
  }

  for (i=1; i < ranges.size(); i++) {
    diff.push_back(ranges[i].first - ranges[i-1].second);
  }

  k = diff.size() - k;
  sort(diff.begin(), diff.end());

  for (i=0; i < k; i++) {
    ans += diff[i];

  }

  printf("%d\n", ans);
  return 0;
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:29:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   29 |   for (i=1; i < ranges.size(); i++) {
      |             ~~^~~~~~~~~~~~~~~
stove.cpp:11:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   11 |   scanf("%d %d", &n, &k);
      |   ~~~~~^~~~~~~~~~~~~~~~~
stove.cpp:17:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   17 |     scanf("%d", &a);
      |     ~~~~~^~~~~~~~~~

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