結果

問題 No.319 happy b1rthday 2 me
コンテスト
ユーザー Yang33
提出日時 2018-10-02 15:52:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,245 bytes
コンパイル時間 1,698 ms
コンパイル使用メモリ 169,960 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-12 10:03:43
合計ジャッジ時間 2,681 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 27 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using VS = vector<string>;    using LL = long long;
using VI = vector<int>;       using VVI = vector<VI>;
using PII = pair<int, int>;   using PLL = pair<LL, LL>;
using VL = vector<LL>;        using VVL = vector<VL>;

#define ALL(a)  begin((a)),end((a))
#define RALL(a) (a).rbegin(), (a).rend()
#define SZ(a) int((a).size())
#define SORT(c) sort(ALL((c)))
#define RSORT(c) sort(RALL((c)))
#define UNIQ(c) (c).erase(unique(ALL((c))), end((c)))
#define FOR(i, s, e) for (int(i) = (s); (i) < (e); (i)++)
#define FORR(i, s, e) for (int(i) = (s); (i) > (e); (i)--)
#define debug(x) cerr << #x << ": " << x << endl
const int INF = 1e9;                          const LL LINF = 1e16;
const LL MOD = 1000000007;                    const double PI = acos(-1.0);
int DX[8] = { 0, 0, 1, -1, 1, 1, -1, -1 };    int DY[8] = { 1, -1, 0, 0, 1, -1, 1, -1 };

/* -----  2018/10/02  Problem: yukicoder 319  / Link: http://yukicoder.me/problems/no/319  ----- */
/* ------問題------

りあんくんは, 正整数 A,B(A≤B) に対し A から B までの数字を十進表記でつなげた文字列を考え, 「(A,B)文字列」と名付けることとしました.
例えば「(3,5)文字列」は "345", 「(8,13)文字列」は "8910111213", 「(102,102)文字列」は "102" となります.
ここでりあんくんは, ある正整数 A,B からなる「(A,B)文字列」に含まれる "12" という連続部分文字列の個数が気になってしまいました.
初めてのぽしゃけで酔っぱらってしまいダメダメなりあんくんの代わりに, 「(A,B)文字列」に含まれる "12" の個数を求めてあげてください.

-----問題ここまで----- */
/* -----解説等-----


----解説ここまで---- */

LL dp[14][2][7];
LL f(const string & s, int i, int preone = 0, int cnt = 0, int less = 0) {
	if (i == -1) {
		return cnt;
	}
	if (dp[i][preone][cnt] != -1 && less == 1)return dp[i][preone][cnt];

	LL res = 0;
	int Max = less ? 9 : s[i] - '0';
	FOR(nx, 0, Max + 1) {
		res += f(s, i - 1, nx == 1, cnt + (preone && (nx == 2)), less || nx < Max);
	}

	if (less == 1)return dp[i][preone][cnt] = res;
	return res;
}

LL dp2[14][2][2][2];
LL g(const string& s, int i, int lz = 1, int firsttwo = 0, int endtwo = 0, int less = 0) {
	if (i == -1) {
		return (!lz) && firsttwo&&endtwo;
	}

	if (less && dp2[i][lz][firsttwo][endtwo] != -1) { return dp2[i][lz][firsttwo][endtwo]; }
	LL res = 0;
	int Max = less ? 9 : s[i] - '0';
	FOR(nx, 0, Max + 1) {
		res += g(s, i - 1, lz&&nx == 0, firsttwo || (lz&&nx == 2), i == 0 && nx == 2, less || nx < Max);
	}

	if (less)return dp2[i][lz][firsttwo][endtwo] = res;
	return res;
}

int main() {
	cin.tie(0);
	ios_base::sync_with_stdio(false);
	string A, B; cin >> A >> B;
	// A--
	FORR(i, SZ(A) - 1, 0 - 1) {
		if (A[i] == '0') {
			A[i] = '9';
		}
		else {
			A[i]--; break;
		}
	}
	if (A[0] == '0')A = A.substr(1);

	LL ans = 0;
	if (A.front() == '2'&&A.back() == '1')ans--;
	fill(**dp, **dp + 14 * 2 * 7, -1);
	fill(***dp2, ***dp2 + 14 * 2 * 2 * 2, -1);

	reverse(ALL(A));
	reverse(ALL(B));

	ans += f(B, SZ(B) - 1) - f(A, SZ(A) - 1);
	LL ret = g(B, SZ(B) - 1) - g(A, SZ(A) - 1);
	ans += ret;

	cout << ans << "\n";

	return 0;
}
0