結果

問題 No.315 世界のなんとか3.5
ユーザー koyumeishi
提出日時 2015-12-07 23:43:13
言語 Go
(1.23.4)
結果
AC  
実行時間 345 ms / 2,000 ms
コード長 2,099 bytes
コンパイル時間 12,030 ms
コンパイル使用メモリ 230,216 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-10 20:30:23
合計ジャッジ時間 16,989 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

package main

import (
	"fmt"
	"bufio"
	"os"
	"strconv"
	"strings"
)

//var sc = bufio.NewScanner(os.Stdin);
var rd = bufio.NewReaderSize(os.Stdin, 1000000);
/*
func nextInt() int{
	sc.Scan();
	ret,_ := strconv.Atoi(sc.Text());
	return ret;
}

func nextStr() string{
	sc.Scan();
	ret := sc.Text();
	return ret;
}
*/

var mod int64 = 1000000007;

func calc(x *string, p int) int64{
	var dp  [2][800][16]int64;
	dp[0][0][0] = 1;
	var sz int = len(*x); 
	for i :=0; i<sz; i++ {
		k := int((*x)[i] - '0');
		mod8 := p;
		if(i<sz-5){
			mod8 = 1;
		}
		for j:=0; j<mod8; j++{
			for a:=0; a<2; a++ {
				for b:=0; b<3; b++ {
					for c:=0; c<2; c++ {
						var s int = (a<<3)|(b<<1)|c;
						var val int64 = dp[i&1][j][s];
						dp[i&1][j][s] = 0;
						for d:=0; d<10; d++ {
							if(a==0 && d>k){
								break;
							}
							var a_ int = a;
							if(d<k){
								a_ |= 1;
							}
							var b_ int = (b+d)%3;
							var c_ int = c;
							if(d==3){
								c_ |= 1;
							}
							var s_ int = (a_<<3)|(b_<<1)|c_;
							var j_ int = 0;
							if( i>=sz-5 ){
								j_ = (j*10+d)%p;
							}
							dp[(i&1)^1][j_][s_] += val;
							if(dp[(i&1)^1][j_][s_] >= mod){
								dp[(i&1)^1][j_][s_] -= mod;
							}
						}
					}
				}
			}
		}
	}

	var ret int64 = 0;
	for j:=1; j<p; j++ {
		for a:=0; a<2; a++ {
			for b:=0; b<3; b++ {
				for c:=0; c<2; c++ {
					if(b!=0 && c==0){
						continue;
					}
					s:=(a<<3)|(b<<1)|c;
					ret += dp[sz&1][j][s];
					if(ret >= mod){
						ret -= mod;
					}
				}
			}
		}
	}
	return ret;
}

func main(){
	//sc.Split(bufio.ScanWords);
	/*
	var a string = nextStr();
	var b string = nextStr();
	var p int = nextInt();
	*/
	in,_,_ := rd.ReadLine();
	tmp := strings.Split(string(in), " ");
	var a, b string = tmp[0], tmp[1];
	p,_ := strconv.Atoi(tmp[2]);

	var ans int64 = 0;
	ans += calc(&b, p);
	ans -= calc(&a, p);
	ans = (ans+mod)%mod;

	var x,y,z int = 0,0,0;
	for i:=0; i<len(a); i++ {
		k := int(a[i] - '0');
		x = (x+k)%3;
		if(k==3){
			y = 1;
		}
		z = (z*10+k)%p;
	}
	if((x==0||y==1)&&z!=0){
		ans++;
		ans %= mod;
	}
	fmt.Println(ans);

}
0