【HDU】4539 郑厂长系列故事——排兵布阵

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others) Total Submission(s): 3527    Accepted Submission(s): 1219   Problem Description 　　郑厂长不是正厂长 　　也不是副厂长 　　他根本就不是厂长 　　事实上 　　他是带兵打仗的团长 　　一天，郑厂长带着他的军队来到了一个n*m的平原准备布阵。 　　根据以往的战斗经验，每个士兵可以攻击到并且只能攻击到与之曼哈顿距离为2的位置以及士兵本身所在的位置。当然，一个士兵不能站在另外一个士兵所能攻击到的位置，同时因为地形的原因平原上也不是每一个位置都可以安排士兵。 　　现在，已知n,m 以及平原阵地的具体地形，请你帮助郑厂长计算该阵地,最多能安排多少个士兵。     Input 输入包含多组测试数据； 每组数据的第一行包含2个整数n和m (n <= 100, m <= 10 )，之间用空格隔开； 接下来的n行，每行m个数，表示n*m的矩形阵地，其中1表示该位置可以安排士兵，0表示该地形不允许安排士兵。     Output 请为每组数据计算并输出最多能安排的士兵数量，每组数据输出一行。     Sample Input   6 6 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0     Sample Output   2     Source 2013腾讯编程马拉松复赛第二场（3月30日）     Recommend liuyiding

 

这个的话，其实和炮兵布阵是一样的了，

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
int dp[2][200][200] ;  
int cannot[101] ;
int N , M ;

int ok(int x){    //同行
    if(x & (x<<2))
        return 0 ;
    if(x & (x>>2))
        return 0 ;
    return 1 ;
}

int Count(int x){ 
    int sum = 0 ;
    while(x){
        if(x&1)
            sum++ ;
        x>>=1 ;
    }
    return sum ;
}


void get_all_states(vector<int> &states , vector<int> &people){
     int i ;
     states.clear() ;
     people.clear() ;
     for(i = 0 ; i < (1<<M) ; i++){
         if(ok(i)){
            states.push_back(i) ;
            people.push_back(Count(i)) ;
         }
     }
}

int DP(){
    int i , j , k ,row , sum = 0 ;
    vector<int> states ;
    vector<int> people ;
    get_all_states(states , people) ;
    memset(dp , -1 , sizeof(dp)) ;
    for(i = 0 ; i < states.size() ; i++){
        if(states[i] & cannot[1])
            continue ;
        dp[1][i][0] = people[i];
    }
    for(row = 2 ; row <= N ; row++){
        for(i = 0 ; i < states.size() ; i++){
            if(states[i] & cannot[row])
                continue ;
            for(j = 0 ; j < states.size() ; j++){
                if(states[j] & cannot[row-1])
                    continue ;
                if(states[i] & (states[j]<<1))
                    continue ;
                if(states[i] & (states[j]>>1))
                    continue ;
                for(k = 0 ; k < states.size() ; k++){
                    if(states[k] & cannot[row-2])
                       continue ;
                    if(states[i] & states[k])
                       continue ;
                    if(states[j] & (states[k]<<1))
                       continue ;
                    if(states[j] & (states[k]>>1))
                       continue ;
                    if(dp[(row-1)&1][j][k] == -1)
                       continue ;
                    dp[row&1][i][j] = max(dp[row&1][i][j] ,
                                          dp[(row-1)&1][j][k] + people[i]) ;
                }
            }
        }
    }
    for(i = 0 ; i < states.size() ; i++)
        for(j = 0 ; j < states.size() ; j++)
            sum = max(sum , dp[N&1][i][j]) ;
    return sum ;
}

int main(){
    int i , j , x ;
    while(cin>>N>>M){
         memset(cannot , 0 , sizeof(cannot)) ;
         for(i = 1 ; i <= N ; i++){
            for(j = 0 ; j < M ; j++){
                cin>>x ;
                if(!x)
                  cannot[i] |= (1<<j) ;
            }
         }
         cout<<DP()<<endl ;
    }
    return 0 ;
}