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;

}

};

相关推荐
初圣魔门首席弟子4 小时前
C++ STL 向量(vector)学习笔记:从基础到实战
c++·笔记·学习
青草地溪水旁4 小时前
Visual Studio Code中launch.json深度解析:C++调试的艺术
c++·vscode·json
m0_552200825 小时前
《UE5_C++多人TPS完整教程》学习笔记62 ——《P63 多人游戏中的开火特效(Fire Effects in Multiplayer)》
c++·游戏·ue5
liu****5 小时前
基于websocket的多用户网页五子棋(九)
服务器·网络·数据库·c++·websocket·网络协议·个人开发
liu****5 小时前
基于websocket的多用户网页五子棋(八)
服务器·前端·javascript·数据库·c++·websocket·个人开发
ajassi20005 小时前
开源 C++ QT QML 开发(十二)通讯--TCP客户端
c++·qt·开源
2401_881244405 小时前
P3808 AC 自动机(简单版)
算法
进击的圆儿6 小时前
【学习笔记05】C++11新特性学习总结(下)
c++·笔记·学习
Jayden_Ruan6 小时前
C++十进制转二进制
数据结构·c++·算法