#448339 (C++11) No.1011 Infinite Stairs

• 問題: No.1011 Infinite Stairs
• ユーザー: DAyamaCTF
• 提出日時: 2020-03-23 14:16:41
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 17 ms / 2,000 ms
• コード長: 1,241 bytes
• コンパイル時間: 2,091 ms
• コンパイル使用メモリ: 145,320 KB
• 実行使用メモリ: 5,044 KB
• 最終ジャッジ日時: 2023-09-24 11:08:01
• 合計ジャッジ時間: 3,752 ms
• ジャッジサーバーID （参考情報）: judge12 / judge14

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	16 ms; 4,864 KB
testcase_01	AC	17 ms; 5,040 KB
testcase_02	AC	16 ms; 4,916 KB
testcase_03	AC	16 ms; 4,908 KB
testcase_04	AC	17 ms; 4,856 KB
testcase_05	AC	17 ms; 4,852 KB
testcase_06	AC	17 ms; 4,852 KB
testcase_07	AC	17 ms; 5,000 KB
testcase_08	AC	16 ms; 5,040 KB
testcase_09	AC	16 ms; 4,852 KB
testcase_10	AC	17 ms; 4,900 KB
testcase_11	AC	16 ms; 4,948 KB
testcase_12	AC	17 ms; 5,044 KB
testcase_13	AC	17 ms; 4,944 KB
testcase_14	AC	17 ms; 4,852 KB
testcase_15	AC	16 ms; 4,864 KB
testcase_16	AC	17 ms; 4,852 KB
testcase_17	AC	17 ms; 4,904 KB
testcase_18	AC	16 ms; 4,900 KB
testcase_19	AC	16 ms; 4,848 KB
testcase_20	AC	16 ms; 4,996 KB
testcase_21	AC	17 ms; 4,908 KB
testcase_22	AC	16 ms; 4,916 KB
testcase_23	AC	16 ms; 4,976 KB
testcase_24	AC	17 ms; 4,912 KB
testcase_25	AC	16 ms; 4,908 KB
testcase_26	AC	16 ms; 4,852 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll> P;

#define fi first
#define se second
#define repl(i,a,b) for(ll i=(ll)(a);i<(ll)(b);i++)
#define rep(i,n) repl(i,0,n)
#define all(x) (x).begin(),(x).end()
#define dbg(x) cout<<#x"="<<x<<endl
#define mmax(x,y) (x>y?x:y)
#define mmin(x,y) (x<y?x:y)
#define maxch(x,y) x=mmax(x,y)
#define minch(x,y) x=mmin(x,y)
#define uni(x) x.erase(unique(all(x)),x.end())
#define exist(x,y) (find(all(x),y)!=x.end())
#define bcnt __builtin_popcountll

#define INF 1e16
#define mod 1000000007

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

ll fac[100010],finv[100010];

ll comb(ll n,ll r){
	if(n<0||r<0||n<r)return 0;
	else return (fac[n]*finv[r]%mod)*finv[n-r]%mod;
}

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

  fac[0]=1;
  rep(i,100000)fac[i+1]=fac[i]*(i+1)%mod;
  rep(i,100001)finv[i]=mod_pow(fac[i],mod-2);

  ll N,D,K;
  cin>>N>>D>>K;
  ll S=K-N;
  ll res=0;
  rep(i,N+1){
  	ll nS=S-i*D;
  	if(nS<0)continue;
  	ll v=comb(nS+(N-1),N-1)*comb(N,i)%mod;
  	if(i%2==0) (res+=v)%=mod;
  	else (res+=mod-v)%=mod;
  }
  cout<<res<<endl;

  return 0;
}