結果
問題 | 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を何個作ったか,smallmemset(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,leadmemset(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;}