#955634 (C++17) No.5020 Averaging

提出ソース

結果

問題	No.5020 Averaging
ユーザー	arimatti
提出日時	2024-02-25 16:27:02
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	TLE
実行時間	-
コード長	6,137 bytes
コンパイル時間	2,662 ms
コンパイル使用メモリ	223,668 KB
実行使用メモリ	6,548 KB
スコア	0
最終ジャッジ日時	2024-02-25 16:29:04
合計ジャッジ時間	67,850 ms
ジャッジサーバーID （参考情報）	judge10 / judge11

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	TLE	-
testcase_01	TLE	-
testcase_02	TLE	-
testcase_03	TLE	-
testcase_04	TLE	-
testcase_05	TLE	-
testcase_06	TLE	-
testcase_07	TLE	-
testcase_08	TLE	-
testcase_09	TLE	-
testcase_10	TLE	-
testcase_11	TLE	-
testcase_12	TLE	-
testcase_13	TLE	-
testcase_14	TLE	-
testcase_15	TLE	-
testcase_16	TLE	-
testcase_17	TLE	-
testcase_18	TLE	-
testcase_19	TLE	-
testcase_20	TLE	-
testcase_21	TLE	-
testcase_22	TLE	-
testcase_23	TLE	-
testcase_24	TLE	-
testcase_25	TLE	-
testcase_26	TLE	-
testcase_27	TLE	-
testcase_28	TLE	-
testcase_29	TLE	-
testcase_30	TLE	-
testcase_31	TLE	-
testcase_32	TLE	-
testcase_33	TLE	-
testcase_34	TLE	-
testcase_35	TLE	-
testcase_36	TLE	-
testcase_37	TLE	-
testcase_38	TLE	-
testcase_39	TLE	-
testcase_40	TLE	-
testcase_41	TLE	-
testcase_42	TLE	-
testcase_43	TLE	-
testcase_44	TLE	-
testcase_45	TLE	-
testcase_46	TLE	-
testcase_47	TLE	-
testcase_48	TLE	-
testcase_49	TLE	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;
// #include <atcoder/all>
// using namespace atcoder;

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

#define rep(i,n) for(int i = 0; i < (n); ++i)
#define drep(i,n) for(int i = (n)-1; i >= 0; --i)
#define rng(a) a.begin(),a.end()
#define rrng(a) a.rbegin(),a.rend()
template<typename T> using vc = vector<T>;
template<typename T> using vv = vc<vc<T>>;
template<typename T> using PQ = priority_queue<T,vc<T>,greater<T> >;
template<typename T>void uni(T& a){sort(rng(a));a.erase(unique(rng(a)),a.end());}
using ll = long long;
using P = pair<int,int>;
int dx[4] = {-1,0,1,0};
int dy[4] = {0,-1,0,1};
constexpr int INF = 1001001001;
constexpr ll INFL = 1001002003004005006ll;

/*
mst で各放送局を連結にする
差分更新しない焼きなまし
近傍：放送局をひとつ選び出力を変更する
*/

inline unsigned long long xor128() {
	static unsigned long long rx = 123456789, ry = 986960440, rz = 738905609, rw = 23140692;
	unsigned long long rt = (rx ^ (rx << 11));
	rx = ry;ry = rz;rz = rw;
	return (rw = (rw ^ (rw >> 19)) ^ (rt ^ (rt >> 8)));
}
inline float rand128(float min = 0, float max = 1){
	return (float)(xor128() % 0xFFFF) / 0xFFFF * (max - min) + min;
}
inline unsigned long long randint(int min, int max){
	return xor128()%(max - min + 1) + min;
}

struct Timer {
  double t = 0.0;
  double lastStop = 0.0;
  bool stopped = false;
  Timer() {
    restart();
  }
  inline void restart() {
    t = now();
    stopped = false;
  }
  inline double time() {
    if (stopped) return lastStop - t;
    else return now() - t;
  }
  inline double now() {
    unsigned long long l, h;
    __asm__ ("rdtsc" : "=a"(l), "=d"(h));
    return (double)(l | h << 32) * 3.333333333333333e-10;
  }
};

// グローバル変数
Timer timer;
int n; // 頂点数
vector<ll> ga;
vector<ll> gb;
void read_input() {
  cin >> n;
  ga.resize(n);
  gb.resize(n);
  rep(i,n) cin >> ga[i] >> gb[i];
}

struct Output {
  int u,v;
  Output (int u=0, int v=0) : u(u), v(v) {}
};

struct State {
  vector<ll> a;
  vector<ll> b;  
  vector<Output> ans;

  State() {
    a.resize(n);
    b.resize(n);
    a = ga;
    b = gb;
  }
  
  void init() {
  }

  void make_init_ans() {
    rep(i,50) {
      int idx1 = randint(0,n-2);
      int idx2 = randint(0,n-1);
      if (idx1 == idx2) idx1++;
      ans.push_back(Output(idx1,idx2));
      ll va = (a[idx1]+a[idx2])/2;
      ll vb = (b[idx1]+b[idx2])/2;
      a[idx1] = a[idx2] = va;
      b[idx1] = b[idx2] = vb;
    }
  }

  void print_ans() {
    cout << ans.size() << endl;
    for (Output o : ans) {
      cout << o.u << " " << o.v << endl;
    }
  }

  void gready() {
    set<P> st;
    rep(i,10) {
      int idx1 = randint(0,n-2);
      int idx2 = randint(0,n-1);
      if (idx1 == idx2) idx1++;
      ans.push_back(Output(idx1+1,idx2+1));
      ll va = (a[idx1]+a[idx2])/2;
      ll vb = (b[idx1]+b[idx2])/2;
      a[idx1] = a[idx2] = va;
      b[idx1] = b[idx2] = vb;
    }
    rep(i,30) {
      ll v1 = abs(5e17-a[0]);
      ll v2 = abs(5e17-b[0]);
      ll best_score = INFL;
      int best_j = -1;
      int best_k = -1;
      for (int j=0; j<n; j++) {
        for (int k=j+1; k<n; k++) {
          ll va = (a[k]+a[j])/2;
          ll vb = (b[k]+b[j])/2;
          ll score = abs(va-vb);
          if (st.count(P(j,k))) continue;
          if (best_score > score) {
            best_score = score;
            best_j = j;
            best_k = k;
          }
        }
      }
      if (best_j == -1) continue;
      ans.push_back(Output(best_j+1,best_k+1));
      ll va = (a[best_k]+a[best_j])/2;
      ll vb = (b[best_k]+b[best_j])/2;
      a[best_k] = a[best_j] = va;
      b[best_k] = b[best_j] = vb;
      st.insert(P(best_j,best_k));
    }
    rep(i,10) {
      ll v1 = abs(5e17-a[0]);
      ll v2 = abs(5e17-b[0]);
      ll best_score = INFL;
      int best_j = -1;
      for (int j=1; j<n; j++) {
        ll va = (a[0]+a[j])/2;
        ll vb = (b[0]+b[j])/2;
        ll vva = abs(5e17-va);
        ll vvb = abs(5e17-vb);
        ll score = max(vva,vvb);
        if (best_score > score) {
          best_score = score;
          best_j = j;
        }
      }
      if (best_j == -1) continue;
      ans.push_back(Output(1,best_j+1));
      ll va = (a[0]+a[best_j])/2;
      ll vb = (b[0]+b[best_j])/2;
      a[0] = a[best_j] = va;
      b[0] = b[best_j] = vb;
      // cerr << "--------------- " << endl;
      // rep(j,n) cerr << j << " " << a[j] << " " << b[j] << endl;
      // cerr << "--------------- " << endl;
    }    
  }

  void show_ans() {
  }

  void transition() {
    int idx = randint(0,49);

  }

  void rollback() {

  }

  ll calc_score() {
    ll V1 = abs(a[0]-5e17);
    ll V2 = abs(b[0]-5e17);
    return 2000000 - 100000*log10(max(V1,V2)+1);
  }

};

void simulatedAnnealing() {
  State state;
  state.gready();
  state.calc_score();
  ll score = state.calc_score();
  ll best_score = score;
  cerr << "best_score:" << "\t" << best_score << endl;
  double time_limit = 0.950;
  double start_temp = 1000.0;
  double end_temp = 10.0;
  auto best_state = state;
  int iter_count = 0;

  while (timer.time() < time_limit) {
    iter_count++;
    State next_state;
    next_state.gready();
    next_state.calc_score();
    ll next_score = next_state.calc_score();    

    // score maximumize
    // double temp = start_temp + (end_temp - start_temp) * timer.time() / time_limit;
    // double probability = exp((next_score - now_score) / temp);
    // bool is_force_next = probability > rand128();
    
    // if (next_score > score || is_force_next) {
    // if (next_score > score) {
    //   score = next_score;
    //   state = next_state;
    // }
    if (next_score > best_score) {
      best_score = next_score;
      best_state = state;
      cerr << "best score: " << best_score << endl;
    }
  }
  cerr << "best_score:" << "\t" << best_score << endl;
  cerr << "iter_count:" << "\t" << iter_count << endl;
  best_state.print_ans();
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  read_input();
  simulatedAnnealing();
  return 0;
}

yukicoder

結果

テストケース

ソースコード