結果

問題 No.1102 Remnants
ユーザー kotatsugamekotatsugame
提出日時 2020-07-03 21:49:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 139 ms / 2,000 ms
コード長 693 bytes
コンパイル時間 555 ms
コンパイル使用メモリ 66,428 KB
実行使用メモリ 8,652 KB
最終ジャッジ日時 2024-09-17 00:09:06
合計ジャッジ時間 2,789 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:10:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   10 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
using namespace std;
const long mod=1e9+7;
long power(long a,long b){return b?power(a*a%mod,b/2)*(b%2?a:1)%mod:1;}
int N,K;
long A[2<<17];
long S[2<<17];
//H(a,K)=C(a+K-1,K)=C(a+K-1,a-1)
long H[2<<17];
main()
{
	cin>>N>>K;
	for(int i=0;i<N;i++)
	{
		cin>>A[i];
		S[i+1]=S[i]+A[i];
		S[i+1]%=mod;
	}
	if(K==0)
	{
		cout<<S[N]<<endl;
		return 0;
	}
	K--;
	H[1]=1;
	for(int i=2;i<=N;i++)
	{
		H[i]=H[i-1]*(K+i-1)%mod*power(i-1,mod-2)%mod;
	}
	long ans=0,cum=0;
	for(int i=1;i<=N;i++)
	{
		(cum+=H[i])%=mod;
		(ans+=S[i]*cum%mod*H[N-i+1]%mod)%=mod;
	}
	cum=0;
	for(int i=N;i>=1;i--)
	{
		(cum+=H[N-i+1])%=mod;
		(ans+=(mod-S[i-1])*cum%mod*H[i]%mod)%=mod;
	}
	cout<<ans<<endl;
}
0