#364885 (C++17) No.573 a^2[i] = a[i]

提出ソース

結果

問題	No.573 a^2[i] = a[i]
ユーザー	Haar
提出日時	2019-08-02 18:12:09
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	12 ms / 2,000 ms
コード長	3,279 bytes
コンパイル時間	1,973 ms
コンパイル使用メモリ	203,808 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-07-05 08:14:03
合計ジャッジ時間	3,235 ms
ジャッジサーバーID （参考情報）	judge4 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	6 ms 6,812 KB
testcase_01	AC	6 ms 6,940 KB
testcase_02	AC	6 ms 6,940 KB
testcase_03	AC	5 ms 6,940 KB
testcase_04	AC	6 ms 6,944 KB
testcase_05	AC	5 ms 6,940 KB
testcase_06	AC	5 ms 6,940 KB
testcase_07	AC	5 ms 6,944 KB
testcase_08	AC	5 ms 6,944 KB
testcase_09	AC	5 ms 6,940 KB
testcase_10	AC	5 ms 6,940 KB
testcase_11	AC	5 ms 6,940 KB
testcase_12	AC	4 ms 6,940 KB
testcase_13	AC	5 ms 6,940 KB
testcase_14	AC	5 ms 6,944 KB
testcase_15	AC	5 ms 6,944 KB
testcase_16	AC	5 ms 6,940 KB
testcase_17	AC	5 ms 6,940 KB
testcase_18	AC	5 ms 6,944 KB
testcase_19	AC	5 ms 6,944 KB
testcase_20	AC	5 ms 6,940 KB
testcase_21	AC	6 ms 6,940 KB
testcase_22	AC	4 ms 6,940 KB
testcase_23	AC	4 ms 6,944 KB
testcase_24	AC	5 ms 6,940 KB
testcase_25	AC	5 ms 6,940 KB
testcase_26	AC	5 ms 6,944 KB
testcase_27	AC	5 ms 6,940 KB
testcase_28	AC	5 ms 6,940 KB
testcase_29	AC	5 ms 6,944 KB
testcase_30	AC	5 ms 6,940 KB
testcase_31	AC	5 ms 6,940 KB
testcase_32	AC	5 ms 6,944 KB
testcase_33	AC	5 ms 6,944 KB
testcase_34	AC	5 ms 6,940 KB
testcase_35	AC	5 ms 6,940 KB
testcase_36	AC	5 ms 6,940 KB
testcase_37	AC	5 ms 6,944 KB
testcase_38	AC	6 ms 6,940 KB
testcase_39	AC	5 ms 6,940 KB
testcase_40	AC	6 ms 6,940 KB
testcase_41	AC	5 ms 6,940 KB
testcase_42	AC	5 ms 6,940 KB
testcase_43	AC	5 ms 6,940 KB
testcase_44	AC	6 ms 6,944 KB
testcase_45	AC	5 ms 6,940 KB
testcase_46	AC	12 ms 6,944 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>
#define LLI long long int
#define FOR(v, a, b) for(LLI v = (a); v < (b); ++v)
#define FORE(v, a, b) for(LLI v = (a); v <= (b); ++v)
#define REP(v, n) FOR(v, 0, n)
#define REPE(v, n) FORE(v, 0, n)
#define REV(v, a, b) for(LLI v = (a); v >= (b); --v)
#define ALL(x) (x).begin(), (x).end()
#define RALL(x) (x).rbegin(), (x).rend()
#define ITR(it, c) for(auto it = (c).begin(); it != (c).end(); ++it)
#define RITR(it, c) for(auto it = (c).rbegin(); it != (c).rend(); ++it)
#define EXIST(c,x) ((c).find(x) != (c).end())
#define fst first
#define snd second
#define popcount __builtin_popcount
#define UNIQ(v) (v).erase(unique(ALL(v)), (v).end())
#define bit(i) (1LL<<(i))

#ifdef DEBUG
#include <misc/C++/Debug.cpp>
#else
#define dump(...) ((void)0)
#endif

#define gcd __gcd

using namespace std;
template <class T> constexpr T lcm(T m, T n){return m/gcd(m,n)*n;}

template <typename I> void join(ostream &ost, I s, I t, string d=" "){for(auto i=s; i!=t; ++i){if(i!=s)ost<<d; ost<<*i;}ost<<endl;}
template <typename T> istream& operator>>(istream &is, vector<T> &v){for(auto &a : v) is >> a; return is;}

template <typename T, typename U> bool chmin(T &a, const U &b){return (a>b ? a=b, true : false);}
template <typename T, typename U> bool chmax(T &a, const U &b){return (a<b ? a=b, true : false);}
template <typename T, size_t N, typename U> void fill_array(T (&a)[N], const U &v){fill((U*)a, (U*)(a+N), v);}


LLI power(LLI n, LLI p, LLI m){
  LLI ret = 1;
  while(p>0){
    if(p&1) (ret *= n) %= m;
    (n *= n) %= m;
    p /= 2;
  }
  return ret;
}

LLI mod_inv(LLI n, LLI p){return power(n,p-2,p);}


/**
 * @attention 使用前にinit関数を呼び出す
 */
template <LLI MOD> class CombUtils{
public:
  static vector<LLI> facto;
  static vector<LLI> ifacto;

  static void init(int N){
    facto.assign(N+1, 1);
    ifacto.assign(N+1, 1);

    FORE(i,1,N){
      (facto[i] = facto[i-1] * i) %= MOD;
    }

    ifacto[N] = mod_inv(facto[N],MOD);

    REV(i,N-1,0){
      ifacto[i] = ifacto[i+1] * (i+1) % MOD;
    }
  }

  static LLI factorial(LLI i){
    assert(i < facto.size());
    return facto[i];
  }
  
  static LLI factorial_inverse(LLI i){
    assert(i < ifacto.size());
    return ifacto[i];
  }

  static LLI P(LLI n, LLI k);
  static LLI C(LLI n, LLI k);
  static LLI H(LLI n, LLI k);
  static LLI stirling_number(LLI n, LLI k);
  static LLI bell_number(LLI n, LLI k);
};

template <LLI MOD> vector<LLI> CombUtils<MOD>::facto = vector<LLI>();
template <LLI MOD> vector<LLI> CombUtils<MOD>::ifacto = vector<LLI>();

template <LLI MOD> LLI CombUtils<MOD>::P(LLI n, LLI k){
  if(n < k or n < 0 or k < 0) return 0;
  return factorial(n) * factorial_inverse(n-k) % MOD;
}

template <LLI MOD> LLI CombUtils<MOD>::C(LLI n, LLI k){
  if(n < k or n < 0 or k < 0) return 0;
  return P(n,k) * factorial_inverse(k) % MOD;
}

template <LLI MOD> LLI CombUtils<MOD>::H(LLI n, LLI k){
  if(n == 0 and k == 0) return 1;
  return C(n+k-1, k);
}


const LLI mod = 1e9+7;

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  CombUtils<mod>::init(200000);

  int N;
  while(cin >> N){
    LLI ans = 0;

    FORE(i,0,N){
      (ans += CombUtils<mod>::C(N,i) * power(i,N-i,mod) % mod) %= mod;
    }
    

    cout << ans << endl;
  }
  
  return 0;
}

yukicoder

結果

テストケース

ソースコード