結果

問題 No.315 世界のなんとか3.5
ユーザー bal4ubal4u
提出日時 2019-08-30 22:44:12
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 102 ms / 2,000 ms
コード長 3,037 bytes
コンパイル時間 1,345 ms
コンパイル使用メモリ 35,400 KB
実行使用メモリ 11,988 KB
最終ジャッジ日時 2024-11-22 01:33:06
合計ジャッジ時間 3,383 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
11,540 KB
testcase_01 AC 5 ms
11,528 KB
testcase_02 AC 5 ms
11,460 KB
testcase_03 AC 5 ms
11,616 KB
testcase_04 AC 4 ms
11,476 KB
testcase_05 AC 5 ms
11,448 KB
testcase_06 AC 5 ms
11,568 KB
testcase_07 AC 5 ms
11,508 KB
testcase_08 AC 5 ms
11,424 KB
testcase_09 AC 5 ms
11,464 KB
testcase_10 AC 5 ms
11,532 KB
testcase_11 AC 4 ms
11,564 KB
testcase_12 AC 28 ms
11,580 KB
testcase_13 AC 28 ms
11,688 KB
testcase_14 AC 51 ms
11,636 KB
testcase_15 AC 52 ms
11,704 KB
testcase_16 AC 52 ms
11,664 KB
testcase_17 AC 51 ms
11,704 KB
testcase_18 AC 28 ms
11,560 KB
testcase_19 AC 29 ms
11,604 KB
testcase_20 AC 29 ms
11,508 KB
testcase_21 AC 29 ms
11,596 KB
testcase_22 AC 53 ms
11,824 KB
testcase_23 AC 53 ms
11,704 KB
testcase_24 AC 51 ms
11,700 KB
testcase_25 AC 53 ms
11,620 KB
testcase_26 AC 53 ms
11,692 KB
testcase_27 AC 53 ms
11,704 KB
testcase_28 AC 50 ms
11,820 KB
testcase_29 AC 95 ms
11,896 KB
testcase_30 AC 94 ms
11,876 KB
testcase_31 AC 92 ms
11,796 KB
testcase_32 AC 99 ms
11,892 KB
testcase_33 AC 53 ms
11,660 KB
testcase_34 AC 95 ms
11,796 KB
testcase_35 AC 102 ms
11,988 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function 'in':
main.c:10:14: warning: implicit declaration of function 'getchar_unlocked' [-Wimplicit-function-declaration]
   10 | #define gc() getchar_unlocked()
      |              ^~~~~~~~~~~~~~~~
main.c:16:24: note: in expansion of macro 'gc'
   16 |         int n = 0, c = gc();
      |                        ^~

ソースコード

diff # 

// yukicoder: 315 世界のなんとか3.5
// 2019.8.30 bal4u

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

typedef long long ll;

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

int in() { // 非負整数の入力
	int n = 0, c = gc();
	do n = 10 * n + (c & 0xf); while ((c = gc()) >= '0');
	return n;
}

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[200005], *A; int WA;
char B[200005]; int WB;
int dp[200005][2][3][2];   // 3である桁, mod 3, 元の数
int dq[10][2][3][805][2];   // 3である桁, mod 3, mod 800, 元の数
int C8[4] = {0,1,2,4};
int C80[5] = {0,1,10,20,40};
int C800[6] = {0,1,10,100,200,400};

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

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

int main()
{
	int i, P, ans;
	int L, *C;
	
	WA = ins(A = AA), WB = ins(B), P = in();
	if (P == 8) C = C8, L = 3;
	else if (P == 80) C = C80, L = 4;
	else if (P == 800) C = C800, L = 5;
	
	if (WA == 1 && *A == 1) ans = calc(WB, B, C, L, P);
	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, C, L, P) - calc(WA, A, C, L, P);
	}
	if (ans < 0) ans += MOD;
	printf("%d\n", ans);
	return 0;
}
0