Problem p02785 (C++)

Buggy submission (status: Time Limit Exceeded):
#include <algorithm>
#include <climits>
#include <cmath>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <vector>
using namespace std;

#define ll long long
#define ld long double
#define pi 3.14159265359;
// %llu %lf

// v:vector  type:型  order:greater(大きい順) or less(小さい順)
#define _sort(v, type, order)                                                  \
  do {                                                                         \
    sort(v.begin(), v.end(), order<type>());                                   \
  } while (0)
// vector 要素の総和算出
// v:vector  default_value:初期値
#define _sum(v, default_value) accumulate(v.begin(), v.end(), default_value)
// vector 最大値( return ite )
#define _max_element(v) max_element(v.begin(), v.end())
// vector 最小値( return ite )
#define _min_element(v) min_element(v.begin(), v.end())
// vector 最大値が格納されている要素値
#define _max_element_number(v)                                                 \
  distance(v.begin(), max_element(v.begin(), v.end()))
// 特定コンテナの中から特定の値をカウントする
#define _count(v, value) count(v.begin(), v.end(), value)

// set<uint32_t> member;	// 重複するデータを保持する事はできない
// member.insert(2) member.insert(2) ⇒ member.count(2)は1
//                                                               member.emplace(2)とかも同じ
// multiset<uint32_t> v;	// 重複するデータも保持する事はできる
// member.insert(2) member.insert(2) ⇒ member.count(2)は2

// 丸め
#define _round(v) round(v)
// 2乗 / 3乗
#define _square(v) pow(v, 2)
#define _cube(v) pow(v, 3)
// 大小判定
#define _max(x, y) max(x, y)
#define _min(x, y) min(x, y)
// string ⇒ int
#define _stringtoi(s) stoi(s)
// double 平方根
#define _sqrt(x) sqrt(x)
// double 引数 x 以上で最小の整数値 ex) 3.30303 ⇒ 4
#define _ceil(x) ceil(x)
// 指定された要素以上の値が現れる最初の位置のイテレータを取得する
#define _lower_bound(v, min) lower_bound(v.begin(), v.end(), min)
// 指定された要素より大きい値が現れる最初の位置のイテレータを取得する
#define _upper_bound(v, min) upper_bound(v.begin(), v.end(), min)

// 順列 n個の数が与えられる
// 0,1,2,...,n-1
// 全ての並べ方を1行ごとに出力する
// ex 0 1 2 / 0 2 1 / 1 0 2 / 1 2 0 / 2 0 1 / 2 1 0
#define _next_permutation(v)                                                   \
  do {                                                                         \
    for (auto num : v) {                                                       \
      printf("%d ", num);                                                      \
    }                                                                          \
    printf("\n");                                                              \
  } while (next_permutation(v.begin(), v.end()))

// 絶対値
template <typename T> static T _abs(const T x) { return (x > 0 ? x : -x); }

// 最大公約数
int64_t gcd(int64_t a, int64_t b) {
  while (b) {
    int64_t c = b;
    b = a % b;
    a = c;
  }
  return a;
}
// 最小公倍数
int64_t lcm(int64_t a, int64_t b) {
  if (!a || !b)
    return 0;
  return a * b / gcd(a, b);
}

// 多次元 std::vector 生成
template <class T> vector<T> make_vec(size_t a) { return vector<T>(a); }
template <class T, class... Ts> auto make_vec(size_t a, Ts... ts) {
  return vector<decltype(make_vec<T>(ts...))>(a, make_vec<T>(ts...));
}
// ex) auto dp = make_vec<uint64_t>(SIZE + 1, 2, 2);

// ■
// UINT32_MAX
// 4294967295 ≒ 4 * 1e9

// ■
// 割り算した結果との比較での丸め対処時
// ABC 161 B
// https://atcoder.jp/contests/abc161/tasks/abc161_b
// double   border = _sum(v, 0 / (double)(4 * m);
//            ↓
// uint32_t border = (_sum(v, 0) + (4 * m) - 1) / (4 * m);

// ■
// N の数が与えられた場合の組合せ N * (N - 1) / 2

/*
 */
// ABC 153 C
// https://atcoder.jp/contests/abc153/tasks/abc153_c

int main() {
  uint32_t n, k;
  cin >> n >> k;

  vector<uint64_t> v(n);

  uint64_t sum = 0;
  for (uint32_t i = 0; i < n; i++) {
    cin >> v[i];
    sum += v[i];
  }

  if (n <= k) {
    printf("0\n");
    return 0;
  }
  _sort(v, uint64_t, greater);

  for (uint32_t i = 0; i < k; i++) {
    sum -= v[0];
    v.erase(v.begin());
  }

  printf("%llu\n", sum);
  return 0;
}


Fixed submission:
#include <algorithm>
#include <climits>
#include <cmath>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <vector>
using namespace std;

#define ll long long
#define ld long double
#define pi 3.14159265359;
// %llu %lf

// v:vector  type:型  order:greater(大きい順) or less(小さい順)
#define _sort(v, type, order)                                                  \
  do {                                                                         \
    sort(v.begin(), v.end(), order<type>());                                   \
  } while (0)
// vector 要素の総和算出
// v:vector  default_value:初期値
#define _sum(v, default_value) accumulate(v.begin(), v.end(), default_value)
// vector 最大値( return ite )
#define _max_element(v) max_element(v.begin(), v.end())
// vector 最小値( return ite )
#define _min_element(v) min_element(v.begin(), v.end())
// vector 最大値が格納されている要素値
#define _max_element_number(v)                                                 \
  distance(v.begin(), max_element(v.begin(), v.end()))
// 特定コンテナの中から特定の値をカウントする
#define _count(v, value) count(v.begin(), v.end(), value)

// set<uint32_t> member;	// 重複するデータを保持する事はできない
// member.insert(2) member.insert(2) ⇒ member.count(2)は1
//                                                               member.emplace(2)とかも同じ
// multiset<uint32_t> v;	// 重複するデータも保持する事はできる
// member.insert(2) member.insert(2) ⇒ member.count(2)は2

// 丸め
#define _round(v) round(v)
// 2乗 / 3乗
#define _square(v) pow(v, 2)
#define _cube(v) pow(v, 3)
// 大小判定
#define _max(x, y) max(x, y)
#define _min(x, y) min(x, y)
// string ⇒ int
#define _stringtoi(s) stoi(s)
// double 平方根
#define _sqrt(x) sqrt(x)
// double 引数 x 以上で最小の整数値 ex) 3.30303 ⇒ 4
#define _ceil(x) ceil(x)
// 指定された要素以上の値が現れる最初の位置のイテレータを取得する
#define _lower_bound(v, min) lower_bound(v.begin(), v.end(), min)
// 指定された要素より大きい値が現れる最初の位置のイテレータを取得する
#define _upper_bound(v, min) upper_bound(v.begin(), v.end(), min)

// 順列 n個の数が与えられる
// 0,1,2,...,n-1
// 全ての並べ方を1行ごとに出力する
// ex 0 1 2 / 0 2 1 / 1 0 2 / 1 2 0 / 2 0 1 / 2 1 0
#define _next_permutation(v)                                                   \
  do {                                                                         \
    for (auto num : v) {                                                       \
      printf("%d ", num);                                                      \
    }                                                                          \
    printf("\n");                                                              \
  } while (next_permutation(v.begin(), v.end()))

// 絶対値
template <typename T> static T _abs(const T x) { return (x > 0 ? x : -x); }

// 最大公約数
int64_t gcd(int64_t a, int64_t b) {
  while (b) {
    int64_t c = b;
    b = a % b;
    a = c;
  }
  return a;
}
// 最小公倍数
int64_t lcm(int64_t a, int64_t b) {
  if (!a || !b)
    return 0;
  return a * b / gcd(a, b);
}

// 多次元 std::vector 生成
template <class T> vector<T> make_vec(size_t a) { return vector<T>(a); }
template <class T, class... Ts> auto make_vec(size_t a, Ts... ts) {
  return vector<decltype(make_vec<T>(ts...))>(a, make_vec<T>(ts...));
}
// ex) auto dp = make_vec<uint64_t>(SIZE + 1, 2, 2);

// ■
// UINT32_MAX
// 4294967295 ≒ 4 * 1e9

// ■
// 割り算した結果との比較での丸め対処時
// ABC 161 B
// https://atcoder.jp/contests/abc161/tasks/abc161_b
// double   border = _sum(v, 0 / (double)(4 * m);
//            ↓
// uint32_t border = (_sum(v, 0) + (4 * m) - 1) / (4 * m);

// ■
// N の数が与えられた場合の組合せ N * (N - 1) / 2

/*
 */
// ABC 153 C
// https://atcoder.jp/contests/abc153/tasks/abc153_c

int main() {
  uint32_t n, k;
  cin >> n >> k;

  vector<uint64_t> v(n);

  uint64_t sum = 0;
  for (uint32_t i = 0; i < n; i++) {
    cin >> v[i];
    sum += v[i];
  }

  if (n <= k) {
    printf("0\n");
    return 0;
  }
  _sort(v, uint64_t, greater);

  for (uint32_t i = 0; i < k; i++) {
    sum -= v[i];
    // v.erase(v.begin());
  }

  printf("%llu\n", sum);
  return 0;
}