yukicoder No.260 世界のなんとか3

問題

http://yukicoder.me/problems/no/260

日.

解法

桁DPです.
を, (番目までの桁を見ていて, との大小関係が, との大小関係が, 3で割った余りが, 8で割った余りが, 数字の中に3が含まれているかがで, そのような状態になるような数の個数)と定義しました.
遷移自体はたいして難しくないですね.

ソースコード

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;

const int kMOD = 1000 * 1000 * 1000 + 7;
const int kMAX_INDEX = 10010;
string A, B;

int ways[kMAX_INDEX][2][2][3][8][2];

void Solve() {
    vector<int> int_A, int_B;
    // AとBを数列に変換する
    for (int i = 0; i < B.size(); i++) {
        if (i + A.size() < B.size()) int_A.push_back(0);
        else int_A.push_back(A[i - B.size() + A.size()] - '0');
        int_B.push_back(B[i] - '0');
    }

    // 桁dp
    ways[0][0][0][0][0][0] = 1;
    for (int i = 0; i < int_A.size(); i++) {
        for (int greater = 0; greater < 2; greater++) {
            for (int smaller = 0; smaller < 2; smaller++) {
                for (int mod3 = 0; mod3 < 3; mod3++) {
                    for (int mod8 = 0; mod8 < 8; mod8++) {
                        for (int has_three = 0; has_three < 2; has_three++) {
                            if (ways[i][greater][smaller][mod3][mod8][has_three] == 0) continue;

                            for (int digit = 0; digit < 10; digit++) {
                                if (!greater && digit < int_A[i]) continue;
                                if (!smaller && digit > int_B[i]) break;

                                ways[i + 1][greater | digit > int_A[i]][smaller | digit < int_B[i]][(mod3 * 10 + digit) % 3][(mod8 * 10 + digit) % 8][has_three | digit == 3] += ways[i][greater][smaller][mod3][mod8][has_three];
                                ways[i + 1][greater | digit > int_A[i]][smaller | digit < int_B[i]][(mod3 * 10 + digit) % 3][(mod8 * 10 + digit) % 8][has_three | digit == 3] %= kMOD;
                            }
                        }
                    }
                }
            }
        }
    }
    int answer = 0;
    for (int greater = 0; greater < 2; greater++) {
        for (int smaller = 0; smaller < 2; smaller++) {
            for (int mod3 = 0; mod3 < 3; mod3++) {
                for (int mod8 = 1; mod8 < 8; mod8++) {
                    for (int has_three = 0; has_three < 2; has_three++) {
                        if (mod3 != 0 && !has_three) continue;
                        answer = (answer + ways[int_A.size()][greater][smaller][mod3][mod8][has_three]) % kMOD;
                    }
                }
            }
        }
    }
    cout << answer << endl;
}

int main() {
    cin >> A >> B;

    Solve();
    return 0;
}

感想

(横に)長すぎィ!