結果

問題 No.802 だいたい等差数列
ユーザー Ryuhei Mori
提出日時 2019-09-21 00:13:18
言語 C
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,240 bytes
コンパイル時間 1,412 ms
コンパイル使用メモリ 30,720 KB
実行使用メモリ 13,568 KB
最終ジャッジ日時 2024-09-15 06:08:15
合計ジャッジ時間 3,381 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 20 WA * 9 RE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>

const int mod = 1000000007;
int n, m, d1, d2;

/*
[x^m] 1/(1-x) x/(1-x) (x^d1 + ... + x^d2)^{n-1}
[x^m] x/(1-x)^2 x^{d1 (n-1)} [(1 - x^{d2-d1+1}) / (1-x) ]^{n-1}
[x^{m-d1(n-1)-1}] (1 - x^{d2-d1+1})^{n-1} / (1-x)^{n+1}

[x^{k}] (1 - x^{e})^{n-1} / (1-x)^{n+1}

\sum_i (-1)^i\binom{n-1}{i} \binom{k-ie+n}{n}
=
\sum_i (-1)^i (k-ie+n)! / (n i! (n-1-i)! (k-ie)!) 
*/

int modpow(int x, int n){
  int r=1;
  for(;n;n>>=1){
    if(n&1) r = (long long) r * x % mod;
    x = (long long) x * x % mod;
  }
  return r;
}

int fact[1500000];
int ifact[1500000];

int main(){
  int i, k, e, r;
  scanf("%d%d%d%d",&n,&m,&d1,&d2);
  fact[0] = 1;
  for(i=1;i<sizeof(fact)/sizeof(fact[0]);i++)
    fact[i] = (long long) fact[i-1] * i % mod;
  i--;
  ifact[i]=modpow(fact[i], mod-2);
  for(i--;i>=0;i--)
    ifact[i] = (long long) ifact[i+1] * (i+1) % mod;
  k = m - d1*(n-1) - 1;
  e = d2 - d1 + 1;

  if(k < 0){
    puts("0");
    return 0;
  }

  r = 0;
  for(i=0; i<=n-1 && i*e <= k; i++){
    int z;
    z = (long long) fact[k-i*e+n] * ifact[i] % mod * ifact[n-1-i] % mod * ifact[k-i*e] % mod;
    if(i&1) r -= z;
    else r += z;
    r = r % mod;
  }
  r = (long long) r * modpow(n, mod-2) % mod;
  printf("%d\n", r);

  return 0;
}
0