10.5 数位dp

lc417

dfs双标记

class Solution {

public:

int dx[4]={0,0,1,-1};

int dy[4]={1,-1,0,0};

int m,n;

vector<vector<bool>> pvis,avis;

vector<vector<int>> ret;

//双标记法 不要搅合在一起

//!!!!!!清晰 正确是首要! 不要偷懒

vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights)

{

m=heights.size();

n=heights[0].size();

pvis.resize(m,vector<bool>(n,false));

avis.resize(m,vector<bool>(n,false));

for(int i=0;i<m;i++)

{

for(int j=0;j<n;j++)

{

if(i==0 || j==0)

{

pvis[i][j]=true;

dfs(heights,i,j,pvis);

}

if(i==m-1 || j==n-1)

{

avis[i][j]=true;

dfs(heights,i,j,avis);

}

}

}

for(int i=0;i<m;i++)

{

for(int j=0;j<n;j++)

{

if(pvis[i][j]==true && avis[i][j]==true)

ret.push_back({i,j});

}

}

return ret;

}

//如何在DFS中区分双标记位呢

void dfs(vector<vector<int>>& heights,int i,int j,vector<vector<bool>>& vis)

{

for(int k=0;k<4;k++)

{

int x=i+dx[k],y=j+dy[k];

if(x>=0 && x<m && y>=0 && y<n

&& heights[i][j]<=heights[x][y]

&& !vis[x][y])

{

vis[x][y]=true;

dfs(heights,x,y,vis);

}

}

}

};

lc3704

数位dp

class Solution {

public:

long long countNoZeroPairs(long long n) {

const int MAXP = 20;

long long f[MAXP][2][2][2];

memset(f, -1, sizeof(f));

vector<int> A;

for (int p = 0; p < MAXP; p++, n /= 10) A.push_back(n % 10);

auto dp = [&](this auto &&dp, int pos, int c, int za, int zb) -> long long {

if (pos < 0) return c == 0 && za == 1 && zb == 1 ? 1 : 0;

long long &ret = f[pos][c][za][zb];

if (ret >= 0) return ret;

ret = 0;

// 枚举 a 这一位填什么

for (int a = 0; a <= 9; a++) {

if (a == 0 && za) continue;

// 枚举前一位是否进位

for (int cc = 0; cc <= 1; cc++) {

// 再根据这一位是否进位,就能算出 b 这一位要填什么

int b = A[pos] + (c ? 10 : 0) - a - cc;

if (b < 0 || b > 9) continue;

if (b == 0 && zb) continue;

ret += dp(pos - 1, cc, za || a > 0 ? 1 : 0, zb || b > 0 ? 1 : 0);

}

}

return ret;

};

return dp(MAXP - 1, 0, 0, 0);

}

};

模板

/*

//<https://oi-wiki.org/dp/number/\>

#include <cstdio>

using namespace std;

constexpr int N = 15;

using ll = long long;

ll l, r, dp[N], mi[N];

ll ans1[N], ans2[N];

int a[N];

void solve(ll n, ll *ans) {

ll tmp = n;

int len = 0;

while (n) a[++len] = n % 10, n /= 10;

for (int i = len; i >= 1; --i) {

for (int j = 0; j < 10; j++) ans[j] += dp[i - 1] * a[i];

for (int j = 0; j < a[i]; j++) ans[j] += mi[i - 1];

tmp -= mi[i - 1] * a[i], ans[a[i]] += tmp + 1;

ans[0] -= mi[i - 1];

}

}

int main() {

scanf("%lld%lld", &l, &r);

mi[0] = 1ll;

for (int i = 1; i <= 13; ++i) {

dp[i] = dp[i - 1] * 10 + mi[i - 1];

mi[i] = 10ll * mi[i - 1];

}

solve(r, ans1), solve(l - 1, ans2);

for (int i = 0; i < 10; ++i) printf("%lld ", ans1[i] - ans2[i]);

return 0;

}

*/

class Solution {

typedef long long ll;

public:

long long countNoZeroPairs(long long n)

{

if(n<=0)

return 0;

ll cnt[10][2][2];

ll m[2][2];

bool init=false;

if(!init)

{

memset(cnt,0,sizeof(cnt));

for(int d=0;d<=9;d++)

{

for(int i=0;i<=1;i++)

{

for(int a=1;a<=9;a++)

{

for(int b=1;b<=9;b++)

{

int s=a+b+i;

if(s%10==d)

{

if(s>=10)

++cnt[d][i][1];

else

++cnt[d][i][0];

}

}

}

}

}

m[0][0]=m[0][1]=m[1][0]=m[1][1]=0;

for(int d=0;d<=9;d++)

{

m[0][0]+=cnt[d][0][0];

m[0][1]+=cnt[d][0][1];

m[1][0]+=cnt[d][1][0];

m[1][1]+=cnt[d][1][1];

}

init=true;

}

auto t_cnt=[&](int c,int l)->ll

{

ll dp0=(c==0);

ll dp1=(c==1);

for(int i=0;i<l;i++)

{

ll ndp0=dp0*m[0][0]+dp1*m[1][0];

ll ndp1=dp0*m[0][1]+dp1*m[1][1];

dp0=ndp0;

dp1=ndp1;

}

return dp0;

};

int a[20],len=0;

ll t=n;

while(t)

{

a[++len]=(int)(t%10);

t/=10;

}

ll ret=0;

ll eq0=1,eq1=0;

for(int i=len;i>=1;i--)

{

int cur=a[i];

if(eq0)

{

for(int d=1;d<cur;d++)

{

ll w0=cnt[d][0][0];

ll w1=cnt[d][0][1];

if(w0)

ret+=eq0*w0*t_cnt(0,i-1);

if(w1)

ret+=eq0*w1*t_cnt(1,i-1);

}

}

if(eq1)

{

for(int d=1;d<cur;d++)

{

ll w0=cnt[d][1][0];

ll w1=cnt[d][1][1];

if(w0)

ret+=eq1*w0*t_cnt(0,i-1);

if(w1)

ret+=eq1*w1*t_cnt(1,i-1);

}

}

ll ne0=0,ne1=0;

if(eq0)

{

ne0+=eq0*cnt[cur][0][0];

ne1+=eq0*cnt[cur][0][1];

}

if(eq1)

{

ne0+=eq1*cnt[cur][1][0];

ne1+=eq1*cnt[cur][1][1];

}

eq0=ne0;

eq1=ne1;

}

ret+=eq0;

return ret;

}

};

相关推荐
一念&3 小时前
每日一个C语言知识:C 头文件
c语言·开发语言·算法
Miraitowa_cheems7 小时前
LeetCode算法日记 - Day 88: 环绕字符串中唯一的子字符串
java·数据结构·算法·leetcode·深度优先·动态规划
B站_计算机毕业设计之家7 小时前
python电商商品评论数据分析可视化系统 爬虫 数据采集 Flask框架 NLP情感分析 LDA主题分析 Bayes评论分类(源码) ✅
大数据·hadoop·爬虫·python·算法·数据分析·1024程序员节
小白菜又菜8 小时前
Leetcode 1518. Water Bottles
算法·leetcode·职场和发展
长存祈月心8 小时前
Rust Option 与 Result深度解析
算法
杭州杭州杭州9 小时前
机器学习(3)---线性算法,决策树,神经网络,支持向量机
算法·决策树·机器学习
星竹晨L9 小时前
C++继承机制:面向对象编程的基石
开发语言·c++
9ilk10 小时前
【仿RabbitMQ的发布订阅式消息队列】--- 模块设计与划分
c++·笔记·分布式·后端·中间件·rabbitmq
不语n10 小时前
快速排序(Quick Sort)详解与图解
数据结构·算法·排序算法·快速排序·双指针排序
恒者走天下10 小时前
面试的时候项目怎么聊,才能发挥最大的价值
c++