// #include <bits/stdc++.h>
#include <iostream>
#define pb push_back
#define s second
#define f first
#define ll long long
#define int ll
using namespace std;
string str;
int dp[15][15][25][5];
int get(int last2,int last,int len,int sm){
if (len >= str.size()) return 1;
if (dp[last2][last][len][sm] != -1)
return dp[last2][last][len][sm];
int l = str[len] - '0',sum = 0;
if (sm){
for (int i = 0; i <= 9; i++){
if (i != last && i != last2)
sum += get(last,i,len + 1,1);
}
}
else{
for (int i = 0; i < l; i++){
if (i != last && i != last2)
sum += get(last,i,len + 1,1);
}
if (l != last && l != last2)
sum += get(last,l,len + 1,0);
}
return dp[last2][last][len][sm] = sum;
}
int G(int x){
if (x < 0) return 0;
memset(dp, -1, sizeof(dp));
str = "";
while(x > 0){
str+=char((x%10)+'0');
x/=10;
}
reverse(str.begin(),str.end());
int l = str[0] - '0',ans = 1;
for (int i = 1; i < l; i++)
ans += get(i,i,1,1);
ans += get(l,l,1,0);
for (int i = 2; i <= str.size(); i++)
for (int j=1;j<=9;j++)
ans += get(j,j,i,1);
return ans;
}
signed main (){
ios_base::sync_with_stdio(0),cin.tie(NULL),cout.tie(NULL);
ll a,b;
cin>>a>>b;
cout<<G(b) - G(a - 1)<<"\n";
}
Compilation message
numbers.cpp: In function 'long long int get(long long int, long long int, long long int, long long int)':
numbers.cpp:14:13: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
14 | if (len >= str.size()) return 1;
| ~~~~^~~~~~~~~~~~~
numbers.cpp: In function 'long long int G(long long int)':
numbers.cpp:38:5: error: 'memset' was not declared in this scope
38 | memset(dp, -1, sizeof(dp));
| ^~~~~~
numbers.cpp:3:1: note: 'memset' is defined in header '<cstring>'; did you forget to '#include <cstring>'?
2 | #include <iostream>
+++ |+#include <cstring>
3 | #define pb push_back
numbers.cpp:45:5: error: 'reverse' was not declared in this scope
45 | reverse(str.begin(),str.end());
| ^~~~~~~
numbers.cpp:51:23: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
51 | for (int i = 2; i <= str.size(); i++)
| ~~^~~~~~~~~~~~~
