[数位DP 多重背包计数] BZOJ5003. 与链

8 篇文章 0 订阅
7 篇文章 0 订阅

每一位二进制分开考虑

那么在一个合法的序列中,一定是前面几个数当前二进制位是1,其他都是0

数位DP,每一位的1最多出现k次,这就是一个多重背包

多重背包转移用前缀和优化就好了

O(nlogn)

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

const int N=100010,P=1e9+9;

int n,K,f[20][N][2];
int tmp[N];

inline void add(int &x,int y){
  (x+=y)%=P;
}

int main(){
  scanf("%d%d",&K,&n);
  f[17][0][1]=1;
  for(int i=17;i;i--){
    int t=1<<i-1;
    for(int k=0;k<=1;k++){
      if(k==1 && !((n>>i-1)&1)){
    for(int j=0;j<=n;j++) add(f[i-1][j][k],f[i][j][k]);
    continue;
      }
      for(int j=0;j<=n;j++) add(f[i-1][j][0],f[i][j][k]);
      for(int j=0;j<=n;j++)
    tmp[j]=(f[i][j][k]+(j-t<0?0:tmp[j-t]))%P;
      for(int j=t;j<=n;j++){
    int nxt=(k==1 && ((n>>i-1)&1));
    add(f[i-1][j][nxt],(tmp[j-t]-(j-1LL*t*(K+1)<0?0:tmp[j-t*(K+1)]))%P);
      }
    }
  }
  int ans=(f[0][n][0]+f[0][n][1])%P;
  cout<<(ans+P)%P<<endl;
  return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值