結果
問題 | No.823 Many Shifts Easy |
ユーザー |
|
提出日時 | 2019-10-12 20:20:59 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 14 ms / 2,000 ms |
コード長 | 534 bytes |
コンパイル時間 | 677 ms |
コンパイル使用メモリ | 65,664 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-11-29 03:30:52 |
合計ジャッジ時間 | 1,284 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 10 |
コンパイルメッセージ
main.cpp:9:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type] 9 | main() | ^~~~
ソースコード
#include<iostream> using namespace std; long mod=1e9+7; long power(long a,long b){return b?power(a*a%mod,b/2)*(b%2?a:1)%mod:1;} long F[1<<17],I[1<<17]; int N,K; long P(int a,int b){return a<b||b<0?0:F[a]*I[a-b]%mod;} long C(int a,int b){return F[a]*I[b]%mod*I[a-b]%mod;} main() { cin>>N>>K; F[0]=1; for(int i=1;i<=N;i++)F[i]=F[i-1]*i%mod; I[N]=power(F[N],mod-2); for(int i=N;i--;)I[i]=I[i+1]*(i+1)%mod; long ans=0; for(int k=1;k<=N;k++) { (ans+=(P(N-1,K)+(k<N)*P(N-2,K-2)*C(K,2)%mod)%mod*k%mod)%=mod; } cout<<ans<<endl; }