#141063 (C++14) No.473 和と積の和

提出ソース

結果

問題	No.473 和と積の和
ユーザー	koyumeishi
提出日時	2016-12-23 20:49:53
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	56 ms / 3,000 ms
コード長	4,957 bytes
コンパイル時間	1,375 ms
コンパイル使用メモリ	120,156 KB
実行使用メモリ	94,540 KB
最終ジャッジ日時	2024-05-09 14:43:18
合計ジャッジ時間	2,737 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	34 ms 62,592 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	3 ms 5,376 KB
testcase_12	AC	3 ms 5,632 KB
testcase_13	AC	3 ms 5,760 KB
testcase_14	AC	3 ms 5,760 KB
testcase_15	AC	3 ms 5,632 KB
testcase_16	AC	2 ms 5,632 KB
testcase_17	AC	2 ms 5,376 KB
testcase_18	AC	2 ms 5,376 KB
testcase_19	AC	2 ms 5,376 KB
testcase_20	AC	3 ms 5,632 KB
testcase_21	AC	2 ms 5,376 KB
testcase_22	AC	2 ms 5,376 KB
testcase_23	AC	5 ms 11,648 KB
testcase_24	AC	4 ms 12,544 KB
testcase_25	AC	56 ms 94,540 KB
testcase_26	AC	27 ms 56,064 KB
testcase_27	AC	30 ms 60,160 KB
testcase_28	AC	4 ms 5,632 KB
testcase_29	AC	3 ms 5,376 KB
testcase_30	AC	3 ms 5,376 KB
testcase_31	AC	3 ms 5,376 KB
testcase_32	AC	2 ms 5,376 KB
testcase_33	AC	3 ms 5,632 KB
testcase_34	AC	35 ms 65,920 KB
testcase_35	AC	36 ms 65,920 KB
testcase_36	AC	2 ms 5,632 KB
testcase_37	AC	3 ms 5,504 KB
testcase_38	AC	37 ms 66,048 KB
testcase_39	AC	23 ms 45,184 KB
testcase_40	AC	23 ms 45,184 KB
testcase_41	AC	22 ms 45,184 KB
testcase_42	AC	51 ms 89,728 KB
testcase_43	AC	50 ms 89,856 KB
testcase_44	AC	54 ms 89,728 KB
testcase_45	AC	3 ms 5,376 KB
testcase_46	AC	3 ms 5,376 KB

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

ソースコード

raw source code

#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <functional>
#include <set>
#include <ctime>
#include <random>
#include <chrono>
#include <cassert>
#include <tuple>
#include <utility>
using namespace std;

namespace {
  using Integer = /*long long; /*/__int128;
  template<class T, class S> istream& operator >> (istream& is, pair<T,S>& p){return is >> p.first >> p.second;}
  template<class T> istream& operator >> (istream& is, vector<T>& vec){for(T& val: vec) is >> val; return is;}
  template<class T> istream& operator ,  (istream& is, T& val){ return is >> val;}
  template<class T, class S> ostream& operator << (ostream& os, const pair<T,S>& p){return os << p.first << " " << p.second;}
  template<class T> ostream& operator << (ostream& os, const vector<T>& vec){for(size_t i=0; i<vec.size(); i++) os << vec[i] << (i==vec.size()-1?"":" "); return os;}
  template<class T> ostream& operator ,  (ostream& os, const T& val){ return os << " " << val;}

  template<class H> void print(const H& head){ cout << head; }
  template<class H, class ... T> void print(const H& head, const T& ... tail){ cout << head << " "; print(tail...); }
  template<class ... T> void println(const T& ... values){ print(values...); cout << endl; }

  template<class H> void eprint(const H& head){ cerr << head; }
  template<class H, class ... T> void eprint(const H& head, const T& ... tail){ cerr << head << " "; eprint(tail...); }
  template<class ... T> void eprintln(const T& ... values){ eprint(values...); cerr << endl; }

  class range{ Integer start_, end_, step_; public: struct range_iterator{ Integer val, step_; range_iterator(Integer v, Integer step) : val(v), step_(step) {} Integer operator * (){return val;} void operator ++ (){val += step_;} bool operator != (range_iterator& x){return step_ > 0 ? val < x.val : val > x.val;} }; range(Integer len) : start_(0), end_(len), step_(1) {} range(Integer start, Integer end) : start_(start), end_(end), step_(1) {} range(Integer start, Integer end, Integer step) : start_(start), end_(end), step_(step) {} range_iterator begin(){ return range_iterator(start_, step_); } range_iterator   end(){ return range_iterator(  end_, step_); } };

  inline string operator "" _s (const char* str, size_t size){ return move(string(str)); }
  constexpr Integer my_pow(Integer x, Integer k, Integer z=1){return k==0 ? z : k==1 ? z*x : (k&1) ? my_pow(x*x,k>>1,z*x) : my_pow(x*x,k>>1,z);}
  constexpr Integer my_pow_mod(Integer x, Integer k, Integer M, Integer z=1){return k==0 ? z%M : k==1 ? z*x%M : (k&1) ? my_pow_mod(x*x%M,k>>1,M,z*x%M) : my_pow_mod(x*x%M,k>>1,M,z);}
  constexpr unsigned long long operator "" _ten (unsigned long long value){ return my_pow(10,value); }

  inline int k_bit(Integer x, int k){return (x>>k)&1;} //0-indexed

  mt19937 mt(chrono::duration_cast<chrono::nanoseconds>(chrono::steady_clock::now().time_since_epoch()).count());

  template<class T> string join(const vector<T>& v, const string& sep){ stringstream ss; for(size_t i=0; i<v.size(); i++){ if(i>0) ss << sep; ss << v[i]; } return ss.str(); }

  inline string operator * (string s, int k){ string ret; while(k){ if(k&1) ret += s; s += s; k >>= 1; } return ret; }
}
constexpr long long mod = 9_ten + 7;

long long f(vector<long long>& v){
  while(v.size() > 1){
    long long a = v.back(); v.pop_back();
    long long b = v.back(); v.pop_back();
    v.push_back(a+b + a*b);
  }
  return v.front();
}


vector<long long> divisors(long long N){
  if(N==1) return {1LL};
  
  vector<long long> ret;
  ret.push_back(1);
  ret.push_back(N);

  for(long long i=2; i*i<=N; i++){
    if(N%i==0){
      ret.push_back(i);
      if(i*i != N) ret.push_back(N/i);
    }
  }

  sort(ret.begin(), ret.end());
  return ret;
}

vector<long long> d;
vector<vector<int>> dd;
#include <unordered_map>
#include <array>
array<array<array<int, 500>, 32>, 1344> memo;
long long rec(int x, long long k, int mx_i){
  if( memo[x][k][mx_i] != -1 ) return memo[x][k][mx_i];
  if(k==1){
    return 1;
  }

  long long ret = 0;
  for(int i=mx_i; i<d.size() && d[i]*d[i]<=d[x]; i++){
    if(dd[x][i] == -1) continue;
    long long a = d[i];
    long long b = d[dd[x][i]];
    if(b<a) break;
    if(b < my_pow(a, k-1)) break;
    ret += rec(dd[x][i], k-1, i);
  }
  return memo[x][k][mx_i] = ret;
}

int main(){
  long long n,x;

  cin >> n,x;
  d = divisors(x+1);
  dd = vector<vector<int>>(d.size(), vector<int>(d.size(), -1));
  for(int i=0; i<d.size(); i++){
    for(int j=0; j<d.size(); j++){
      if( d[i]%d[j] == 0 ){
        dd[i][j] = lower_bound(d.begin(), d.end(), d[i]/d[j]) - d.begin();
      }
    }
  }
  for(int k=0; k<d.size(); k++){
    for(int i=0; i<32; i++)
    for(int j=0; j<500; j++){
      memo[k][i][j] = -1;
    }
  }
  if(n>=32){
    println(0);
    return 0;
  }
  cout << rec(d.size()-1, n, 1) << endl;
  return 0;
}

yukicoder

結果

テストケース

ソースコード