結果
問題 | No.995 タピオカオイシクナーレ |
ユーザー | 37zigen |
提出日時 | 2020-02-21 23:04:59 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 707 ms / 2,000 ms |
コード長 | 1,448 bytes |
コンパイル時間 | 3,888 ms |
コンパイル使用メモリ | 77,780 KB |
実行使用メモリ | 59,552 KB |
最終ジャッジ日時 | 2024-10-09 02:05:49 |
合計ジャッジ時間 | 14,572 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
import java.io.PrintWriter; import java.util.Arrays; import java.util.HashMap; import java.util.Scanner; import java.math.*; class Main { public static void main(String[] args) throws Exception { new Main().run(); } long[][] mul(long[][] a,long[][] b){ long[][] ret=new long[2][2]; for(int i=0;i<a.length;++i){ for(int j=0;j<b[i].length;++j){ for(int k=0;k<a[i].length;++k){ ret[i][j]+=a[i][k]*b[k][j]%M; ret[i][j]%=M; } } } return ret; } long[][] mat_pow(long[][] a,long n){ long[][] ret=new long[2][2]; ret[0][0]=ret[1][1]=1; for(;n>0;n>>=1,a=mul(a,a))if(n%2==1)ret=mul(ret,a); return ret; } long pow(long a,long n){ long ret=1; for(;n>0;n>>=1,a=a*a%M)if(n%2==1)ret=ret*a%M; return ret; } long inv(long a){ return pow(a,M-2); } final long M=(long)1e9+7; void run() { Scanner sc=new Scanner(System.in); int n=sc.nextInt(); int m=sc.nextInt(); long K=sc.nextLong(); long p=sc.nextLong(); long q=sc.nextLong(); long[] b=new long[n]; for(int i=0;i<n;++i){ b[i]=sc.nextLong(); } long prob=p*inv(q)%M; long[][] mat={{(M+1-prob)%M,prob},{prob,(M+1-prob)%M}}; long[][] A=mat_pow(mat,K); long ans=0; for(int i=0;i<n;++i){ if(i<m){ ans+=A[0][0]*b[i]%M; }else{ ans+=A[0][1]*b[i]%M; } ans%=M; } System.out.println((ans+M)%M); } static void tr(Object... objects) { System.out.println(Arrays.deepToString(objects)); } }