【数位DP】【HDU 4507】吉哥系列故事——恨7不成妻

来源：互联网发布：张无忌最爱赵敏知乎编辑：程序博客网时间：2024/04/30 21:04

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

无聊写个水DP

#include <cstdio>#include <cstring>#include <cstdlib>#include <iostream>#include <cmath>#include <algorithm>using namespace std;#define rep(i,l,r) for(int i=(l),_=(r);i<=_;i++)#define per(i,r,l) for(int i=(r),_=(l);i>=_;i--)#define MS(arr,x) memset(arr,x,sizeof(arr))#define INE(i,u) for(int i=head[u];~i;i=e[i].next)#define LL long long#define X first#define Y secondinline const int read(){int r=0,k=1;char c=getchar();for(;c<'0'||c>'9';c=getchar())if(c=='-')k=-1;for(;c>='0'&&c<='9';c=getchar())r=r*10+c-'0';return k*r;}////////////////////////////////////////////////const LL mod=1000000007;LL l,r;struct data{LL cnt,s1,s2;};data dp[20][7][7];int bit[20];LL pow10[20];////////////////////////////////////////////////data dfs(int pos,int bitsum,int sum,bool limit){if(!pos) return bitsum!=0&&sum!=0?(data){1,0,0}:(data){0,0,0};if(!limit && ~dp[pos][bitsum][sum].cnt) return dp[pos][bitsum][sum];int upp=limit?bit[pos]:9;data ans=(data){0,0,0};rep(i,0,upp) if(i!=7){data ret=dfs(pos-1,(bitsum+i)%7,(sum*10+i)%7,limit&&i==upp);ans.cnt=ans.cnt+ret.cnt;if(ans.cnt>=mod) ans.cnt-=mod;LL x=i*pow10[pos]%mod;ans.s1=ans.s1+ret.s1+ret.cnt*x%mod;if(ans.s1>=mod) ans.s1-=mod;if(ans.s1>=mod) ans.s1-=mod;ans.s2+=ret.s2;if(ans.s2>=mod) ans.s2-=mod;ans.s2+=2*ret.s1%mod*x%mod;if(ans.s2>=mod) ans.s2-=mod;ans.s2+=ret.cnt*x%mod*x%mod;if(ans.s2>=mod) ans.s2-=mod;}if(!limit) dp[pos][bitsum][sum]=ans;return ans;}LL cal(LL n){for(*bit=0;n;n/=10) bit[++*bit]=n%10;return dfs(*bit,0,0,1).s2;}////////////////////////////////////////////////void input(){    cin>>l>>r;}void solve(){LL ans=cal(r)-cal(l-1);if(ans<0) ans+=mod;cout<<ans<<endl; }////////////////////////////////////////////////int main(){    freopen("_.in","r",stdin); freopen("_.out","w",stdout);    MS(dp,-1);    pow10[1]=1; rep(i,2,20) pow10[i]=pow10[i-1]*10%mod;    rep(i,1,read())    input(),solve();    return 0;}

0 0