結果

問題 No.260 世界のなんとか3
ユーザー bal4ubal4u
提出日時 2019-08-31 05:27:11
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 6 ms / 2,000 ms
コード長 2,443 bytes
コンパイル時間 1,355 ms
コンパイル使用メモリ 35,140 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-05-02 21:06:58
合計ジャッジ時間 2,750 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 6 ms
5,376 KB
testcase_04 AC 6 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 4 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 5 ms
5,376 KB
testcase_11 AC 4 ms
5,376 KB
testcase_12 AC 3 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 4 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 4 ms
5,376 KB
testcase_17 AC 3 ms
5,376 KB
testcase_18 AC 3 ms
5,376 KB
testcase_19 AC 4 ms
5,376 KB
testcase_20 AC 3 ms
5,376 KB
testcase_21 AC 3 ms
5,376 KB
testcase_22 AC 4 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 3 ms
5,376 KB
testcase_25 AC 3 ms
5,376 KB
testcase_26 AC 4 ms
5,376 KB
testcase_27 AC 1 ms
5,376 KB
testcase_28 AC 6 ms
5,376 KB
testcase_29 AC 6 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// yukicoder: 260 世界のなんとか3
// 2019.8.31 bal4u

#include <stdio.h>
#include <string.h>

typedef long long ll;

#if 0
#define gc() getchar_unlocked()
#else
#define gc() getchar()
#endif

int ins(char *s) { // 文字列の入力　スペース以下の文字で入力終了
	char *p = s; int c;
	do c = gc(), *s++ = c & 0xf;
	while (c > ' ');
	*--s = 0;
	return s - p;
}

#define MOD 1000000007

char AA[10005], *A; int WA;
char B[10005]; int WB;
int dq[10005][2][3][2];   // 3の有無, mod 3, mod 8, 元の数
int dp[5][2][3][8][2];   // 3である桁, mod 3, mod 8, 元の数
int C[4] = {0,1,2,4};

inline static void chmod(int *a, int b) { *a = ((ll)(*a) + b) % MOD; }

int calc(int w, char *s) {
	int i, j, k, n, i3, m3, m8;
	
	memset(dp, 0, sizeof(dp));
	if (w > 3) {
		int ww = w-3;
		memset(dq, 0, sizeof(dq));
		dq[0][0][0][1] = 1;
		for (i=0; i<ww; i++) for (i3=0; i3<2; i3++) for (m3=0; m3<3; m3++) for (k=0; k<2; k++) if (dq[i][i3][m3][k]) {
			if (k) {
               	for (j = s[i]-1; j >= 0; j--) chmod(&dq[i+1][i3|(j==3)][(m3+j)%3][0], dq[i][i3][m3][k]);
           		chmod(&dq[i+1][i3|(s[i]==3)][(m3+s[i])%3][k], dq[i][i3][m3][k]);
           	} else for (j = 9; j >= 0; j--) chmod(&dq[i+1][i3|(j==3)][(m3+j)%3][k], dq[i][i3][m3][k]);
        }
		for (i3=0; i3<2; i3++) for (m3=0; m3<3; m3++) for (k=0; k<2; k++) dp[0][i3][m3][0][k] = dq[ww][i3][m3][k];
	} else dp[0][0][0][0][1] = 1, i = 0;

    for(n=0;i<w;i++,n++)for(i3=0;i3<2;i3++)for(m3=0;m3<3;m3++)for(m8=0;m8<8;m8++)for(k=0;k<2;k++)if(dp[n][i3][m3][m8][k]){
		if (k) {
            for (j = s[i]-1; j >= 0; j--) chmod(&dp[n+1][i3|(j==3)][(m3+j)%3][(m8+C[w-i]*j)&7][0], dp[n][i3][m3][m8][k]);
           	chmod(&dp[n+1][i3|(s[i]==3)][(m3+s[i])%3][(m8+C[w-i]*s[i])&7][k], dp[n][i3][m3][m8][k]);
        } else for (j = 9; j >= 0; j--) chmod(&dp[n+1][i3|(j==3)][(m3+j)%3][(m8+C[w-i]*j)&7][k], dp[n][i3][m3][m8][k]);
    }
	
    k = 0;
	for (m8 = 1; m8 < 8; m8++) chmod(&k, dp[n][0][0][m8][0] + dp[n][0][0][m8][1]);
	for (m3 = 0; m3 < 3; m3++) for (m8 = 1; m8 < 8; m8++) chmod(&k, dp[n][1][m3][m8][0] + dp[n][1][m3][m8][1]);
	return k;
}

int main()
{
	int i, ans;
	
	WA = ins(A = AA), WB = ins(B);
	if (WA == 1 && *A == 1) ans = calc(WB, B);
	else {
		for (i = WA-1; i >= 0; i--) {
			if (A[i] > 0) { A[i]--; break; }
			A[i] = 9;
		}
		if (A[0] == 0) WA--, A++;
		ans = calc(WB, B) - calc(WA, A);
	}
	if (ans < 0) ans += MOD;
	printf("%d\n", ans);
	return 0;
}
0