結果

問題 No.995 タピオカオイシクナーレ
コンテスト
ユーザー hipotaf
提出日時 2020-06-11 13:19:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 1,846 bytes
コンパイル時間 647 ms
コンパイル使用メモリ 75,024 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-24 02:55:13
合計ジャッジ時間 2,083 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <cmath>

using namespace std;

using ll = long long;

template <class T>
using grid = vector<vector<T>>;

#define REP(i,n) for(ll i=0;i<(ll)(n);i++)
#define REPD(i,n) for(ll i=n-1;i>=0;i--)
#define FOR(i,a,b) for(ll i=a;i<=(ll)(b);i++)
#define FORD(i,a,b) for(ll i=a;i>=(ll)(b);i--)

#define input(...) __VA_ARGS__; in(__VA_ARGS__)

void print() {
  std::cout << std::endl;
}

template <class Head, class... Tail>
void print(Head&& head, Tail&&... tail) {
  std::cout << head << " ";
  print(std::forward<Tail>(tail)...);
}

void in() { }

template <class Head, class... Tail>
void in(Head&& head, Tail&&... tail) {
  cin >> head;
  in(std::forward<Tail>(tail)...);
}

ll MOD = 1000000007;
ll powmod(ll v, ll x) {
  if (x == 0) return 1;
  if (x % 2 == 0) {
    ll tmp = powmod(v, x / 2);
    return (tmp * tmp) % MOD;
  } else {
    return (powmod(v, x - 1) * v) % MOD;
  }
}

ll complementary(ll x) {
  return (1 - x + MOD) % MOD;
}

ll P, Q;
ll e_value(ll k) {
  if (k == 1) {
    // return 1 - P / Q;
    return complementary((P * powmod(Q, MOD - 2)) % MOD);
  }

  if (k % 2 == 0) {
    ll even = e_value(k / 2);
    ll odd = complementary(even);

    return ((even * even) % MOD + (odd * odd) % MOD) % MOD;
  } else {
    ll even_one = e_value(1);
    ll even_other = e_value(k - 1);

    ll odd_one = complementary(even_one);
    ll odd_other = complementary(even_other);

    return ((even_one * even_other) % MOD + (odd_one * odd_other) % MOD) % MOD;
  }
}

int main() {
  ll input(n, m, k, p, q);
  P = p;
  Q = q;
  vector<ll> b(n);
  REP(i, n) cin >> b[i];

  ll frn = e_value(k) % MOD;
  ll bck = (1 - frn + MOD) % MOD;

  ll prob = 0;
  REP(i, m) {
    (prob += b[i] * frn) %= MOD;
  }
  FOR(i, m, n - 1) {
    (prob += b[i] * bck) %= MOD;
  }

  print(prob % MOD);
}
0