結果
問題 | No.319 happy b1rthday 2 me |
ユーザー |
|
提出日時 | 2021-09-15 02:49:48 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 3 ms / 2,000 ms |
コード長 | 4,198 bytes |
コンパイル時間 | 2,552 ms |
コンパイル使用メモリ | 196,280 KB |
最終ジャッジ日時 | 2025-01-24 14:04:05 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 29 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define rep(i,n) for(ll i=0;i<n;i++) #define repl(i,l,r) for(ll i=(l);i<(r);i++) #define per(i,n) for(ll i=(n)-1;i>=0;i--) #define perl(i,r,l) for(ll i=r-1;i>=l;i--) #define fi first #define se second #define pb push_back #define ins insert #define pqueue(x) priority_queue<x,vector<x>,greater<x>> #define all(x) (x).begin(),(x).end() #define CST(x) cout<<fixed<<setprecision(x) #define vtpl(x,y,z) vector<tuple<x,y,z>> #define rev(x) reverse(x); using ll=long long; using vl=vector<ll>; using vvl=vector<vector<ll>>; using pl=pair<ll,ll>; using vpl=vector<pl>; using vvpl=vector<vpl>; const ll MOD=1000000007; const ll MOD9=998244353; const int inf=1e9+10; const ll INF=4e18; const ll dy[9]={0,1,-1,0,1,1,-1,-1,0}; const ll dx[9]={1,0,0,-1,1,-1,1,-1,0}; template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; } template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; } ll f(ll m){ string s=to_string(m); ll n=s.size(); ll dp[n+1][2][10][2];//i桁目、直前が1か?、12を何個作ったか,small memset(dp,0,sizeof(dp)); dp[0][0][0][0]=1; rep(i,n){ ll p=s[i]-'0'; rep(j,9){ rep(to,10){ if(to==1){ dp[i+1][1][j][1]+=dp[i][0][j][1]; dp[i+1][1][j][1]+=dp[i][1][j][1]; } else if(to==2){ dp[i+1][0][j][1]+=dp[i][0][j][1]; dp[i+1][0][j+1][1]+=dp[i][1][j][1]; } else { dp[i+1][0][j][1]+=dp[i][0][j][1]; dp[i+1][0][j][1]+=dp[i][1][j][1]; } } rep(to,p){ if(to==1){ dp[i+1][1][j][1]+=dp[i][0][j][0]; dp[i+1][1][j][1]+=dp[i][1][j][0]; } else if(to==2){ dp[i+1][0][j][1]+=dp[i][0][j][0]; dp[i+1][0][j+1][1]+=dp[i][1][j][0]; } else { dp[i+1][0][j][1]+=dp[i][0][j][0]; dp[i+1][0][j][1]+=dp[i][1][j][0]; } } if(p==1){ dp[i+1][1][j][0]+=dp[i][0][j][0]; dp[i+1][1][j][0]+=dp[i][1][j][0]; } else if(p==2){ dp[i+1][0][j][0]+=dp[i][0][j][0]; dp[i+1][0][j+1][0]+=dp[i][1][j][0]; } else { dp[i+1][0][j][0]+=dp[i][0][j][0]; dp[i+1][0][j][0]+=dp[i][1][j][0]; } } } ll ans=0; rep(i,10)rep(k,2)rep(t,2)ans+=dp[n][k][i][t]*i; return ans; } ll g(ll m){ string s=to_string(m); ll n=s.size(); ll dp[n+1][3][2][2];//i桁め、どこまで進んだか、small,lead memset(dp,0,sizeof(dp)); dp[0][0][0][0]=1; rep(i,n){ ll p=s[i]-'0'; if(p==2)dp[i+1][1][0][1]+=dp[i][0][0][0]; rep(to,p){ if(to==2)dp[i+1][1][1][1]+=dp[i][0][0][0]; else if(to==0)dp[i+1][0][1][0]+=dp[i][0][0][0]; //else dp[i+1][0][1][0]+=dp[i][0][0][0];もう可能性がない } rep(to,10){ if(to==2)dp[i+1][1][1][1]+=dp[i][0][1][0]; else if(to==0)dp[i+1][0][1][0]+=dp[i][0][1][0]; } dp[i+1][1][0][1]+=dp[i][1][0][1]; rep(to,p)dp[i+1][1][1][1]+=dp[i][1][0][1]; rep(to,10)dp[i+1][1][1][1]+=dp[i][1][1][1]; if(i==n-1){ if(p==1)dp[i+1][2][0][1]+=dp[i][1][0][1]; rep(to,p){ if(to==1)dp[i+1][2][1][1]+=dp[i][1][0][1]; } rep(to,10){ if(to==1)dp[i+1][2][1][1]+=dp[i][1][1][1]; } } } return dp[n][2][0][1]+dp[n][2][1][1]; } int main(){ ll a,b;cin >> a >> b; //cout << f(b)-f(a-1) << endl; //cout << g(b)-g(a-1) << endl; ll ans=f(b)-f(a-1)+g(b)-g(a-1); if(a==1&&b>=2)ans++; string sb=to_string(b); if(sb.front()=='2'&&sb.back()=='1')ans--; cout << ans << endl; }