제출 #228372

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
228372		FieryPhoenix	Linear Garden (IOI08_linear_garden)	C++11	80 / 100	155 ms	65540 KiB

linear_garden

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

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
typedef long long ll;
typedef long double ld;
#define INF 2001001001
#define MOD 1000000007

int N,M;
string S;
ll dp[1000001][3][3][2];//index, max(L-P), max(P-L), 0 for L

int main()
{
  //ios_base::sync_with_stdio(0);cin.tie(0);
  //freopen (".in","r",stdin);
  //freopen (".out","w",stdout);
  cin>>N>>M;
  cin>>S;
  dp[N][0][0][0]=1;
  for (int i=N-1;i>=0;i--){
    //adding L
    for (int j=0;j<2;j++)
      for (int k=0;k<=2;k++)
	for (int l=0;l<2;l++){
	  dp[i][j+1][max(0,k-1)][0]+=dp[i+1][j][k][l];
	  dp[i][j+1][max(0,k-1)][0]%=M;
	}
    //adding P
    for (int j=0;j<=2;j++)
      for (int k=0;k<2;k++)
	for (int l=0;l<2;l++){
	  dp[i][max(0,j-1)][k+1][1]+=dp[i+1][j][k][l];
	  dp[i][max(0,j-1)][k+1][1]%=M;
	}
  }
  int L_P=0,P_L=0;
  int ans=0;
  for (int i=0;i<N;i++){
    //cout<<"I: "<<i<<endl;
    if (S[i]=='P'){
      for (int j=0;j<=2;j++)
	for (int k=0;k<=2;k++)
	  if (j+L_P<=2 && k+P_L<=2){
	    ans+=dp[i][j][k][0];
	    //cout<<"add "<<dp[i][j][k][0]<<endl;
	    ans%=M;
	  }
    }
    if (S[i]=='L'){
      L_P++;
      P_L=max(0,P_L-1);
    }
    else{
      P_L++;
      L_P=max(0,L_P-1);
    }
  }
  cout<<ans+1<<endl;
  return 0;
}

• 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: 0 / 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