結果

問題 No.319 happy b1rthday 2 me
コンテスト
ユーザー koyumeishi
提出日時 2015-12-12 01:30:32
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 2,634 bytes
コンパイル時間 1,034 ms
コンパイル使用メモリ 85,044 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-15 08:33:31
合計ジャッジ時間 1,826 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 24 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <set>
using namespace std;
/*
1***2 1***3 1***4 ... 
2***1 2***2

dp[sml][last] := cnt 12
dp[head][edge]

*/

//resize(vector, {v1,v2,v3});

void resize(int& val , vector<int>::iterator itr){val = 0;}
void resize(long long& val , vector<int>::iterator itr){val = 0;}
void resize(double& val , vector<int>::iterator itr){val = 0;}

template<class T>
void resize(vector<T>& vec, vector<int>::iterator itr){
	vec.resize(*itr);
	for(int i=0; i<*itr; i++){
		resize(vec[i], itr+1);
	}
}

template<class T>
void resize(T& vec, vector<int> sz){
	resize(vec, sz.begin());
}

template<class T>
ostream& operator << (ostream& os, vector<T> vec){
	for(int i=0; i<vec.size(); i++){
		os << vec[i] << " ";
	}
	os << endl;
	return os;
}

long long calc(string& x){
	vector<vector<vector<long long>>> dp(2, vector<vector<long long>>(10,vector<long long>(11, 0)));
	dp[0][0][0] = 1;

	for(int i=0; i<x.size(); i++){
		int k = x[i]-'0';
		vector<vector<vector<long long>>> dp_(2, vector<vector<long long>>(10,vector<long long>(11, 0)));

		for(int a=0; a<2; a++)
		for(int b=0; b<10; b++)
		for(int c=0; c<10; c++){
			if(dp[a][b][c] == 0) continue;
			for(int d=0; d<10; d++){
				if(a==0 && k<d) break;
				int a_ = a|(d<k);
				int b_ = d;
				int c_ = c + ((b*10+d==12)?1:0);

				dp_[a_][b_][c_] += dp[a][b][c];
			}
		}
		swap(dp, dp_);
	}
	long long ret = 0;
	for(int a=0; a<2; a++)
	for(int b=0; b<10; b++)
	for(int c=1; c<11; c++){
		ret += dp[a][b][c] * c;
	}
	return ret;
}
/*
10 11 12 13 14
*/

long long calc2(string &x){
	vector<vector<long long>> dp(4, vector<long long>(2, 0));

	long long ret = 0;
	dp[0][1] = 1;
	for(int i=0; i<x.size(); i++){
		int k = x[i] - '0';
		vector<vector<long long>> dp_(4, vector<long long>(2, 0));

		for(int a=0; a<4; a++)
		for(int b=0; b<2; b++){
			if(dp[a][b]==0) continue;
			for(int d=0; d<10; d++){
				if(b==1 && d>k) break;
				int a_ = (a>0?a:min(d,3));
				int b_ = b & (d==k);

				dp_[a_][b_] += dp[a][b];

				if(i==x.size()-1){
					//if(a_==1 && d==2 && b_==0){
					//	ret += dp[a][b];
					//}
					if(a_==2 && d==1 && b_==0){
						ret += dp[a][b];
					}
				}
			}
		}
		swap(dp, dp_);
	}
	return ret;
}

int main(){
	long long a,b;
	cin >> a >> b;
	a--;

	string a_,b_;
	char buff[100];
	sprintf(buff,"%lld", a);
	a_ = buff;
	sprintf(buff,"%lld", b);
	b_ = buff;

	long long ans = calc(b_);
	ans += calc2(b_);

	ans -= calc(a_);
	ans -= calc2(a_);

	if(a<1 && b>=2) ans++;

	cout << ans << endl;

	return 0;
}
0