Submission #262861

#	Time	Username	Problem	Language	Result	Execution time	Memory
262861		ElyesChaabouni	Linear Garden (IOI08_linear_garden)	C++14	85 / 100	109 ms	65540 KiB

linear_garden

/*#pragma GCC optimize("O3")*/
#include<bits/stdc++.h>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp> 
//#define ordered_set tree<int, null_type,less<int >, rb_tree_tag,tree_order_statistics_node_update> 
#define eps 1e-9
#define MOD1 998244353
#define MOD2 1000000007
#define INV_10 299473306
#define INF 1000000000
#define PI 3.14159265358979323846
using namespace std;
int dp[1000000][9][2];
bool a[1000000];
int n, m, pp, ll;
int solve(int idx, int d, bool b)
{
	if(idx==n)
		return 1;
	if(dp[idx][d][b]!=-1)
		return dp[idx][d][b];
	dp[idx][d][b]=0;
	if(a[idx])
	{
		if(d/3 < 2)
		{
			pp=d/3;
			ll=d%3;
			pp++;
			ll--;
			ll=max(ll, 0);
			dp[idx][d][b]+=solve(idx+1, pp*3+ll, b);
			dp[idx][d][b]%=m;
		}
		if(d%3 < 2)
		{
			pp=d/3;
			ll=d%3;
			pp--;
			ll++;
			pp=max(0, pp);
			dp[idx][d][b]+=solve(idx+1, pp*3+ll, 1);
			dp[idx][d][b]%=m;
		}
	}
	else
	{
		if(d%3 < 2)
		{
			pp=d/3;
			ll=d%3;
			pp--;
			ll++;
			pp=max(0, pp);
			dp[idx][d][b]+=solve(idx+1, pp*3+ll, b);
			dp[idx][d][b]%=m;
		}
		if(d/3 < 2 && b)
		{
			pp=d/3;
			ll=d%3;
			pp++;
			ll--;
			ll=max(ll, 0);
			dp[idx][d][b]+=solve(idx+1, pp*3+ll, b);
			dp[idx][d][b]%=m;
		}
	}
	return dp[idx][d][b];
}
int main()
{
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin >> n >> m;
	char c;
	for(int i = 0; i < n; i++)
	{
		cin >> c;
		if(c=='P')
			a[i]=1;
	}
	for(int i = 0; i <= n; i++)
			for(int k = 0; k < 9; k++)
				for(int l = 0; l < 3; l++)
					dp[i][k][l]=-1;
	cout << solve(0, 0, 0) << '\n';
}
//size

Compilation message (stderr)

linear_garden.cpp: In function 'int main()':
linear_garden.cpp:85:17: warning: iteration 2 invokes undefined behavior [-Waggressive-loop-optimizations]
   85 |      dp[i][k][l]=-1;
      |      ~~~~~~~~~~~^~~
linear_garden.cpp:84:22: note: within this loop
   84 |     for(int l = 0; l < 3; l++)
      |                    ~~^~~
linear_garden.cpp:85:16: warning: array subscript 2 is above array bounds of 'int [2]' [-Warray-bounds]
   85 |      dp[i][k][l]=-1;
      |      ~~~~~~~~~~^

• Subtask 1: 2 / 2
• Subtask 2: 2 / 2
• Subtask 3: 4 / 4
• Subtask 4: 4 / 4
• Subtask 5: 4 / 4
• Subtask 6: 4 / 4
• Subtask 7: 4 / 4
• Subtask 8: 4 / 4
• Subtask 9: 4 / 4
• Subtask 10: 4 / 4
• Subtask 11: 4 / 4
• Subtask 12: 5 / 5
• Subtask 13: 5 / 5
• Subtask 14: 5 / 5
• Subtask 15: 5 / 5
• Subtask 16: 5 / 5
• Subtask 17: 5 / 5
• Subtask 18: 5 / 5
• Subtask 19: 5 / 5
• Subtask 20: 5 / 5
• Subtask 21: 0 / 5
• Subtask 22: 0 / 5
• Subtask 23: 0 / 5