Submission #2403326

---

Source Code Expand

#include<stdio.h>
#include<vector>
#include<algorithm>
#include<string>
#include<iostream>
#include<set>
using namespace std;
typedef long long ll;
ll mod = 998244353;
ll dp[303030][3][3][2];
set<string>se;
void dfs(string s)
{
	if (se.find(s) != se.end())return;
	se.insert(s);
	for (int i = 0; i < s.size() - 1; i++)
	{
		if (s[i] != s[i + 1])
		{
			string z = s;
			z[i] = z[i + 1] = int('a') + int('b') + int('c') - s[i] - s[i + 1];
			dfs(z);
		}
	}
}
int main()
{
	string s;
	cin >> s;
	se.clear();
	for (int i = 0; i <= s.size(); i++)for (int j = 0; j < 3; j++)for (int k = 0; k < 3; k++)for (int l = 0; l < 2; l++)dp[i][j][k][l] = 0;
	if (s.size() <= 3)
	{
		dfs(s);
		printf("%d\n", se.size());
	}
	else
	{
		bool f1 = false, f2 = false;
		int cnt = 0;
		for (int i = 0; i < s.size(); i++)cnt += s[i] - 'a';
		for (int i = 0; i < s.size() - 1; i++)
		{
			if (s[i] != s[i + 1])f1 = true;
			else f2 = true;
		}
		cnt %= 3;
		if (!f1)printf("1\n");
		else
		{
			dp[1][0][0][0] = dp[1][1][1][0] = dp[1][2][2][0] = 1;
			for (int i = 1; i < s.size(); i++)
			{
				for (int j = 0; j < 3; j++)
				{
					for (int k = 0; k < 3; k++)
					{
						for (int l = 0; l < 3; l++)
						{
							if (k == l)
							{
								dp[i + 1][(j + l) % 3][l][1] += dp[i][j][k][0] + dp[i][j][k][1];
							}
							else
							{
								dp[i + 1][(j + l) % 3][l][0] += dp[i][j][k][0];
								dp[i + 1][(j + l) % 3][l][1] += dp[i][j][k][1];
							}
						}
					}
				}
				for (int j = 0; j < 3; j++)for (int k = 0; k < 3; k++)for (int l = 0; l < 2; l++)dp[i + 1][j][k][l] %= mod;
			}
			ll ans = 0;
			for (int i = 0; i < 3; i++)ans += dp[s.size()][cnt][i][1];
			if (!f2)ans++;
			printf("%lld\n", ans%mod);
		}
	}
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:35:27: warning: format ‘%d’ expects argument of type ‘int’, but argument 2 has type ‘std::set<std::basic_string<char> >::size_type {aka long unsigned int}’ [-Wformat=]
   printf("%d\n", se.size());
                           ^

Judge Result