이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#pragma GCC optimize ("Ofast,unroll-loops")
#pragma GCC target ("avx2")
using namespace std;
typedef long long ll;
typedef pair<int, int> pp;
#define er(args ...) cerr << __LINE__ << ": ", err(new istringstream(string(#args)), args), cerr << endl
#define per(i,r,l) for(int i = (r); i >= (l); i--)
#define rep(i,l,r) for(int i = (l); i < (r); i++)
#define all(x) x.begin(), x.end()
#define sz(x) (int)(x).size()
#define pb push_back
#define ss second
#define ff first
void err(istringstream *iss){}template<typename T,typename ...Args> void err(istringstream *iss,const T &_val, const Args&...args){string _name;*iss>>_name;if(_name.back()==',')_name.pop_back();cerr<<_name<<" = "<<_val<<", ",err(iss,args...);}
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const ll mod = 1e9 + 7, maxn = 5e5+5, d = 11, lg = 19, inf = ll(1e9) + 5;
ll pw(ll a,ll b,ll md=mod){if(!b)return 1;ll k=pw(a,b>>1ll);return k*k%md*(b&1ll?a:1)%md;}
bool chk(ll a){
vector<int> d;
while(a) d.pb(a%10), a /= 10;
rep(i,1,sz(d)) if(d[i-1] == d[i] || (i>1 && d[i-2] == d[i])) return false;
return true;
}
ll dp[lg][d][d];
ll f(ll a){
vector<int> dig;
while(a) dig.pb(a%10), a /= 10; reverse(all(dig));
ll res = 0;
if(sz(dig)) res = dp[sz(dig)-1][10][10];
rep(i,0,sz(dig)){
rep(k,0,dig[i]){
if((i && k == dig[i-1]) || (i > 1 && k == dig[i-2]) || (!k && !i)) continue;
res += dp[sz(dig)-i-1][(i?dig[i-1]:10)][k];
}
if((i && dig[i] == dig[i-1]) || (i > 1 && dig[i] == dig[i-2])) break;
}
return res;
}
void init(){
rep(i,0,d) rep(j,0,d) dp[0][i][j] = 1;
rep(l,1,lg){
rep(i,0,d){
rep(j,0,d){
rep(k,0,d-1) if(k - i && k - j) {
if(i == 10 && j == 10 && k == 0) dp[l][i][j] += dp[l-1][i][i];
else dp[l][i][j] += dp[l-1][j][k];
}
}
}
}
}
ll slv1(ll l, ll r){
return f(r+1) - f(l);
}
ll slv2(ll l, ll r){
ll res = 0;
rep(i,l,r+1) res += chk(i);
return res;
}
int main(){
cin.tie(0) -> sync_with_stdio(0); init();
ll l, r; cin >> l >> r;
cout << slv1(l, r) << '\n';
return 0;
}
컴파일 시 표준 에러 (stderr) 메시지
numbers.cpp: In function 'll f(ll)':
numbers.cpp:31:6: warning: this 'while' clause does not guard... [-Wmisleading-indentation]
31 | while(a) dig.pb(a%10), a /= 10; reverse(all(dig));
| ^~~~~
numbers.cpp:31:38: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'while'
31 | while(a) dig.pb(a%10), a /= 10; reverse(all(dig));
| ^~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |