結果
問題 | No.315 世界のなんとか3.5 |
ユーザー |
![]() |
提出日時 | 2019-08-30 22:44:12 |
言語 | C (gcc 13.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 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 36 |
コンパイルメッセージ
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(); | ^~
ソースコード
// 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; }