結果

問題 No.1102 Remnants
ユーザー saxofone111saxofone111
提出日時 2021-04-07 23:11:15
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 180 ms / 2,000 ms
コード長 4,367 bytes
コンパイル時間 5,032 ms
コンパイル使用メモリ 203,716 KB
実行使用メモリ 9,440 KB
最終ジャッジ日時 2023-09-05 07:01:08
合計ジャッジ時間 8,127 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 103 ms
8,176 KB
testcase_01 AC 104 ms
8,048 KB
testcase_02 AC 103 ms
8,116 KB
testcase_03 AC 103 ms
8,068 KB
testcase_04 AC 103 ms
8,016 KB
testcase_05 AC 138 ms
9,044 KB
testcase_06 AC 113 ms
8,280 KB
testcase_07 AC 180 ms
9,440 KB
testcase_08 AC 104 ms
8,068 KB
testcase_09 AC 105 ms
8,204 KB
testcase_10 AC 103 ms
8,072 KB
testcase_11 AC 105 ms
8,104 KB
testcase_12 AC 105 ms
8,136 KB
testcase_13 AC 105 ms
8,252 KB
testcase_14 AC 128 ms
8,480 KB
testcase_15 AC 122 ms
8,224 KB
testcase_16 AC 153 ms
9,176 KB
testcase_17 AC 152 ms
9,136 KB
testcase_18 AC 170 ms
9,120 KB
testcase_19 AC 125 ms
8,408 KB
testcase_20 AC 108 ms
8,068 KB
testcase_21 AC 144 ms
8,592 KB
testcase_22 AC 162 ms
8,876 KB
testcase_23 AC 147 ms
8,648 KB
testcase_24 AC 134 ms
8,284 KB
testcase_25 AC 171 ms
9,060 KB
testcase_26 AC 106 ms
8,132 KB
testcase_27 AC 120 ms
8,064 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

#define MOD 1000000007
#define rep(i, n) for(ll i=0; i < (n); i++)
#define rrep(i, n) for(ll i=(n)-1; i >=0; i--)
#define ALL(v) v.begin(),v.end()
#define rALL(v) v.rbegin(),v.rend()
#define FOR(i, j, k) for(ll i=j;i<k;i++)
#define debug_print(var) cerr << #var << "=" << var <<endl;
#define DUMP(i, v)for(ll i=0;i<v.size();i++)cerr<<v[i]<<" "
#define fi first
#define se second

using namespace std;
typedef long long int ll;
typedef vector<ll> llvec;
typedef vector<double> dvec;
typedef pair<ll, ll> P;
typedef long double ld;
struct edge{ll x, c;};

ll mod(ll a, ll mod){
  ll res = a%mod;
  if(res<0)res=res + mod;
  return res;
}

ll modpow(ll a, ll n, ll mod){
  ll res=1;
  while(n>0){
    if(n&1) res=res*a%mod;
    a=a*a%mod;
    n>>=1;
  }
  return res;
}

ll modinv(ll a, ll mod){
  ll b=mod, u=1, v=0;
  while(b){
    ll t=a/b;
    a-=t*b; swap(a, b);
    u-=t*v; swap(u, v);
  }
  u%=mod;
  if(u<0)u+=mod;
  return u;
}

ll gcd(ll a, ll b){
  ll r = a%b;
  if(r==0) return b;
  else return gcd(b, a%b);
}

// @param b 1
// @return pair(g, x) s.t. g = gcd(a, b), xa = g (mod b), 0 <= x < b/g
std::pair<long long, long long> inv_gcd(long long a, long long b) {
  a = mod(a, b);
  if (a == 0) return {b, 0};

  // Contracts:
  // [1] s - m0 * a = 0 (mod b)
  // [2] t - m1 * a = 0 (mod b)
  // [3] s * |m1| + t * |m0| <= b
  long long s = b, t = a;
  long long m0 = 0, m1 = 1;

  while (t) {
    long long u = s / t;
    s -= t * u;
    m0 -= m1 * u;  // |m1 * u| <= |m1| * s <= b

    // [3]:
    // (s - t * u) * |m1| + t * |m0 - m1 * u|
    // <= s * |m1| - t * u * |m1| + t * (|m0| + |m1| * u)
    // = s * |m1| + t * |m0| <= b

    auto tmp = s;
    s = t;
    t = tmp;
    tmp = m0;
    m0 = m1;
    m1 = tmp;
  }
  // by [3]: |m0| <= b/g
  // by g != b: |m0| < b/g
  if (m0 < 0) m0 += b / s;
  return {s, m0};
}

// (rem, mod)
std::pair<long long, long long> crt(const std::vector<long long>& r,
                                    const std::vector<long long>& m) {
  assert(r.size() == m.size());
  int n = int(r.size());
  // Contracts: 0 <= r0 < m0
  long long r0 = 0, m0 = 1;
  for (int i = 0; i < n; i++) {
    assert(1 <= m[i]);
    long long r1 = mod(r[i], m[i]), m1 = m[i];
    if (m0 < m1) {
      std::swap(r0, r1);
      std::swap(m0, m1);
    }
    if (m0 % m1 == 0) {
      if (r0 % m1 != r1) return {0, 0};
      continue;
    }
    // assume: m0 > m1, lcm(m0, m1) >= 2 * max(m0, m1)

    // (r0, m0), (r1, m1) -> (r2, m2 = lcm(m0, m1));
    // r2 % m0 = r0
    // r2 % m1 = r1
    // -> (r0 + x*m0) % m1 = r1
    // -> x*u0*g % (u1*g) = (r1 - r0) (u0*g = m0, u1*g = m1)
    // -> x = (r1 - r0) / g * inv(u0) (mod u1)

    // im = inv(u0) (mod u1) (0 <= im < u1)
    long long g, im;
    std::tie(g, im) = inv_gcd(m0, m1);

    long long u1 = (m1 / g);
    // |r1 - r0| < (m0 + m1) <= lcm(m0, m1)
    if ((r1 - r0) % g) return {0, 0};

    // u1 * u1 <= m1 * m1 / g / g <= m0 * m1 / g = lcm(m0, m1)
    long long x = (r1 - r0) / g % u1 * im % u1;

    // |r0| + |m0 * x|
    // < m0 + m0 * (u1 - 1)
    // = m0 + m0 * m1 / g - m0
    // = lcm(m0, m1)
    r0 += x * m0;
    m0 *= u1;  // -> lcm(m0, m1)
    if (r0 < 0) r0 += m0;
  }
  return {r0, m0};
}


bool is_prime(ll n){
  ll i = 2;
  if(n==1)return false;
  if(n==2)return true;
  bool res = true;
  while(i*i <n){
    if(n%i==0){
      res = false;
    }
    i = i+1;
  }

  //if(i==1)res = false;
  if(n%i==0)res=false;
  return res;
}

#define fracMax  300000
ll frac[fracMax];
ll ifrac[fracMax];

void gen(){
  frac[0] = 1;
  ifrac[0] = 1;
  rep(i, fracMax-1){
    frac[i+1] = mod((i+1)*frac[i], MOD);
    ifrac[i+1] = modinv(frac[i+1], MOD);   
  }
  return;
}

ll binom(ll n, ll k){
  if(k<0 or k>n){
    return 0;
  }else{
    return mod(mod(frac[n]*ifrac[n-k], MOD)*ifrac[k], MOD);
  }
}


/**************************************
** A main function starts from here  **
***************************************/
int main(){
  ll N, K;
  cin >> N >> K;
  ll ans = 0;
  gen();
  llvec frac2(N+1, 0);
  frac2[0] = 1;
  rep(i, N){
    frac2[i+1] = mod(frac2[i]*(K+1+i), MOD);
  }
  rep(i, N){
    ll a;
    cin >> a;
    ll tmp = a;
    tmp = mod(tmp*frac2[i], MOD);
    tmp = mod(tmp*ifrac[i], MOD);
    tmp = mod(tmp*frac2[N-i-1], MOD);
    tmp = mod(tmp*ifrac[N-i-1], MOD);
    ans = mod(ans + tmp, MOD);
  }
  cout << ans << endl;
  return 0;
}
0