제출 #444936

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
444936		aryan12	Palindrome-Free Numbers (BOI13_numbers)	C++17	72.50 / 100	2 ms	356 KiB

numbers

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

mt19937_64 RNG(chrono::steady_clock::now().time_since_epoch().count());

int dp[20][12][12][2];

vector<int> Convert(int x) {
	vector<int> digits;
	while(x != 0) {
		digits.push_back(x % 10);
		x /= 10;
	}
	reverse(digits.begin(), digits.end());
	/*for(int i = 0; i < digits.size(); i++) {
		cout << digits[i] << " ";
	}
	cout << "\n";*/
	return digits;
}

void Ini() {
	for(int i = 0; i < 20; i++) {
		for(int j = 0; j < 12; j++) {
			for(int k = 0; k < 12; k++) {
				for(int l = 0; l < 2; l++) {
					dp[i][j][k][l] = -1;
				}
			}
		}
	}
}

int DigitDP(vector<int> s, int pos, int second_last, int last, bool edge) {
	//cout << "pos = " << pos << ", second_last = " << second_last << ", last = " << last << ", edge = " << edge << "\n";
	if(pos == s.size()) {
		return 1;
	}
	if(dp[pos][second_last][last][(edge) ? 1 : 0] != -1) {
		return dp[pos][second_last][last][(edge) ? 1 : 0];
	}
	int ans = 0;
	if(edge) {
		for(int i = 0; i < s[pos]; i++) {
			if(i != second_last && i != last) {
				ans += DigitDP(s, pos + 1, last, i, false);
			}
		}
		if(s[pos] != second_last && s[pos] != last) {
			ans += DigitDP(s, pos + 1, last, s[pos], true);
		}
	}
	else {
		for(int i = 0; i < 10; i++) {
			if(i != second_last && i != last) {
				ans += DigitDP(s, pos + 1, last, i, false);
			}
		}
	}
	return dp[pos][second_last][last][(edge) ? 1 : 0] = ans;
}

void Solve() {
	int left, right;
	cin >> left >> right;
	vector<int> l = Convert(left - 1), r = Convert(right);
	Ini();
	int ans = DigitDP(r, 0, 10, 10, true);
	Ini();
	//cout << "ans = " << ans << "\n";
	if(left != 0) {
		ans -= DigitDP(l, 0, 10, 10, true);
	}
	cout << ans << "\n";
}

int32_t main() {
	auto begin = std::chrono::high_resolution_clock::now();
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int t = 1;
	//cin >> t;
	while(t--) {
		Solve();
	}
	auto end = std::chrono::high_resolution_clock::now();
    auto elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - begin);
    cerr << "Time measured: " << elapsed.count() * 1e-9 << " seconds.\n"; 
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

numbers.cpp: In function 'long long int DigitDP(std::vector<long long int>, long long int, long long int, long long int, bool)':
numbers.cpp:37:9: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |  if(pos == s.size()) {
      |     ~~~~^~~~~~~~~~~

• Subtask 1: 12.5 / 25
• Subtask 2: 60 / 75