可以选最大数为aj,
选次大数为ai,
选ai为c,
这样的话,
次大数的值就加到最大数里面去了,
次大数就变成0了。
那么基于这个思路,
我们选前k+1个数加起来就好了
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
const int N3 = 1117, N4 = 1e4 + 10, N5 = 1e5 + 10, N6 = 1e6 + 10;
#define mp make_pair
#define pb push_back
#define eb emplace_back
#define sum(a) ( accumulate ((a).begin(), (a).end(), 0ll))
#define mine(a) (*min_element((a).begin(), (a).end()))
#define maxe(a) (*max_element((a).begin(), (a).end()))
#define mini(a) ( min_element((a).begin(), (a).end()) - (a).begin())
#define maxi(a) ( max_element((a).begin(), (a).end()) - (a).begin())
#define lowb(a, x) ( lower_bound((a).begin(), (a).end(), (x)) - (a).begin())
#define uppb(a, x) ( upper_bound((a).begin(), (a).end(), (x)) - (a).begin())
#define all(a) (a).begin(), (a).end()
template<typename T> vector<T>& operator-- (vector<T> &v){for (auto& i : v) --i; return v;}
template<typename T> vector<T>& operator++ (vector<T> &v){for (auto& i : v) ++i; return v;}
template<typename T> istream& operator>>(istream& is, vector<T> &v){for (auto& i : v) is >> i; return is;}
template<typename T> ostream& operator<<(ostream& os, vector<T> v){for (auto& i : v) os << i << ' '; return os;}
template<typename T, typename U> pair<T,U>& operator-- (pair<T, U> &p){--p.first; --p.second; return p;}
template<typename T, typename U> pair<T,U>& operator++ (pair<T, U> &p){++p.first; ++p.second; return p;}
template<typename T, typename U> istream& operator>>(istream& is, pair<T, U> &p){is >> p.first >> p.second; return is;}
template<typename T, typename U> ostream& operator<<(ostream& os, pair<T, U> p){os << p.first << ' ' << p.second; return os;}
template<typename T, typename U> pair<T,U> operator-(pair<T,U> a, pair<T,U> b){return mp(a.first-b.first, a.second-b.second);}
template<typename T, typename U> pair<T,U> operator+(pair<T,U> a, pair<T,U> b){return mp(a.first+b.first, a.second+b.second);}
template<typename T, typename U> void umin(T& a, U b){if (a > b) a = b;}
template<typename T, typename U> void umax(T& a, U b){if (a < b) a = b;}
// bool prim[N6];
// void getpr()
// {
// memset(prim, true, sizeof prim);
// int u = 1e6;
// for(int i = 2; i <= u; i ++)
// if(prim[i])
// {
// int j = 2;
// while(i * j <= u) prim[i*j] = false, j ++;
// }
// }
int main()
{
ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
int T;
cin >> T;
ll t[N5];
t[0] = 1;
for(int i = 1; i <= 1e5; i ++) t[i] = 2 * t[i-1];
// getpr();
// vector<int> prims;
// for(int i = 2; i <= 1e6; i ++) if(prim[i]) prims.pb(i);
while(T --)
{
int n, k;
cin >> n >> k;
vector<ll> a(n);
cin >> a;
sort(all(a));
reverse(all(a));
ll res = 0;
for(int i = 0; i < min(n, k + 1); i ++) res += a[i];
cout << res << endl;
}
return 0;
}
这样把写会更有效率吗,写这一大堆define不会更耗时吗