数位dp初探

2022-09-11 14:24:45

我这种蒟蒻就一直不会写数位dp。。

于是开了个坑。。

1833: [ZJOI2010]count 数字计数

这道被KPM大爷说是入门题。。嗯似乎找找规律然后减掉0的情况后乱搞就可以了。。（但是还是写了很久TAT

#include<cstring>

#include<iostream>

#include<cstdio>

#include<queue>

#include<cmath>

#include<algorithm>

#define rep(i,l,r) for (int i=l;i<=r;i++)

#define down(i,l,r) for (int i=l;i>=r;i--)

#define clr(x,y) memset(x,y,sizeof(x))

#define low(x) (x&(-x))

#define maxn 505

#define inf int(1e9)

#define mm 1000000007

#define ll long long

using namespace std;

ll ans[][];

ll a,b,bin[];

ll read(){

    ll x=,f=; char ch=getchar();

    while (!isdigit(ch)) {if (ch=='-') f=-; ch=getchar();}

    while (isdigit(ch)) {x=x*+ch-''; ch=getchar();}

    return x*f;

}

int get(ll x){

    if (x==) return ;

    int ans=;

    while(x){ans++; x/=;}

    return ans;

}

void cal(ll x,int f){

    int L,l=get(x),a; L=l;

    rep(i,,) ans[i][f]=(l-)*bin[l-];

    rep(i,,l-) ans[][f]-=bin[i];

    while (l){

        a=x/bin[l-];

        if (l==L) rep(i,,a-) {

            ans[i][f]+=bin[l-];

            if (l->=) rep(j,,) ans[j][f]+=1LL*(l-)*bin[l-];

        }

        if (l!=L) rep(i,,a-) {

            ans[i][f]+=bin[l-];

            if (l->=) rep(j,,) ans[j][f]+=1LL*(l-)*bin[l-];

        }

        ans[a][f]+=x%bin[l-]+;

        x=x%bin[l-];

        l--;

    }

}

int main(){

    bin[]=; rep(i,,) bin[i]=bin[i-]*;

    a=read(); b=read();

    cal(a-,);

    cal(b,);

    rep(i,,) printf("%lld ",ans[i][]-ans[i][]);

    printf("%lld\n",ans[][]-ans[][]);

    return ;

}

1026: [SCOI2009]windy数

这道写个dp f[i][j]表示第i位最前一位数是j，特判一下0的情况。。

#include<cstring>

#include<iostream>

#include<cstdio>

#include<queue>

#include<cmath>

#include<algorithm>

#define rep(i,l,r) for (int i=l;i<=r;i++)

#define down(i,l,r) for (int i=l;i>=r;i--)

#define clr(x,y) memset(x,y,sizeof(x))

#define low(x) (x&(-x))

#define maxn 505

#define inf int(1e9)

#define mm 1000000007

#define ll long long

using namespace std;

int f[][],bin[];

int a,b;

int read(){

    int x=,f=; char ch=getchar();

    while (!isdigit(ch)) {if (ch=='-') f=-; ch=getchar();}

    while (isdigit(ch)) {x=x*+ch-''; ch=getchar();}

    return x*f;

}

int get(ll x){

    if (x==) return ;

    int ans=;

    while(x){ans++; x/=;}

    return ans;

}

int cal(int x){

    int L,l=get(x),ans=,a,tmp=-; L=l;

    if (l>) ans++;//0的情况

    rep(i,,l-) {

        rep(j,,) ans+=f[i][j];

    }

    while (l){

        a=x/bin[l-];

        if (l==L) rep(i,,a-) if (abs(i-tmp)>=){

            rep(j,,) if (abs(i-j)>=) ans+=f[l-][j];

        }

        if (l!=L) rep(i,,a-) if (abs(i-tmp)>=){

            rep(j,,) if (abs(i-j)>=) ans+=f[l-][j];

        }

        if (l==) rep(i,,a) if (abs(tmp-i)>=) ans++;

        if (abs(tmp-a)<) break;

        tmp=a;

        x%=bin[l-];

        l--;

    }

    return ans;

}

int main(){

    bin[]=; rep(i,,) bin[i]=bin[i-]*;

    a=read(); b=read();

    int l=get(b);

    rep(i,,) f[][i]=;

    rep(i,,) rep(j,,) if (f[i][j]){

        rep(k,,) if (abs(j-k)>=) f[i+][k]+=f[i][j];

    }

    printf("%d\n",cal(b)-cal(a-));

    return ;

}

1072: [SCOI2007]排列perm

为什么我一直认为这道是状压。。 f[s][i]表示当前状态是s，模数为i。然后枚举每一个数是否被选过转移就可以了。

最后除掉相同数字个数的阶乘。

#include<cstring>

#include<iostream>

#include<cstdio>

#include<queue>

#include<cmath>

#include<algorithm>

#define rep(i,l,r) for (int i=l;i<=r;i++)

#define down(i,l,r) for (int i=l;i>=r;i--)

#define clr(x,y) memset(x,y,sizeof(x))

#define low(x) (x&(-x))

#define maxn 505

#define inf int(1e9)

#define mm 1000000007

#define ll long long

using namespace std;

int f[][];

int n,t,l,bin[],a[],num[],p[],ans;

char s[];

int read(){

    int x=,f=; char ch=getchar();

    while (!isdigit(ch)) {if (ch=='-') f=-; ch=getchar();}

    while (isdigit(ch)) {x=x*+ch-''; ch=getchar();}

    return x*f;

}

int main(){

    t=read();

    bin[]=; rep(i,,) bin[i]=bin[i-]*;

    p[]=; rep(i,,) p[i]=p[i-]*i;

    while (t--){

        clr(f,); clr(num,);

        scanf("%s",s); l=strlen(s);

        int d=read();

        rep(i,,l-) a[i]=s[i]-'',num[a[i]]++;

        //printf("%d\n",bin[l]-1);

        f[][]=;

        rep(s,,bin[l]-) rep(j,,d-) if (f[s][j]){

            rep(k,,l-) if ((s&bin[k])==) f[s+bin[k]][(j*+a[k])%d]+=f[s][j];

        }

        ans=f[bin[l]-][];

        rep(i,,) if (num[i]) ans/=p[num[i]];

        printf("%d\n",ans);

    }

    return ;

}

2425: [HAOI2010]计数

题目其实就是给你一些数字让你用这些数字构造一个个比给定数小的数。。

比如说我现在这一位最高为x，那就取0~x-1(可以有前导0)为这一位然后剩下的数*组合。

对于这个*组合，比如剩下有n位，这个数字还剩下有x个，那每一次就乘上C(n,x)然后n-=x。

#include<cstring>

#include<iostream>

#include<cstdio>

#include<queue>

#include<cmath>

#include<algorithm>

#define rep(i,l,r) for (int i=l;i<=r;i++)

#define down(i,l,r) for (int i=l;i>=r;i--)

#define clr(x,y) memset(x,y,sizeof(x))

#define low(x) (x&(-x))

#define maxn 505

#define inf int(1e9)

#define mm 1000000007

#define ll long long

using namespace std;

ll c[][],ans;

int cnt[],a[];

char s[];

int read(){

    int x=,f=; char ch=getchar();

    while (!isdigit(ch)) {if (ch=='-') f=-; ch=getchar();}

    while (isdigit(ch)) {x=x*+ch-''; ch=getchar();}

    return x*f;

}

ll cal(int n){

    ll ans=;

    int now=n;

    rep(i,,) ans*=c[now][cnt[i]],now-=cnt[i];

    return ans;

}

int main(){

    c[][]=;

    rep(i,,) {

        c[i][]=;

        rep(j,,) c[i][j]=c[i-][j-]+c[i-][j];

    }

    scanf("%s",s); int l=strlen(s);

    rep(i,,l-) a[i+]=s[i]-'',cnt[a[i+]]++;

    rep(i,,l) {

        rep(j,,a[i]-) if (cnt[j]){

            cnt[j]--;

            ans+=cal(l-i);

            cnt[j]++;

        }

        cnt[a[i]]--;

    }

    printf("%lld\n",ans);

    return ;

}

BZOJ1799: [Ahoi2009]self 同类分布

http://www.cnblogs.com/ctlchild/p/5126952.html

（还有几道题目慢慢填TAT。。。

码农公寓

相关文章