hihocoder 1033 인터레이싱 및 디지털 DP

pp[site][sum][p]는site 비트 숫자가 있고 교차와sum가 있으며 site 위치의 기호는 +(p==1) 또는 -(p==0)이다.마지막으로 k라는 것을 알았기 때문에sum의 초기값을 k+100(마이너스가 있기 때문)으로 설정합니다.어떤 분에 대해서 만약에 추가하려면 우리는sum로 줄여서 마지막까지 진행합니다. 만약sum==100이라면 앞의 교차와 바로 k를 확신할 수 있습니다.서로 다른 자릿수의 숫자에 대해 본위 숫자에 대해 가감 두 가지 상황이 있기 때문에 p라는 일차원으로 헷갈리는 것을 방지했다.
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define N 1000000007
#define LL long long
struct node
{
    LL cnt;
    LL sum;
}dp[20][202][2];
int num[20];
LL c[20],k;
node dfs(int site,int sum,int f,int p,int flag)
{
    if(site==0)
    {
        node a;
        if(flag) a.cnt=0;
        else
        a.cnt=sum==100?1:0;
        a.sum=0;
        return a;
    }
    if(!f&&!flag&&dp[site][sum][p].cnt!=-1)
        return dp[site][sum][p];
    node ans,a;
    ans.cnt=ans.sum=0;
    int len=f?num[site]:9;
    for(int i=0;i<=len;i++)
    {
        if(flag&&i==0)
        {
            a=dfs(site-1,sum,f&&i==len,p,flag);
        }
        else
        {
        if(p)
            a=dfs(site-1,sum-i,f&&i==len,p^1,0);
        else
            a=dfs(site-1,sum+i,f&&i==len,p^1,0);
        }
        ans.cnt=(ans.cnt+a.cnt)%N;
        ans.sum=(ans.sum+((c[site]*a.cnt)%N)*i+a.sum)%N;
    }
    if(!f&&!flag)
    {
        dp[site][sum][p].cnt=ans.cnt;
        dp[site][sum][p].sum=ans.sum;
    }
    return ans;
}
LL solve(LL n)
{
    int i=0;
    while(n)
    {
        num[++i]=n%10;
        n=n/10;
    }
    node ans;
    ans=dfs(i,k+100,1,1,1);
    return ans.sum;
}
int main()
{
    LL l,r;
    memset(dp,-1,sizeof(dp));
    c[1]=1;
    for(int i=2;i<=18;i++)
        c[i]=(c[i-1]*10)%N;
    while(scanf("%lld %lld %d",&l,&r,&k)!=EOF)
    {
        printf("%lld
",((solve(r)-solve(l-1))%N+N)%N); } return 0; }

좋은 웹페이지 즐겨찾기