POJ - 1185 주병 진지 상태 압축 DP

2952 단어
상태 압축 DP...쉬워요..한 줄에 최대 10개...1 대표로 포병을...0 은 놓지 않는다는...행당 최대 1024가지 상태...그리고 합법적으로..이렇게 되면...한 줄의 상태가 최대 60가지가 되는데...
현재 줄에 영향을 줄 수 있는 건 두 줄밖에 없는데..그래서 3차원 DP로...dp[t][x1][x2]....t는 어느 층을 대표하는지...x1은 윗층의 상태를 대표하는데...x2 현재 레이어의 상태...
    dp[t][x1][x2]= max ( dp[t-1][y][x1] + cnt[x2]) ...y는 t-2층 상태...cnt는 현재 층 상태의 포병 수, 즉 1의 수량...여기 제약조건이 보이는데...그냥 상태 x1, x2, y 충돌 불가...
Program:
#include<iostream>
#include<cmath>
#include<stack>
#include<queue>
#include<set>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#define ll long long
#define oo 1000000007
using namespace std;   
int n,m,a[105],anum;
char arc[105][12];
int cnt[65],dp[105][65][65];
bool ok1(int x)   //           
{
       int p,i;
       p=-10; 
       while (x)
       {
             i++;
             if (x%2)
             {
                   if (i-p<=2) return false;
                   p=i;
             }
             x/=2;
       }       
       return true;
}
bool ok2(int t,int x)  //               
{
       int i; 
       for (i=m;i>=1 && x;i--)
       {
             if (x%2 && arc[t][i]=='H') return false;
             x/=2;
       }
       return true;
}
bool ok3(int y,int x) //    y  ,  x  ,     
{
       while (y && x)
       {
             if (y%2 && x%2) return false;
             y/=2;  x/=2;
       }
       return true;
}
int main()
{       
       int i,t,j,x,ans;
       while (~scanf("%d%d",&n,&m))
       { 
               anum=0;
               a[0]=0;
               for (i=0;i<(1<<m);i++)
                  if (ok1(i)) 
                  {
                          a[++anum]=i;
                          cnt[anum]=0;
                          j=i;
                          while (j)
                          {
                                 j=j&(j-1);
                                 cnt[anum]++;
                          }
                  }
               for (i=0;i<=n;i++) gets(arc[i]+1);
               memset(dp,0,sizeof(dp));
               for (i=1;i<=anum;i++)
                 if (ok2(1,a[i]))  dp[1][0][i]=cnt[i];  
               for (t=2;t<=n;t++)
                  for (i=0;i<=anum;i++)
                    if (ok2(t,a[i]))
                      for (j=0;j<=anum;j++)
                        if (ok2(t-1,a[j]) && ok3(a[j],a[i]))
                          for (x=0;x<=anum;x++)
                            if (dp[t][j][i]<dp[t-1][x][j]+cnt[i] && ok3(a[x],a[i]))
                                dp[t][j][i]=dp[t-1][x][j]+cnt[i];
                         
               ans=0;
               for (i=0;i<=anum;i++)
                 for (j=0;j<=anum;j++)
                   if (ans<dp[n][i][j]) 
                       ans=dp[n][i][j];
               printf("%d
",ans); } return 0; }

좋은 웹페이지 즐겨찾기