ICPC20上海C——小白也能秒懂的数位dp讲解

最新推荐文章于 2022-10-28 23:29:15 发布
最新推荐文章于 2022-10-28 23:29:15 发布
阅读量205 收藏 2
点赞数 2
分类专栏： 数位dp 文章标签： 算法 c++
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/hans774882968/article/details/118651643
版权

dp定义有很多种。采用这个最直观的：添加链接描述

i&j=0，意味着相加不会进位。log2(i+j)+1向下取整，意味着它等于i、j中的最长len值。

接下来为了解决原问题，我们定义一个更为简单的子问题为：
∑ i = 0 X ∑ j = [ i = 0 ] Y [ i & j = 0 ] \sum_{i=0}^X \sum_{j=[i=0]}^Y [i\&j=0] i=0∑X​j=[i=0]∑Y​[i&j=0]
为什么做此定义？以up1=up2=1（注：对应已知的数位dp模板的up变量，比如hdu2089，见下文“附”）为例：枚举i和j的最高位（使得最高位固定），当两者最高位均为0，则取原问题。否则
[ l o g 2 ( i + j ) + 1 ] [log_2(i+j)+1] [log2​(i+j)+1]
已经确定（“[]”表示向下取整），在我的代码里就是idx+1，于是可以把它提到求和式外面，这样就构造出上述更为简单的子问题。

所以我们定义dp[idx,zero]为：当前考虑到下标为idx的位，求的是原问题/子问题（zero=1/0）。之所以变量名叫zero，只是因为一个巧合：该变量恰好和表示i和j的更高位是否全都是0的变量是同一个。

因为原问题的转移方程里有可以提取的公因子，所以我们定义一个权值weight，它等于1或idx+1。下面看weight什么时候等于idx+1。由定义可以知道，如果zero=0，则无论如何权值都只乘1；如果zero=1，表示求解原问题，则i和j不都等于0时，weight=idx+1。

伪代码

    LL ans = 0;
    int up1 = f1 ? x[idx] : 1,up2 = f2 ? y[idx] : 1;
    rep(i,0,up1){
        rep(j,0,up2){
            if(i & j) continue;
            int weight = zero && (i || j) ? idx+1 : 1;
            (ans += dfs(idx-1,f1 && i == up1,f2 && j == up2,NXTZERO) * weight % mod) %= mod;
        }
    }

接下来确定NXTZERO。由以上讨论已经知道，zero=1（求解原问题）且i和j都为0时，NXTZERO=1。其他情况NXTZERO=0。

NXTZERO = zero && !(i || j)
  • 因为dp定义与test case无关，所以memset只需要在最开始做1次，这一技巧在hdu2089中也用到。
  • 需要减1，因为数位dp的代码把i == 0 且 j == 0的情况也算进去了。

参考已有数位dp模板，有如下代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
#define rep(i,a,b) for(int i = (a);i <= (b);++i)
#define re_(i,a,b) for(int i = (a);i < (b);++i)
#define dwn(i,a,b) for(int i = (a);i >= (b);--i)

const int N = 40;
const int mod = 1e9 + 7;

int x[N],y[N];
LL dp[N][2];

void dbg(){puts("");}
template<typename T, typename... R>void dbg(const T &f, const R &... r) {
    cout << f << " ";
    dbg(r...);
}
template<typename Type>inline void read(Type &xx){
    Type f = 1;char ch;xx = 0;
    for(ch = getchar();ch < '0' || ch > '9';ch = getchar()) if(ch == '-') f = -1;
    for(;ch >= '0' && ch <= '9';ch = getchar()) xx = xx * 10 + ch - '0';
    xx *= f;
}

LL dfs(int idx,int f1,int f2,bool zero){
    if(idx == -1) return 1;
    LL &d = dp[idx][zero];
    if(~d && !f1 && !f2) return d;
    LL ans = 0;
    int up1 = f1 ? x[idx] : 1,up2 = f2 ? y[idx] : 1;
    rep(i,0,up1){
        rep(j,0,up2){
            if(i & j) continue;
            int weight = zero && (i || j) ? idx+1 : 1;
            (ans += dfs(idx-1,f1 && i == up1,f2 && j == up2,zero && !(i || j)) * weight % mod) %= mod;
        }
    }
    if(!f1 && !f2) d = ans;
    return ans;
}

LL solve(int u,int v){
    int l1 = 0,l2 = 0;
    memset(x,0,sizeof x);memset(y,0,sizeof y);
    for(;u;u >>= 1) x[l1++] = u & 1;
    for(;v;v >>= 1) y[l2++] = v & 1;
    return (dfs(max(l1,l2)-1,1,1,true) + mod-1) % mod;
}

int main(int argc, char** argv) {
    memset(dp,-1,sizeof dp);
    int T;read(T);
    while(T--){
        int u,v;read(u);read(v);
        printf("%lld\n",solve(u,v));
    }
    return 0;
}

可惜它TLE了。这就引出一个优化：定义dp[idx,f1,f2,zero]，其中f1和f2是以上代码的f1和f2。它们不实际参与决策，但放在状态里。这是一个空间换时间的技巧。那么

if(~d && !f1 && !f2) return d;

if(!f1 && !f2) d = ans;

就变成了

if(~d) return d;

return d = ans;

相应地，必须在每个case都memset一次。
这就得到我们在参考链接中看到的4维dp，本文指出它本质上是2维的dp+空间换时间技巧。这样就得到一个性能更优（可以AC）的代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
#define rep(i,a,b) for(int i = (a);i <= (b);++i)
#define re_(i,a,b) for(int i = (a);i < (b);++i)
#define dwn(i,a,b) for(int i = (a);i >= (b);--i)

const int N = 40;
const int mod = 1e9 + 7;

int x[N],y[N];
LL dp[N][2][2][2];

void dbg(){puts("");}
template<typename T, typename... R>void dbg(const T &f, const R &... r) {
    cout << f << " ";
    dbg(r...);
}
template<typename Type>inline void read(Type &xx){
    Type f = 1;char ch;xx = 0;
    for(ch = getchar();ch < '0' || ch > '9';ch = getchar()) if(ch == '-') f = -1;
    for(;ch >= '0' && ch <= '9';ch = getchar()) xx = xx * 10 + ch - '0';
    xx *= f;
}

LL dfs(int idx,int f1,int f2,bool zero){
    if(idx == -1) return 1;
    LL &d = dp[idx][f1][f2][zero];
    if(~d) return d;
    LL ans = 0;
    int up1 = f1 ? x[idx] : 1,up2 = f2 ? y[idx] : 1;
    rep(i,0,up1){
        rep(j,0,up2){
            if(i & j) continue;
            int weight = zero && (i || j) ? idx+1 : 1;
            (ans += dfs(idx-1,f1 && i == up1,f2 && j == up2,zero && !(i || j)) * weight % mod) %= mod;
        }
    }
    return d = ans;
}

LL solve(int u,int v){
    int l1 = 0,l2 = 0;
    memset(x,0,sizeof x);memset(y,0,sizeof y);memset(dp,-1,sizeof dp);
    for(;u;u >>= 1) x[l1++] = u & 1;
    for(;v;v >>= 1) y[l2++] = v & 1;
    return (dfs(max(l1,l2)-1,1,1,true) + mod-1) % mod;
}

int main(int argc, char** argv) {
    int T;read(T);
    while(T--){
        int u,v;read(u);read(v);
        printf("%lld\n",solve(u,v));
    }
    return 0;
}

附：hdu2089的AC代码

2维且memset只做1次的版本

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
#define rep(i,a,b) for(int i = (a);i <= (b);++i)
#define re_(i,a,b) for(int i = (a);i < (b);++i)
#define dwn(i,a,b) for(int i = (a);i >= (b);--i)

const int SZ = 20;

int dl,d[SZ],dp[SZ][2];

template<typename Type>inline void read(Type &xx){
    Type f = 1;char ch;xx = 0;
    for(ch = getchar();ch < '0' || ch > '9';ch = getchar()) if(ch == '-') f = -1;
    for(;ch >= '0' && ch <= '9';ch = getchar()) xx = xx * 10 + ch - '0';
    xx *= f;
}

int dfs(int idx,bool is6,bool isup){
    if(idx == -1) return 1;
    if(!isup && ~dp[idx][is6]) return dp[idx][is6];
    int up = isup ? d[idx] : 9;
    int ans = 0;
    rep(i,0,up){
        if(i == 4) continue;
        if(is6 && i == 2) continue;
        ans += dfs(idx - 1,i == 6,isup && i == up);
    }
    if(!isup) dp[idx][is6] = ans;
    return ans;
}

int solve(int x){
    dl = 0;
    for(;x;x /= 10) d[dl++] = x % 10;
    return dfs(dl - 1,false,true);
}

int main(int argc, char** argv) {
    memset(dp,-1,sizeof dp);
    int a,b;
    while(~scanf("%d%d",&a,&b) && (a || b)){
        printf("%d\n",solve(b) - solve(a - 1));
    }
    return 0;
}

3维且memset做多次的版本

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
#define rep(i,a,b) for(int i = (a);i <= (b);++i)
#define re_(i,a,b) for(int i = (a);i < (b);++i)
#define dwn(i,a,b) for(int i = (a);i >= (b);--i)

const int SZ = 20;

int dl,d[SZ],dp[SZ][2][2];

template<typename Type>inline void read(Type &xx){
    Type f = 1;char ch;xx = 0;
    for(ch = getchar();ch < '0' || ch > '9';ch = getchar()) if(ch == '-') f = -1;
    for(;ch >= '0' && ch <= '9';ch = getchar()) xx = xx * 10 + ch - '0';
    xx *= f;
}

int dfs(int idx,bool is6,bool isup){
    if(idx == -1) return 1;
    int &v = dp[idx][isup][is6];
    if(~v) return v;
    int up = isup ? d[idx] : 9;
    int ans = 0;
    rep(i,0,up){
        if(i == 4) continue;
        if(is6 && i == 2) continue;
        ans += dfs(idx - 1,i == 6,isup && i == up);
    }
    return v = ans;
}

int solve(int x){
    dl = 0;
    for(;x;x /= 10) d[dl++] = x % 10;
    memset(dp,-1,sizeof dp);
    return dfs(dl - 1,false,true);
}

int main(int argc, char** argv) {
    int a,b;
    while(~scanf("%d%d",&a,&b) && (a || b)){
        printf("%d\n",solve(b) - solve(a - 1));
    }
    return 0;
}
hans774882968
关注
  • 2
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
ACM-ICPC 2020年上海区域赛正式赛试题
12-16
【ACM-ICPC 2020年上海区域赛正式赛试题】是国际大学生程序设计竞赛（ICPC）的一部分，由国际计算机协会（ACM）主办，旨在考验参赛大学生的创新思维、团队协作和在高压环境下编程、分析问题及解决问题的能力。...
2020icpc沈阳站K题（csdn）————程序.pdf
12-01
2020icpc沈阳站K题（csdn）————程序 本文讨论的知识点包括： 1. ROC曲线和AUC计算：在 Pattern Recognition 和 Machine Learning 中，Receiver Operating Characteristic Curve（ROC曲线）是一种常用的评估...
数位DP题目
最新发布
qq_61130688的博客
10-28 251
f(x)定义为x二进制下1的个数，定义一个m位的01串a，看 [0,L] 有多少数字满足对于任意 i 属于 [0, m-1]，f(x+i) mod2 = ai。
Sum of Log(icpc上海站2020) 数位dp
m0_50623076的博客
12-22 1519
思路： 如果算上i=0&&j=0的情况，其实就是求[0,x][0,y]之间任意两个整数中，满足i&j=0的个数乘以log2(i+j)+1取下整数。因为数据范围到了1e9并且测试样例1e5。所以联想到对位进行操作，于是考虑到---数位dp。 数位dp: https://blog.csdn.net/m0_50623076/article/details/111564154 这个题模板题，赛场上没有想到，赛场下后悔不已。 根据题意进行明确，i...
2020ICPC(上海) - Sum of Log(数位dp)
Falcon的博客
12-14 2029
题目链接：点击查看 题目大意：给出X 和 Y，求 题目分析：因为涉及到了位运算且看似可以递推，所以考虑数位dp，因为统计答案时的 i 和 j 的与为 0，所以 i + j = i & j，那么取 log 其实就是最高位，也就是 max( highbit_i , highbit_j ) 最简单的状态就是：dp[ pos ][ highbit_x ][ highbit_y ][ x <= X ][ y <= Y ]，不过很可惜会 TLE 因为我们到达终点后，我们只关心 max( ..
2020ICPC上海 C.Sum of Log（数位DP）
tomjobs的博客
12-26 517
思路： 思路很快就出来了，但是实现过程还是有点细节，一直在T。。。 可以观察发现，题中式子的结果就是i∣ji|ji∣j的最高位。那么我们按照X,Y的二进制位跑数位DP就好了。 我初始的状态定义是dp[len][high]dp[len][high]dp[len][high]，代表两个数枚举到了第lenlenlen位，最高位为highhighhigh时候的结果。但是这样定义状态有个问题，由于limitlimitlimit的存在，就是很多状态无法记录不下来，比如X=1000000000，Y=1的时候，前面部分几.
2020 ICPC 亚洲区域赛（上海）C Sum of Log （数位dp）
birdmanqin的博客
12-17 1324
题链：https://ac.nowcoder.com/acm/contest/9925/C 题意：求 。 思路：首先，暴力的话就是枚举i，j。那么算法的话就很容易想到数位dp。看到&运算，肯定是二进制，那就是数二进制位。 再看，因为要求i&j==0，那么i和j中每一位中都最多只能有一个1（每一位只能有00，01，10，3种情况）； 又因为log运算，那么i,j中最高位的1是第几位就是的值。 那么，我们枚举每一个最高位（也就是枚举的值），然后数位dp算i&j==0的个数就行了
2020icpc上海（重温经典）（csdn）————程序.pdf
12-01
2020icpc上海（重温经典）（csdn）————程序
2021ICPC沈阳L（树形dp+容斥）（csdn）————程序.pdf
12-01
本题是2021年ICPC沈阳地区比赛的一道题目，涉及到的主要算法是树形动态规划（Tree DP）和容斥原理（Inclusion-Exclusion Principle）。题目要求在一个拥有2n个顶点的完全图中，有2n-1条边构成了一棵树，现在要从剩余...
2019ICPC南京K题 Triangle（csdn）————程序.pdf
12-01
本题是2019年ICPC南京站的一道编程题目，编号为K，主要涉及到了几何中的点、线段和三角形的知识，以及在编程中处理浮点数精度问题。下面我们将详细解析这个问题的解决方案。 首先，题目要求判断一个点是否位于某条...
2019年icpc上海网络赛 L Digit sum（数位DP）
邵光亮的博客
09-15 918
题目链接：点击这里 这道题对友用的暴力超时，他们打表过了，不过应该是数据水，这道题的正解应该是数位DP。 就是找1-n所有数，b进制下各个位的数的和，这不就是数位DP吗，虽然我DP很差，但是我套了个板子就过了。 AC代码: #include<cstdio> #include<cstring> #include<iostream> #include&lt...
2019icpc上海网络赛 Digit sum（预处理）（数位dp）
Nerdant
09-17 252
https://nanti.jisuanke.com/t/41422 题意：直接看图，简单明了 分析： 这个我觉得有必要从时间复杂度层面分析这道题目。首先如果是无脑采取暴力的做法，时间复杂度是O(1e5*1e6),必然超时，做都不用做，肯定要换算法或者优化。我们可以观察到本题有T（100000）种case，如果我们每次case都要对N这么大的范围进行暴力操作查找，那么肯定是会超时的，所以...
2020ICPC济南 Bit Sequence（数位DP）
tomjobs的博客
01-11 797
思路： 赛中最后3min过的，过的时候都直接叫出来了（虽然还是银，对结果并没有影响 定义dp[len][one][sta][sum]dp[len][one][sta][sum]dp[len][one][sta][sum]，代表遍历到了第iii位，第[8,len][8,len][8,len]位oneoneone的个数为奇数/偶数，第[0,7][0,7][0,7]位选择的值，以及第8位开始连续1的个数的奇偶值sumsumsum。 复杂度O(T∗60∗2∗2∗27)O(T*60*2*2*2^7)O(T∗60∗..
HDU 5898 odd-even number （数位DP） 2016 ACM/ICPC Asia Regional Shenyang Online
blanKey的博客
09-18 600
题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=5898 odd-even number Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others) Total Submission(s): 114    Accepted
2015年ACM/ICPC北京赛区 K题（找规律+数位dp）
LSD20164388的博客
09-30 385
提交连接：http://hihocoder.com/problemset/solution/1395808 题意：给你两个数n(n&lt;=1e18)，mod(mod=3,5,7,257,65537) 定义f(x): f(1)=1; 3*f(n)*f(2*n+1)=f(2*n)*(1+3*f(n)); f(2*n)&lt;6*f(n); 求f(1)~f(n)%mod。 输出最后m...
2019ICPC香港J题（数位dp）
Karmen Yang
10-06 413
题意：求区间L~R中有多少个数X是与f(x)模m同余的，f(x)定义是每一位上的数与其他所有位上数的乘积取和。 思路：裸的数位dp，dp[i][j][k]表示第i位，前面的位数和为j%m,（x-f(x)）%m为k的情况下最后x与f（x）模m同余的方案数，状态转移就非常明显了，可以直接看代码 #include<bits/stdc++.h> using namespace std; #define mod 1000000007 typedef long long ll; typedef long d
Hihocoder 1971 幸运数字 && 2018 上海大都会 ICPC J Beautiful Number【枚举+数位DP】
Irish_Moonshine的博客
07-22 439
http://hihocoder.com/problemset/problem/1791 #include&amp;lt;bits/stdc++.h&amp;gt; using namespace std; const int maxn=1e5+45; #define INF 0x3f3f3f3f; #define ll long long int #define charmax(x,y) x=max(x,y...
2015 ACM/ICPC 北京赛区 现场赛 K — A Math Problem【规律+数位dp】
lml11111的博客
10-01 376
http://media.hihocoder.com/contests/icpcbeijing2015/problems.pdf 题意： 给你两个数n(n&lt;=1e18)，mod(mod=3,5,7,257,65537) 定义f(x): f(1)=1; 已知：3*f(n)*f(2*n+1)=f(2*n)*(1+3*f(n));            f(2*n)&lt;6*f(...
2020ICPC济南站L-Bit Sequence(数位dp)
qq_35577488的博客
02-02 1046
题目链接 题目思路: 题目条件比较繁杂，一步一步分析. 1.要求同时满足mmm个条件，那么递归出口就不是O(1)O(1)O(1)的判断，而是循环判断.
2019 ICPC亚洲上海站在线比赛题解：数据结构与算法应用
在2019年的ICPC亚洲上海区域赛的在线竞赛中，参赛者们面对了一系列具有挑战性的算法题目。本文将解析其中的三个题目，包括A-LightningRoutingI、B-Lightbulbs和C-Triple。 A-LightningRoutingI: 这道题目涉及在一个...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值