結果

問題 No.315 世界のなんとか3.5
ユーザー koyumeishi
提出日時 2015-09-20 00:57:22
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
TLE  
実行時間 -
コード長 1,784 bytes
コンパイル時間 686 ms
コンパイル使用メモリ 80,100 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-07-19 08:15:19
合計ジャッジ時間 10,547 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 15 TLE * 2 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

//TLE
//桁DP

#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <set>
using namespace std;

#define MOD 1000000007

int P;

long long func(string& s){
	
	//dp[mod 3*P][lower | has3]

	long long my_mod = 3*P;

	vector<vector<long long>> dp (my_mod, vector<long long>(4, 0));

	dp[0][0] = 1;

	for(int i=0; i<s.size(); i++){
		vector<vector<long long>> dp_(my_mod, vector<long long>(4, 0));

		for(int mod3P=0; mod3P<my_mod; mod3P++)
		for(int lower=0; lower<2; lower++)
		for(int has3=0; has3<2; has3++)
		{
			int state = (lower<<1) | (has3);
			for(int d=0; d<10; d++){
				if(lower==0 && d>s[i]-'0') continue;

				int next_state = (lower || d<s[i]-'0')<<1;
				next_state |= (has3 || d==3);

				int next_mod = (mod3P*10 + d) % my_mod;
			
				long long& dp_now  = dp[mod3P][state];
				long long& dp_next = dp_[next_mod][next_state];
				dp_next += dp_now;
				if(dp_next>=MOD) dp_next -= MOD;
			}
		}

		swap(dp, dp_);
	}

	long long ret = 0;
	for(int mod3P=0; mod3P<my_mod; mod3P++)
	for(int lower=0; lower<2; lower++)
	for(int has3=0; has3<2; has3++)
	{
		int state = (lower<<1) | (has3);
		if( mod3P % P == 0 ) continue;
		if( has3==0 && mod3P%3!=0 ) continue;

		ret += dp[mod3P][state];
		if(ret>=MOD) ret -= MOD;
	}

	return ret;
}

int main(){
	string a,b;
	cin >> a >> b;
	cin >> P;

	long long ans = 0;
	ans += func(b);
	ans -= func(a);
	ans = (ans+MOD)%MOD;

	int mod3 = 0;
	int has3 = 0;
	int modP = 0;
	for(int i=0; i<a.size(); i++){
		if(a[i] == '3') has3 = 1;
		mod3 = (mod3*10 + a[i]-'0') % 3;
		modP = (modP*10 + a[i]-'0') % P;
	}
	if( (mod3==0 || has3==1) && modP!=0 ) ans += 1;

	ans = (ans+MOD)%MOD;

	cout << ans << endl;

	return 0;
}
0