結果

問題 No.995 タピオカオイシクナーレ
ユーザー hiro71687k
提出日時 2023-04-16 18:23:05
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 1,232 bytes
コンパイル時間 3,899 ms
コンパイル使用メモリ 251,696 KB
最終ジャッジ日時 2025-02-12 09:15:57
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using ld=long double;
ld pie=3.141592653589793;
ll inf=1444999999;
ll mod=1000000007;
ll modpow(ll x, ll n) {
  x = x%mod;
  if(n==0) return 1;  //再帰の終了条件

  else if(n%2==1) {
    return (x*modpow(x, n-1))%mod;  //nが奇数ならnを1ずらす
  }
  else return modpow((x*x)%mod, n/2)%mod;  //nが偶数ならnが半分になる
}
int main(){
    ll n,m,k,p,q;
    cin >> n >> m >> k >> p >> q;
    vector<ll>c(n);
    for (ll i = 0; i < n; i++)
    {
        cin >> c[i];
    }
    ll a=p;
    ll b=q-p;
    ll x=modpow((a+b)%mod,k);
    ll y=modpow((b-a+mod)%mod,k);
    ll ans=0;
    ll qq=inv_mod(modpow(q,k),mod);
    for (ll i = 0; i < m; i++)
    {
        ll xx=x+y;
        xx%=mod;
        xx*=qq;
        xx%=mod;
        xx*=inv_mod(2,mod);
        xx%=mod;
        xx*=c[i];
        xx%=mod;
        ans+=xx;
        ans%=mod;
    }
    for (ll i = m; i < n; i++)
    {
        ll xx=x-y+mod;
        xx%=mod;
        xx*=qq;
        xx%=mod;
        xx*=inv_mod(2,mod);
        xx%=mod;
        xx*=c[i];
        xx%=mod;
        ans+=xx;
        ans%=mod;
    }
    cout << ans << endl;
}
0