結果

問題 No.5020 Averaging
ユーザー tabae326
提出日時 2024-02-25 14:07:02
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 604 ms / 1,000 ms
コード長 6,612 bytes
コンパイル時間 4,466 ms
コンパイル使用メモリ 275,484 KB
実行使用メモリ 6,548 KB
スコア 15,490,811
最終ジャッジ日時 2024-02-25 14:07:44
合計ジャッジ時間 37,261 ms
ジャッジサーバーID
(参考情報)
judge13 / judge12
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll = long long;
using pl = std::pair<long long, long long>;
//using mint = atcoder::modint1000000007;
using mint = atcoder::modint998244353;
#define rep(i, srt, end) for (long long i = (srt); i < (long long)(end); i++)
#define all(v) (v).begin(), (v).end()
#define dump(var) do{ if(debug::enable) { std::cerr << #var << " : "; debug::print(var); } } while(0);
constexpr int di[4] = {1, -1, 0, 0};
constexpr int dj[4] = {0, 0, 1, -1};
constexpr long long inf = 1LL<<60;
template<typename T> inline void chmax(T &a, T b) { a = std::max(a, b); };
template<typename T> inline void chmin(T &a, T b) { a = std::min(a, b); };
template<typename T> void vprint(const std::vector<T>& v) { for(int i = 0; i < v.size(); i++) { std::cout << v[i] << (i == v.size()-1 ? "\n" : " ");
    } }
template<typename T> void vprintln(const std::vector<T>& v) { for(int i = 0; i < v.size(); i++) { std::cout << v[i] << "\n"; } }
template<int M> void vprint(const std::vector<atcoder::static_modint<M>>& v) { for(int i = 0; i < v.size(); i++) { std::cout << v[i].val() << (i == v
    .size()-1 ? "\n" : " "); } }
template<int M> void vprintln(const std::vector<atcoder::static_modint<M>>& v) { for(int i = 0; i < v.size(); i++) { std::cout << v[i].val() << "\n";
    } }
namespace debug {
#if defined(ONLINE_JUDGE)
const bool enable = false;
#else
const bool enable = true;
#endif
template<typename T> void push(const T& e) { std::cerr << e; }
template<int M> void push(const atcoder::static_modint<M>& e) { std::cerr << e.val(); }
template<typename T1, typename T2> void push(const std::pair<T1, T2>& e) { std::cerr << "("; push(e.first); std::cerr << ","; push(e.second); std
        ::cerr << ")"; }
template<typename T1, typename T2, typename T3> void push(const std::tuple<T1, T2, T3>& e) { std::cerr << "("; push(get<0>(e)); std::cerr << ",";
        push(get<1>(e)); std::cerr << ","; push(get<2>(e)); std::cerr << ")"; }
template<typename T> void print(const T& e) { push(e); std::cerr << "\n"; }
template<typename T> void print(const std::vector<T>& v) { for(int i = 0; i < v.size(); i++) { push(v[i]); std::cerr << " "; } std::cerr << "\n";
        }
template<typename T> void print(const std::vector<std::vector<T>>& v) { std::cerr << "\n"; for(int i = 0; i < v.size(); i++) { std::cerr << i <<
        ": "; print(v[i]); } }
};
struct rand_generator {
random_device seed_gen;
mt19937 engine;
rand_generator() : engine(seed_gen()) {}
int rand(int mod) {
return engine() % mod;
}
} ryuka;
#include <sys/time.h>
struct timer {
double global_start;
double gettime() {
struct timeval tv;
gettimeofday(&tv, NULL);
return tv.tv_sec + tv.tv_usec * 1e-6;
}
void init() {
global_start = gettime();
}
double elapsed() {
return gettime() - global_start;
}
} toki;
constexpr ll TARGET = 500000000000000000;
ll N;
vector<pl> V;
// 2
bool is_2beki(ll val) {
ll cnt = 0;
while(val) {
cnt += val % 2;
val /= 2;
}
return cnt == 1;
}
//
// TARGET
pair<ll, vector<ll>> search_lowest_diff_pattern(const vector<ll>& idx) {
const ll size = idx.size();
ll best_diff = inf;
ll best_mask = -1;
rep(mask, 0, 1<<size) {
unsigned long long sum_front = 0;
unsigned long long sum_back = 0;
ll pp = __builtin_popcountll(mask);
if(!is_2beki(pp)) continue;
rep(i, 0, size) {
if(mask & (1<<i)) {
sum_front += V[idx[i]].first;
sum_back += V[idx[i]].second;
}
}
ll max_ave = max(sum_front / size, sum_back / size);
if(abs(TARGET - max_ave) < best_diff) {
best_diff = abs(TARGET - max_ave);
best_mask = mask;
}
}
assert(best_mask != -1);
vector<ll> res;
rep(i, 0, size) if(best_mask & (1<<i)) res.push_back(i);
return {best_diff, res};
}
// N=10
// 5*10^17
// N=10unsiged long long
vector<ll> decide_indicies_to_use(const double time_limit = 0.6, const ll size = 16) {
const double start_time = toki.gettime();
ll best_diff = inf;
vector<ll> best_idx_to_use;
vector<ll> idx(N);
iota(all(idx), 0);
ll counter = 0;
while(toki.gettime() - start_time < time_limit) {
counter++;
shuffle(all(idx), ryuka.engine);
vector<ll> tmp_idx;
bool zero = false;
for(ll i = 0; i < size; i++) {
tmp_idx.push_back(idx[i]);
if(idx[i] == 0) zero = true;
}
if(!zero) tmp_idx[0] = 0;
sort(all(tmp_idx));
auto [diff, idx_to_use] = search_lowest_diff_pattern(tmp_idx);
if(diff < best_diff) {
best_diff = diff;
swap(best_idx_to_use, idx_to_use);
}
}
dump(counter);
dump(toki.gettime() - start_time);
assert(!best_idx_to_use.empty());
return best_idx_to_use;
}
vector<pl> add_uniformly(const vector<ll>& idx) {
assert(idx[0] == 0);
assert(is_2beki(idx.size()));
vector<pl> debug_v = V;
vector<pl> ans;
for(ll b = 0; (1<<b) < idx.size(); b++) {
for(ll i = 0; i < idx.size(); i += (1<<b) * 2) {
ll u = idx[i];
ll v = idx[i + (1<<b)];
ans.push_back({u, v});
ll ave_front = (debug_v[u].first + debug_v[v].first) / 2;
ll ave_back = (debug_v[u].second + debug_v[v].second) / 2;
debug_v[u] = debug_v[v] = {ave_front, ave_back};
}
}
dump(debug_v);
return ans;
}
void print_ans(const vector<pl>& ans) {
cout << ans.size() << endl;
for(auto [u, v]: ans) {
cout << u+1 << " " << v+1 << endl;
}
}
void solve() {
//
cin >> N;
V.resize(N);
for(auto& [a, b]: V) cin >> a >> b;
// 使
vector<ll> idx_to_use = decide_indicies_to_use();
dump(idx_to_use);
//
vector<pl> ans = add_uniformly(idx_to_use);
//
print_ans(ans);
}
int main() {
toki.init();
ios::sync_with_stdio(false);
cin.tie(nullptr);
solve();
return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0