Submission #144798

#	Time	Username	Problem	Language	Result	Execution time	Memory
144798		emilem	Ljetopica (COI19_ljetopica)	C++14	0 / 100	63 ms	26216 KiB

ljetopica

#include <algorithm>
#include <iostream>
#include <vector>
#include <string>
#include <set>
using namespace std;
const long long mod = 1000000007;

int n, k;
string path, a, b;
vector<int> pow2(2000);
int dp[1005][1005][2][2], cnt[1005][1005][2][2];

void Add(int depth, int steps, int ok, int atBound, int depth2, int steps2, int ok2, int atBound2, char dir)
{
	if (max(depth, depth2) > n || max(steps, steps2) > k)
		return;
	dp[depth][steps][ok][atBound] += 2 * dp[depth2][steps2][ok2][atBound2];
	if (dir == 'R')
		dp[depth][steps][ok][atBound] += cnt[depth2][steps2][ok2][atBound2];
	cnt[depth][steps][ok][atBound] += cnt[depth2][steps2][ok2][atBound2];
}
int Solve(string bound)
{
	dp[1][0][0][1] = dp[1][0][1][1] = 1;
	cnt[1][0][0][1] = cnt[1][0][1][1] = 1;
	for (int depth = 1; depth < n; ++depth)
		for (int steps = 0; steps <= k; ++steps)
			for (int ok = 0; ok <= 1; ++ok)
			{
				if (depth == 3)
					cerr << "";
				// += dp[][][][0]
				if (path[depth - 1] == (ok ? 'L' : 'R'))
					Add(depth + 1, steps, ok, 0, depth, steps, ok, 0, 'L');
				else
					Add(depth + 1, steps, ok, 0, depth, steps, ok, 0, 'R');
				if (path[depth - 1] == (!ok ? 'L' : 'R'))
					Add(depth + 1, steps + 1, !ok, 0, depth, steps, ok, 0, 'L');
				else
					Add(depth + 1, steps + 1, !ok, 0, depth, steps, ok, 0, 'R');
				// += dp[][][][1]
				if (ok)
				{
					if (path[depth - 1] == 'L')
						Add(depth + 1, steps, ok, bound[depth - 1] == 'R' ? 0 : 1, depth, steps, ok, 1, 'L');
					else if (bound[depth - 1] == 'R')
						Add(depth + 1, steps, ok, 1, depth, steps, ok, 1, 'R');
				}
				else
				{
					if (path[depth - 1] == 'R')
						Add(depth + 1, steps, ok, bound[depth - 1] == 'R' ? 0 : 1, depth, steps, ok, 1, 'L');
					else if (bound[depth - 1] == 'R')
						Add(depth + 1, steps, ok, 1, depth, steps, ok, 1, 'R');
				}
				if (!ok)
				{
					if (path[depth - 1] == 'L')
						Add(depth + 1, steps + 1, !ok, bound[depth - 1] == 'R' ? 0 : 1, depth, steps, ok, 1, 'L');
					else if (bound[depth - 1] == 'R')
						Add(depth + 1, steps + 1, !ok, 1, depth, steps, ok, 1, 'R');
				}
				else
				{
					if (path[depth - 1] == 'R')
						Add(depth + 1, steps + 1, !ok, bound[depth - 1] == 'R' ? 0 : 1, depth, steps, ok, 1, 'L');
					else if (bound[depth - 1] == 'R')
						Add(depth + 1, steps + 1, !ok, 1, depth, steps, ok, 1, 'R');
				}
			}
	int ans = 0;
	for (int i = 0; i <= 1; ++i)
		for (int j = 0; j <= 1; ++j)
			ans += dp[n][k][i][j];
	return ans;
}
int main()
{
	pow2[0] = 1;
	for (int i = 1; i < pow2.size(); ++i)
		pow2[i] = (pow2[i - 1] + pow2[i - 1]) % mod;
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	cin >> n >> k;
	cin >> path;
	cin >> a >> b;
	reverse(a.begin(), a.end());
	for (int i = 0; i < a.length(); ++i)
		if (a[i] == '0')
			a[i] = '1';
		else
		{
			a[i] = '0';
			if (i + 1 == a.length())
				a.pop_back();
			break;
		}
	reverse(b.begin(), b.end());
	a.pop_back();
	b.pop_back();
	for (int i = 0; i < a.length(); ++i)
		a[i] = (a[i] == '0') ? 'L' : 'R';
	for (int i = 0; i < b.length(); ++i)
		b[i] = (b[i] == '0') ? 'L' : 'R';
	reverse(a.begin(), a.end());
	reverse(b.begin(), b.end());
	cout << -Solve(b) + Solve(a) << endl;

	char I;
	cin >> I;
}

Compilation message (stderr)

ljetopica.cpp: In function 'int main()':
ljetopica.cpp:81:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 1; i < pow2.size(); ++i)
                  ~~^~~~~~~~~~~~~
ljetopica.cpp:90:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < a.length(); ++i)
                  ~~^~~~~~~~~~~~
ljetopica.cpp:96:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    if (i + 1 == a.length())
        ~~~~~~^~~~~~~~~~~~~
ljetopica.cpp:103:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < a.length(); ++i)
                  ~~^~~~~~~~~~~~
ljetopica.cpp:105:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < b.length(); ++i)
                  ~~^~~~~~~~~~~~

• Subtask 1: 0 / 8
• Subtask 2: 0 / 14
• Subtask 3: 0 / 17
• Subtask 4: 0 / 61