結果
問題 | No.1060 素敵な宝箱 |
ユーザー |
![]() |
提出日時 | 2020-05-22 23:08:14 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 21 ms / 2,000 ms |
コード長 | 1,108 bytes |
コンパイル時間 | 2,576 ms |
コンパイル使用メモリ | 204,072 KB |
最終ジャッジ日時 | 2025-01-10 15:01:50 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include <bits/stdc++.h> using namespace std; #ifdef LOCAL #include "debug.h" #else #define DEBUG(...) #endif int main() { cin.tie(nullptr); ios::sync_with_stdio(false); int n, m; cin >> n >> m; vector a(n, vector(m, 0LL)); vector c(m, 0LL); for (int i = 0; i < n; ++i) { for (int j = 0; j < m; ++j) { cin >> a[i][j]; c[j] += a[i][j]; } } vector b(n, 0LL); for (int i = 0; i < n; ++i) { for (int j = 0; j < m; ++j) { b[i] += a[i][j] * c[j]; } } vector<int> ord(n); iota(begin(ord), end(ord), 0); sort(begin(ord), end(ord), [&](int i, int j) { return b[i] > b[j]; }); vector x(m, 0LL), y(m, 0LL); for (int t = 0; t < n; ++t) { if (t % 2 == 0) { for (int j = 0; j < m; ++j) { x[j] += a[ord[t]][j]; } } else { for (int j = 0; j < m; ++j) { y[j] += a[ord[t]][j]; } } } auto res = accumulate(begin(x), end(x), 0LL, [](auto s, auto e) { return s += e * e; }); res -= accumulate(begin(y), end(y), 0LL, [](auto s, auto e) { return s += e * e; }); cout << res << '\n'; }