MLE 20pts 求助

P2704 [NOI2001] 炮兵阵地

zhi_hui_kan_ti_jie @ 2024-06-20 12:29:16


#include<bits/stdc++.h>
using namespace std;
//#define int long long
const int maxn=1e6+10;
#define pii pair<int,int>
inline int max(int a,int b){return a>b?a:b;}
inline int min(int a,int b){return a<b?a:b;}
int mod=2017;
int dp[2][1025][1025];
int n,m;
inline int count1(int x)
{
    int ret=0;
    while(x)
    {
        if(x&1)ret++;
        x>>=1;
    }
    return ret;
}
inline bool check(int x)
{
    for(int i=0;i<=m-2;i++)
    {
        if((x&1) && (  ((x>>1)&1)  ||  ((x>>2)&1)))
        {
            return false;
        }
        x>>=1;
    }
    return true;
}
void solve()
{
  cin>>n>>m;
  vector<int>mp2(n+1);
  //memset(dp,0,sizeof(dp));
  vector<vector<pii>>state(2);
  for(int i=1;i<=n;i++)
  {
    string s;cin>>s;
    int ret=0;
    for(int j=0;j<m;j++)
    {
        if(s[j]=='P');
        else ret|=1;
        ret<<=1;
    }
    ret>>=1;
    mp2[i]=ret;
  }
  state[0].push_back({0,0});
  int ans=0;
  for(int i=1;i<=n;i++)
  {
    for(int st=0;st<(1<<m);st++)
    {
        if(st&mp2[i])continue;
        if(!check(st))continue;
        for(pii lst:state[(i&1)^1])
        {
            if(st&(lst.first|lst.second))continue;
            else
            {
                dp[i&1][lst.second][st]=max(dp[i&1][lst.second][st],dp[(i&1)^1][lst.first][lst.second]+count1(st));
                state[i&1].push_back({lst.second,st});
                ans=max(ans,dp[i&1][lst.second][st]);
            }
        }
    }
    for(int j=0;j<(1<<(m));j++)
    {
        for(int k=0;k<(1<<m);k++)
        {
            dp[(i&1)^1][j][k]=0;
        }
    }
    state[(i&1)^1].clear();
  }
   cout<<ans<<endl;
}
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    int t=1;
    //cin>>t;
    while(t--)
    {
        solve();
    }
    return 0;
}

|