hdu 4507 吉哥系列故事——恨7不成妻

http://acm.hdu.edu.cn/showproblem.php?pid=4507

用dp结构体里面存有符合要求的个数cnt，各位数和的sum1，符合要求的数的平方和sum2三个值。在维护sum2时需要用到sum1和cnt两个值。

数位dp

 1 #include <cstdio>

 2 #include <cstring>

 3 #include <algorithm>

 4 #define LL __int64

 5 using namespace std;

 6 const LL mod=1000000007;

 7 

 8 struct node

 9 {

10     LL cnt;

11     LL sum1;

12     LL sum2;

13 } dp[30][30][30];

14 int num[30];

15 LL c[30];

16 

17 node dfs(int pos,int sum1,int sum2,bool flag)

18 {

19     if(pos==-1)

20     {

21         node st;

22         st.cnt=(sum1!=0&&sum2!=0);

23         st.sum1=0;

24         st.sum2=0;

25         return st;

26     }

27     if(!flag&&dp[pos][sum1][sum2].cnt!=-1) return dp[pos][sum1][sum2];

28     node ans;

29     ans.cnt=0;

30     ans.sum1=0;

31     ans.sum2=0;

32     node st1;

33     int xx=flag?num[pos]:9;

34     for(int i=0; i<=xx; i++)

35     {

36         if(i==7) continue;

37         st1=dfs(pos-1,(sum1+i)%7,(sum2*10+i)%7,flag&&(i==xx));

38         ans.cnt+=st1.cnt;

39         ans.cnt%=mod;

40         ans.sum1+=(st1.sum1+((c[pos]*i%mod)*st1.cnt)%mod)%mod;

41         ans.sum1%=mod;

42         ans.sum2+=(((st1.cnt*c[pos])%mod*c[pos]%mod*i*i)%mod);

43         ans.sum2%=mod;

44         ans.sum2+=(st1.sum2+((2*c[pos]*i)%mod)*st1.sum1)%mod;

45         ans.sum2%=mod;

46     }

47     if(!flag) dp[pos][sum1][sum2]=ans;

48     return ans;

49 }

50 

51 LL get(LL n)

52 {

53     int cnt=0;

54     while(n)

55     {

56         num[cnt++]=n%10;

57         n=n/10;

58     }

59     node a=dfs(cnt-1,0,0,true);

60     return a.sum2;

61 }

62 

63 void inti()

64 {

65     c[0]=1;

66     for(int i=1; i<20; i++)

67     {

68         c[i]=(c[i-1]*10)%mod;

69     }

70     for(int i=0; i<20; i++)

71     {

72         for(int j=0; j<20; j++)

73         {

74             for(int k=0; k<20; k++)

75             {

76               dp[i][j][k].cnt=-1;

77             }

78         }

79     }

80 }

81 int main()

82 {

83     int t;

84     scanf("%d",&t);

85     inti();

86     while(t--)

87     {

88         LL n,m;

89         scanf("%I64d%I64d",&n,&m);

90         printf("%I64d\n",((get(m)-get(n-1))%mod+mod)%mod);

91     }

92     return 0;

93 }

View Code