HDU1565+状态压缩dp

简单的压缩状态

dp

/*
状态压缩dp
同hdu2167
利用滚动数组！！
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<stack>
#include<math.h>
#include<map>
using namespace std;
const int maxn = 20;
int binary[ maxn+5 ];
int s[ 1<<(maxn) ];
//int sum[ maxn ][ 1<<maxn ];
//int dp[ maxn ][ 1<<maxn ];
int dp[2][1<<maxn];
int mat[ maxn ][ maxn ];
void init_binary(){
    binary[0] = 1;
    for( int i=1;i<=20;i++ ){
        binary[i] = 2*binary[i-1];
    }
}

int solve( int n ){
    //memset( sum,0,sizeof( sum ) );
    memset( dp,0,sizeof( dp ) );
    int cnt = 0;
    int N = (1<<n);
    for( int i=0;i<N;i++ ){
        if( ( (i<<1)&i )==0 ){
            s[ cnt++ ] = i;
        }
    }//保存合格的状态
    //for( int i=0;i<n;i++ ){
        //for( int j=0;j<cnt;j++ ){
            //for( int k=0;k<n;k++ ){
                //if( binary[k]&s[j] ){
                    //sum[ i ][ j ] += mat[ i ][ k ];
                //}
            //}
        //}
    //}
    int ans = 0;
    for( int i=0;i<cnt;i++ ){
        int sum = 0;
        for( int j=0;j<n;j++ ){
            if( binary[j]&s[i] ){
                sum += mat[0][j];
            }
        }
        dp[0][i] = sum;
        ans = max( ans,dp[0][i] );
    }
    for( int i=1;i<n;i++ ){
        for( int j=0;j<cnt;j++ ){
            for( int k=0;k<cnt;k++ ){
                if( ( s[j]&s[k] )==0 ){
                    int sum = 0;
                    for( int kk=0;kk<n;kk++ ){
                        if( binary[kk]&s[j] ){
                            sum += mat[i][kk];
                        }
                    }
                    if( i%2==1 ) dp[ 1 ][ j ] = max( dp[1][j],dp[0][k]+sum );
                    else dp[ 0 ][ j ] = max( dp[0][j],dp[1][k]+sum );
                    //dp[ i ][ j ] = max( dp[i][j],dp[i-1][k]+sum );
                    //dp[i][j] = max( dp[i][j],dp[i-1][k]+sum[i][j] );
                }
            }
            ans = max( ans,max( dp[1][j],dp[0][j] ) );
        }
    }
    return ans;
}  

int main(){
    int n;
    init_binary();
    while( scanf("%d",&n)==1 ){
        for( int i=0;i<n;i++ ){
            for( int j=0;j<n;j++ ){
                scanf("%d",&mat[i][j]);
            }
        }
        int ans = solve( n );
        printf("%d\n",ans);
    }
    return 0;
}