結果

問題 No.1025 Modular Equation
コンテスト
ユーザー chocorusk
提出日時 2020-03-04 22:55:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 861 bytes
コンパイル時間 826 ms
コンパイル使用メモリ 70,116 KB
実行使用メモリ 13,636 KB
最終ジャッジ日時 2024-11-30 23:57:00
合計ジャッジ時間 155,310 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 7 TLE * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
typedef long long ll;
using namespace std;
const int MOD=1e9+7;
ll powmod(ll a, ll k, ll mod){
    ll ap=a, ans=1;
    while(k){
        if(k&1){
            ans*=ap;
            ans%=mod;
        }
        ap=ap*ap;
        ap%=mod;
        k>>=1;
    }
    return ans;
}
ll a[505], q[100010];
int dp[2][100010];
int main()
{
    int n, k; ll b, p;
	scanf("%lld %d %d %lld", &p, &n, &k, &b);
	for(int i=0; i<n; i++) scanf("%lld", &a[i]);
	for(ll i=0; i<p; i++){
		q[i]=powmod(i, k, p);
	}
	dp[0][0]=1;
	for(int i=0; i<n; i++){
		fill(dp[(i&1)^1], dp[(i&1)^1]+p, 0);
		for(int j=0; j<p; j++){
			for(int l=0; l<p; l++){
				ll m=(j+a[i]*q[l])%p;
				dp[(i&1)^1][m]+=dp[i&1][j];
				if(dp[(i&1)^1][m]>=MOD) dp[(i&1)^1][m]-=MOD;
			}
		}
	}
	printf("%d\n", dp[n&1][b]);
    return 0;
}
0