【newcoder 】数圈圈 【数位DP 模板】

来源:互联网 发布:尊而光律所怎么样知乎 编辑:程序博客网 时间:2024/06/04 18:55

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld
题目描述
tabris有一个习惯,无聊的时候就会数圈圈,无论数字还是字母。
现在tabris更无聊啦,晚上睡不着觉就开始数羊,从a只数到b只。
顺便还数了a到b之间有多少个圈。

但是tabris笨啊,虽然数羊不会数错,但很可能数错圈的个数。
但是tabris很难接受自己笨这个事实,所以想问问你他一共应该数出多少个圈,这样tabris才好判断他到底笨不笨啊。
输入描述:
输入一个T,表示数据组数
每组测试数据包含两个正整数a,b。
T∈[1,1000]
a,b∈[1,1014]
输出描述:
每组数据输出结果,并换行。
示例1
输入

11
1 1
2 2
3 3
4 4
5 5
6 6
7 7
8 8
9 9
10 10
1 100
输出

0
0
0
1
0
1
0
2
1
1
111
备注:
数字的圈的个数请根据样例自行理解。

分析:模板。
原文链接
代码

#include <bits/stdc++.h>  using namespace std;  typedef long long LL;  int get(int x){   // 每种数字的贡献是多少。    if(x==0||x==4||x==6||x==9) return 1;      else if(x==1||x==2||x==3||x==5||x==7) return 0;      else return 2;  }  LL dp[20][20][20]; //dp[pos][num][sum]前pos位,num已经出现了sum次  LL digit[20];  LL dfs(int pos, int num, int sum, bool lead, bool limit){      if(pos==0) return sum;      if(!limit&&!lead&&dp[pos][num][sum]!=-1) return dp[pos][num][sum];      int up=limit?digit[pos]:9;      LL ans=0;      if(!lead||pos==1) ans+=dfs(pos-1, num, sum+(num==0), 0, limit&&digit[pos]==0);      else ans+=dfs(pos-1, num, sum, 1, limit&&digit[pos]==0);      for(int i=1; i<=up; i++){          ans+=dfs(pos-1, num, sum+(num==i), 0, limit&&digit[pos]==i);      }      if(!limit&&!lead) dp[pos][num][sum]=ans;      return ans;  }  LL f(LL x, int num){      if(x<0) return 0;      if(x==0) return num==0?1:0;      int pos=0;      while(x){          digit[++pos]=x%10;          x/=10;      }      return dfs(pos,num,0,1,1);  }  int main(){      int T;      LL a, b;      scanf("%d", &T);      memset(dp, -1, sizeof(dp));      while(T--){          LL ans = 0;          scanf("%lld%lld", &a,&b);          for(int i=0;i<=9;i++) ans+=(f(b,i)-f(a-1,i))*get(i);          printf("%lld\n", ans);      }  }