CF 258B Little Elephant and Elections

数位dp。 先找出含不同数目幸运数字的数量。然后直接暴力求解。

本题是第一次做这种找不同数目特定数字的数位dp，加一个参量。在函数外面加一重循环，指定数目，最后判断是否与预期数目相同即可。

暴力求种类数每一个政党可能选的所有数目是相加的，每一种数目的下的种类与后面政党的所有可能数相乘。一开始全部相乘搞错了。波神是dp过的，10^5,优化了两个数量级。感觉没有太大必要

提交时db了两次，第一是交错题，然后INF是1000000007，我令为1000000009了。。。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
#define INF 1000000007
#define LL long long
char a[20];
LL sum[20];
LL dp[20][20];
int len;
LL dfs(int id, int s, bool d, int aim){
    if(s > aim) return 0;
    if(id >= len) return s == aim;
    if(dp[id][s] != -1 && !d)return dp[id][s];
    int edge = d ? a[id] - '0' : 9;
    LL ans = 0;
    for(int i = 0; i <= edge; i++){
        if(i == 4 || i == 7) ans += dfs(id + 1, s + 1, d && i == edge, aim);
        else ans += dfs(id + 1, s , d && i == edge, aim);
    }
    return d ? ans : dp[id][s] = ans;
}
LL dfs2(int id, int num, int sn){
    if(id >= 6)return num < sn ? 1 : 0;
    LL ans = 0;
    for(int i = 0; i < sn - num; i++){
        if(sum[i] > 0){
            sum[i] --;
            ans += (sum[i] + 1) * dfs2(id + 1, num + i, sn) % INF;
            ans %= INF;
            sum[i] ++;
        }

    }
    return ans;
}
int main(){
    scanf("%s", a);
    len = strlen(a);
    memset(sum, 0, sizeof(sum));
    for(int i = 0; i <= 9; i++){
        memset(dp, -1, sizeof(dp));
        sum[i] = dfs(0, 0, true, i);
        if(i == 0)sum[0]--;
    }//printf("\n");

    LL ans = 0;
    for(int i = 0; i <= 9; i++){
        if(sum[i] > 0){
            ans += sum[i] * dfs2(0, 0, i) % INF;
            ans %= INF;
        }
    }
    printf("%I64d\n", ans);
    return 0;
}