#973455 (C++17) No.2748 Strange Clock

• 問題: No.2748 Strange Clock
• ユーザー: Nzt3
• 提出日時: 2024-04-19 16:07:09
• 言語: C++17 (gcc 13.3.0 + boost 1.89.0)
• 結果: TLE
• 実行時間: -
• コード長: 2,612 bytes
• コンパイル時間: 2,590 ms
• コンパイル使用メモリ: 220,776 KB
• 最終ジャッジ日時: 2025-02-21 03:22:12
• ジャッジサーバーID （参考情報）: judge5 / judge3

ファイルパターン	結果
sample	-- * 2
other	TLE * 1 -- * 36

ソースコード

#include<bits/stdc++.h>
using namespace std;
using ll=long long;

constexpr ll MAX_N=15;

ll ipow(ll a,int n){
  if(n==0)return 1ll;
  if(n%2)return a*ipow(a,n-1);
  ll ret=ipow(a,n/2);
  return ret*ret;
}
array<int,MAX_N> ItoTx(ll n,int x,int N){
  array<int,MAX_N> ret;
  for(int i=0;i<N;i++){
    ret[i]=(n%x);
    n/=x;
  }
  assert(n==0);
  return ret;
}
ll TtoI(array<int,MAX_N> v,int x,int N){
  ll ret=0;
  for(int i=N-1;i>=0;i--){
    ret=ret*x+v[i];
  }
  return ret;
}
namespace Lib {
ll extGCD(ll a, ll b, ll &x, ll &y) {
  if (b == 0) {
    x = 1, y = 0;
    return a;
  }
  ll d = extGCD(b, a % b, y, x);
  y = y - a / b * x;
  return d;
}
ll modinv(ll a, ll m) {
  ll x = 0, y = 0, d = extGCD(a, m, x, y);
  if (d != 1) {
    return -1;
  }
  return x;
}
}  // namespace Lib
namespace Lib {
pair<ll, ll> crt(array<ll,2> r,array<ll,2> m) {
  ll ret_r = r[0], ret_m = m[0];
  constexpr int n = 2;
  for (int i = 1; i < n; i++) {
    ll x = 0, y = 0, d = extGCD(ret_m, m[i], x, y);
    if ((ret_r - r[i]) % d != 0) {
      return make_pair(-1ll, -1ll);
    }
    ll tmp = x * (r[i] - ret_r) / d % (m[i] / d);
    ret_r += tmp * ret_m;
    ret_m *= m[i] / d;
    ret_r %= ret_m;
    if (ret_r < 0) ret_r += ret_m;
  }
  if (ret_r < 0) ret_r += ret_m;
  return make_pair(ret_r, ret_m);
}
}  // namespace Lib


int main(){
  vector Ts(MAX_N,vector(0,0ll));
  vector Cnts(MAX_N,vector(0,0ll));
  for(ll N=1;N<=MAX_N;N++){
    ll V_MAX3=ipow(3,N),V_MAX4=ipow(4,N);
    ll p2=ipow(2,N);
    unordered_map<ll,vector<ll>> cnt;
    for(int i=0;i<V_MAX3;i++){
      auto v3=ItoTx(i,3,N);
      // for(int j:v3)cerr<<j;
      // cerr<<'\n';
      ll i3=TtoI(v3,3,N),i4=TtoI(v3,4,N),i6=TtoI(v3,6,N);
      ll d=((((i4-i6)%p2+p2)%p2)<<30)+(((i3-i6)%V_MAX3+V_MAX3)%V_MAX3);
      array<ll,2> r={i3,i4},m={V_MAX3,V_MAX4};
      auto ret=Lib::crt(r,m);
      cnt[d].push_back(ret.first);
    }
    map<ll,ll>ans;
    for(auto [i,v]:cnt){
      if(v.empty())continue;
      sort(v.begin(),v.end());
      v.push_back(v[0]+V_MAX3*V_MAX4);
      for(int i=1;i<(int)v.size();i++){
        ans[v[i]-v[i-1]]+=1;
      }
    }
    for(auto i:ans){
      Ts[N-1].push_back(i.first);
      Cnts[N-1].push_back(i.second);
    }
  }
  cout<<"vector<vector<ll>> Ts={\n";
  for(int i=0;i<MAX_N;i++){
    cout<<"{";
    for(int j=0;j<(int)Ts[i].size();j++)cout<<Ts[i][j]<<(j==(int)Ts[i].size()-1?i==MAX_N-1?"}\n":"},\n":",");
  }
  cout<<"},\nCnts={\n";
  for(int i=0;i<MAX_N;i++){
    cout<<"{";
    for(int j=0;j<(int)Ts[i].size();j++)cout<<Cnts[i][j]<<(j==(int)Ts[i].size()-1?i==MAX_N-1?"}\n":"},\n":",");
  }
  cout<<"};\n";
}