ABC100 D - Patisserie ABC
問題ページ
D: Patisserie ABC - AtCoder Beginner Contest 100 | AtCoder
考えたこと
- 与えられる値が全て正だったらx[i]+y[i]+z[i]が大きい方から取ればいいだけ
- 最大化するのは絶対値で負がある
- それぞれ正負を試せばいいのでは?
- 正負を固定すればソートして大きい方から貪欲に取っていけばよさそう
- O(nlogn) で通りそうなので投げると通る
未証明貪欲でWJで10分待つの心臓に悪い
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; #define int ll using VI = vector<int>; using VVI = vector<VI>; using PII = pair<int, int>; #define FOR(i, a, n) for (ll i = (ll)a; i < (ll)n; ++i) #define REP(i, n) FOR(i, 0, n) #define ALL(x) x.begin(), x.end() #define PB push_back const ll LLINF = (1LL<<60); const int INF = (1LL<<30); const int MOD = 1000000007; template <typename T> T &chmin(T &a, const T &b) { return a = min(a, b); } template <typename T> T &chmax(T &a, const T &b) { return a = max(a, b); } template <typename T> bool IN(T a, T b, T x) { return a<=x&&x<b; } template<typename T> T ceil(T a, T b) { return a/b + !!(a%b); } template<class S,class T> ostream &operator <<(ostream& out,const pair<S,T>& a){ out<<'('<<a.first<<','<<a.second<<')'; return out; } template<class T> ostream &operator <<(ostream& out,const vector<T>& a){ out<<'['; REP(i, a.size()) {out<<a[i];if(i!=a.size()-1)out<<',';} out<<']'; return out; } int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0}; signed main(void) { cin.tie(0); ios::sync_with_stdio(false); int n, m; cin >> n >> m; VI x(n), y(n), z(n); REP(i, n) cin >> x[i] >> y[i] >> z[i]; int ans = 0; REP(i, 1<<3) { VI v; REP(j, n) { int tmp = 0; if(i&1) tmp += x[j]; else tmp -= x[j]; if(i&2) tmp += y[j]; else tmp -= y[j]; if(i&4) tmp += z[j]; else tmp -= z[j]; v.PB(tmp); } sort(ALL(v)); reverse(ALL(v)); int ret = 0; REP(j, m) ret += v[j]; chmax(ans, ret); } cout << ans << endl; return 0; }