結果
問題 | No.1823 Tricolor Dango |
ユーザー |
![]() |
提出日時 | 2022-01-28 21:25:21 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 43 ms / 2,000 ms |
コード長 | 1,422 bytes |
コンパイル時間 | 2,472 ms |
コンパイル使用メモリ | 220,128 KB |
最終ジャッジ日時 | 2025-01-27 16:04:45 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 25 |
ソースコード
#include<bits/stdc++.h> namespace { #pragma GCC diagnostic push #pragma GCC diagnostic ignored "-Wunused-function" #include<atcoder/all> #pragma GCC diagnostic pop using namespace std; using namespace atcoder; #define rep(i,n)for (int i = 0; i < int(n); ++i) #define rrep(i,n)for (int i = int(n)-1; i >= 0; --i) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() template<class T> void chmax(T& a, const T& b) { a = max(a, b); } template<class T> void chmin(T& a, const T& b) { a = min(a, b); } using ll = long long; using P = pair<int,int>; using VI = vector<int>; using VVI = vector<VI>; using VL = vector<ll>; using VVL = vector<VL>; template <class T> long long max_distinct_k(vector<T> a, int k) { const int n = a.size(); if (n < k) return 0; sort(a.begin(), a.end()); const int base = n - k; long long rest = accumulate(a.begin(), a.begin() + base, 0LL); for (int i = base + 1; i < n; i++) { long long mx = rest / (i - base); if (mx < a[i] - a[i - 1]) { return a[i - 1] + mx; } rest -= (a[i] - a[i - 1]) * (i - base); } return a[n - 1] + rest / k; } } int main() { ios::sync_with_stdio(false); cin.tie(0); int tt; cin >> tt; while(tt--) { int n; cin >> n; VL a(n); rep(i, n) cin >> a[i]; ll tot = accumulate(all(a), 0LL); ll cnt = max_distinct_k(move(a), 3); cout << (cnt * 3 == tot ? "Yes\n" : "No\n"); } }