CF1073E Segment Sum 自闭了

CF1073E Segment Sum

题意翻译

给定\(K,L,R\)，求\(L\)~\(R\)之间最多不包含超过\(K\)个数码的数的和。

\(K<=10，L,R<=1e18\)

---

我 写 晕 了

我 自 闭 了

根本不知道自己在写什么？？？？

告辞。。。

---

错误Code:

#include <cstdio>
#define ll long long
const ll mod=998244353;
ll dp[2][20][1<<10],cnt[2][20][1<<10],po[20],l,r;
int k;
void init()
{
    po[0]=1;
    for(int i=1;i<=18;i++) po[i]=po[i-1]*10%mod;
    cnt[0][0][0]=1;
    for(int i=1;i<=18;i++)
        for(int s=0;s<1<<10;s++)
        {
            for(int j=1;j<=9;j++)
                if(s>>j&1)
                {
                    int t=(1<<j)^s;
                    (cnt[1][i][s]+=(cnt[0][i-1][s]+cnt[1][i-1][s]
                                   +cnt[0][i-1][t]+cnt[1][i-1][t]))%=mod;

                }
            int t=s^1;
            if(s&1)
                (cnt[0][i][s]+=(cnt[0][i-1][s]+cnt[1][i-1][s]
                               +cnt[0][i-1][t]+cnt[1][i-1][t]))%=mod;
        }
    for(int i=1;i<=18;i++)
        for(int s=0;s<1<<10;s++)
        {
            for(int j=1;j<=9;j++)
                if(s>>j&1)
                {
                    int t=(1<<j)^s;
                    ll tmp0=(cnt[0][i-1][s]+cnt[1][i-1][s]
                            +cnt[0][i-1][t]+cnt[1][i-1][t])%mod;
                    ll tmp1=(dp[0][i-1][s]+dp[1][i-1][s]
                            +dp[0][i-1][t]+dp[1][i-1][t])%mod;
                    (dp[1][i][s]+=(tmp1+j*po[i-1]%mod*tmp0))%=mod;

                }
            if(s&1)
            {
                int t=s^1;
                ll tmp1=(dp[0][i-1][s]+dp[1][i-1][s]
                        +dp[0][i-1][t]+dp[1][i-1][t])%mod;
                dp[0][i][s]=tmp1;
            }
        }
}
int bit[20];
ll solve(ll d)
{
    int rt=0;
    for(ll i=d;i;i/=10) bit[++rt]=i%10;
    int cho=0;//高位已选状态
    ll res=0,ans=0;//高位剩余
    for(int i=rt;i;i--)
    {
        for(int s=0;s<1<<10;s++)
        {
            int ct=0;
            for(int t=s;t;t>>=1) ct+=t&1;
            if(ct>k+1) continue;
            if(s&1)
            {
                int t=s^1;
                ll tmp1=(dp[0][i-1][s]+dp[1][i-1][s]
                        +dp[0][i-1][t]+dp[1][i-1][t])%mod;
                ans+=tmp1;
            }
            if(ct>k) continue;
            for(int j=1;j<bit[i]+(i==1);j++)//高位选填
                if((s>>j&1)&&((s|cho)==s))
                {
                    int t=(1<<j)^s;
                    ll tmp0=(cnt[0][i-1][s]+cnt[1][i-1][s]
                            +cnt[0][i-1][t]+cnt[1][i-1][t])%mod;
                    ll tmp1=(dp[0][i-1][s]+dp[1][i-1][s]
                            +dp[0][i-1][t]+dp[1][i-1][t])%mod;
                    (ans+=tmp1+tmp0*(j+res)%mod*po[i-1]%mod)%=mod;
                }
        }
        res=(res+bit[i])*10%mod;
        cho|=1<<bit[i];
    }
    return ans;
}
int main()
{
    scanf("%lld%lld%d",&l,&r,&k);
    init();
    printf("%lld\n",((solve(r)-solve(l-1))%mod+mod)%mod);
    return 0;
}

updata2019.2.9

写了好几个月，终于肝出来了

Code:

#include <cstdio>
#include <cstring>
#define ll long long
const int mod=998244353;
inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
#define mul(a,b) (1ll*(a)*(b)%mod)
int po[20],bit[20],len,k;
struct node
{
    int val,cnt;
    node(){}
    node(int v,int c){val=v,cnt=c;}
    node friend operator +(node a,node b){return node(add(a.val,b.val),add(a.cnt,b.cnt));}
}dp[20][1<<10];
node dfs(int pos,int sta,int lead,int lim)//前导0和最高位限制
{
	int cnt=0;
	for(int i=0;i<10;i++) cnt+=sta>>i&1;
	if(cnt>pos) return node(0,0);
	if(!pos) return node(0,1);
	if(!lim&&!lead&&~dp[pos][sta].val) return dp[pos][sta];
	node ret=node(0,0),bee;
	if(lead) ret=ret+dfs(pos-1,sta,lead,lim&&!bit[pos]);
	else if(sta&1) ret=ret+dfs(pos-1,sta,lead,lim&&!bit[pos])+dfs(pos-1,sta^1,lead,lim&&!bit[pos]);
	for(int i=1,up=lim?bit[pos]:9;i<=up;i++)
		if(sta>>i&1)
		{
		    bee=dfs(pos-1,sta,0,lim&&i==up)+dfs(pos-1,sta^(1<<i),0,lim&&i==up);
		    ret=ret+bee;
            ret.val=add(ret.val,mul(bee.cnt,mul(i,po[pos-1])));
		}
	return !lim&&!lead?dp[pos][sta]=ret:ret;
}
int cal(ll x)
{
	len=0;while(x) bit[++len]=x%10,x/=10;
	memset(dp,-1,sizeof dp);int ans=0;
	for(int s=0;s<1<<10;s++)
	{
		int cnt=0;
		for(int i=0;i<10;i++) cnt+=s>>i&1;
		if(cnt<=k) ans=add(ans,dfs(len,s,1,1).val);
	}
	return ans;
}
int main()
{
	ll l,r;
	scanf("%lld%lld%d",&l,&r,&k);
	po[0]=1;for(int i=1;i<=18;i++) po[i]=mul(po[i-1],10);
	printf("%d\n",add(cal(r),mod-cal(l-1)));
	return 0;
}