【洛谷2704】炮兵阵地(状压dp)

题目:

我是超链接

题解:

经典的状压dp,一开始的自己傻傻很naive,虽然预处理了高地和左右,但还是遍历了所有的可能。

那么直接预处理出满足左右和高地的排列,就快多了

代码:

#include 
#include 
#include 
using namespace std;
int s[1030],sum[1030],pl[120],f[105][105][105],n,m,man;
void dfs(int now,int sl,int num,int l1,int l2)
{
	if (now>m)
	{
		s[++man]=num;
		sum[man]=sl;
		return;
	}
	for (int i=0;i<=1;i++)
	  if (i<=1-l1 && i<=1-l2)//左右差两个 
	    dfs(now+1,sl+i,num+i*(1<>1),l2,i);
}
int main()
{
	int i,j,k,ii;char c;
	scanf("%d%d",&n,&m);
	for (i=1;i<=n;i++)
	  {
	  	c=getchar();
	  	while (c!='P' && c!='H') c=getchar();
	  	int hh=1; 	
	  	while (c=='P' || c=='H')
	  	{
	  		if (c=='H')
	  		  pl[i]+=hh;
	  		hh<<=1;
	  		c=getchar();
		}
	  } 
	dfs(1,0,0,0,0);
	int ans=0;
	for (i=1;i<=n;i++)
	  for (j=1;j<=man;j++)
	    {
	    	if (s[j]&pl[i]) continue;
	    	for (k=1;k<=man;k++)
	    	{
	    		if ((s[k]&pl[i-1])||(s[j]&s[k]))continue;
	    	  for (ii=1;ii<=man;ii++)
	    	  {
	    	  	if ((s[ii]&pl[i-2])||(s[ii]&s[k])||(s[ii]&s[j])) continue;
	    	  	f[i][j][k]=max(f[i][j][k],f[i-1][k][ii]+sum[j]);
	    	  	ans=max(ans,f[i][j][k]);
			  }
			}
	    	  
		}
    printf("%d",ans);
}


你可能感兴趣的:(dp)