POJ 2282 數位DP

鏈接：

http://poj.org/problem?id=2282

題意：

給你一個區間a，b，問a到b之間每個數字出現了多少次

題解：

看過算法設計與分析的人都很熟悉這道題，畢竟是課后練習的第一道，感覺用數位dp比模擬更好理解啊

dp[pos][sta]表示到從最低位到第pos位，第pos位為sta（0<=sta<=9）時的 num的個數。

代碼：

int a[20];
int dp[20][10];
int num;

int dfs(int pos, int sta, bool lead, bool limit) {
    if (pos == -1) return sta;
    if (!lead && !limit && dp[pos][sta] != -1) return dp[pos][sta];
    int up = limit ? a[pos] : 9;
    int res = 0;
    rep(i, 0, up + 1) {
        if (lead && i == 0) res += dfs(pos - 1, sta, true, limit && i == a[pos]);
        else res += dfs(pos - 1, sta + (i == num), false, limit && i == a[pos]);
    }
    if (!lead && !limit) dp[pos][sta] = res;
    return res;
}

int solve(int x) {
    int pos = 0;
    while (x) {
        a[pos++] = x % 10;
        x /= 10;
    }
    return dfs(pos - 1, 0, true, true);
}

int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    int a, b;
    memset(dp, -1, sizeof(dp));
    while (cin >> a >> b, a) {
        if (a > b) swap(a, b);
        num = 0;
        cout << solve(b) - solve(a - 1);
        for (num = 1; num <= 9; num++)
            cout << ' ' << solve(b) - solve(a - 1);
        cout << endl;
    }
    return 0;
}