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;
}