poj 1191 棋盘分割 记忆化搜索

纠结了一天，奇葩的代码，dp初始化的时候，用memset(dp, 0x3f, sizeof(dp)),记忆化搜索判断用 if( dp[tim][x1][y1][x2][y2] == 0x3f3f3f3f),TLE了整整一晚上，后来随便换成用-1做判断条件就A掉了，还0ms……又去试了下是不是poj里面dp[tim][x1][y1][x2][y2] == 0x3f3f3f3f会出问题，用a+b代码试了下，完全没问题啊……

还是不知道到底哪挫了…………………………

 

白天写的时候，没有化简方差公式，用一个area存每一块的面积，用一个th数组存储dfs累加的 (xi - avg)*(xi - avg) 的最小值，不知道哪有问题，写了好多遍也不对，后来看黑书上，按化简的方差公式写，过样例，然后就开始上面的 endless TLE …………

 

 

很简单的记忆化搜索

for( k= x1; k<x2; k++){
        mm =min( mm, dfs( tim-1, x1, y1, k, y2) + get_var( k+1, y1, x2, y2));
        mm =min( mm, dfs( tim-1, k+1, y1, x2, y2) + get_var( x1, y1, k, y2));
    }
    for( k= y1; k<y2; k++){
        mm =min( mm, dfs( tim-1, x1, y1, x2, k) + get_var( x1, k+1, x2, y2));
        mm =min( mm, dfs( tim-1, x1, k+1, x2, y2) + get_var( x1, y1, x2, k));
    }

dp[tim][x1][y1][x2][y2] = mm;

把目前的这块（范围 x1, y1, x2, y2 ) 分为两部分， 一部分算分出去的，计算其大小，供最后处理方差，另一部分dfs，继续进行分割

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
const int INF= 0x3f3f3f3f;
int dp[16][9][9][9][9], n, mit[9][9], ans[9][9], area[9][9][9][9];
int get_var(int x1, int y1, int x2, int y2){
    if( area[x1][y1][x2][y2] != -1) return area[x1][y1][x2][y2];
    int tmp = ans[x2][y2] - ans[x1-1][y2] - ans[x2][y1-1] + ans[x1-1][y1-1];
    return area[x1][y1][x2][y2] = tmp*tmp;
}
int dfs(int tim, int x1, int y1, int x2, int y2){
    if( dp[tim][x1][y1][x2][y2] != -1) return dp[tim][x1][y1][x2][y2] ;
    if( tim == 0){
        dp[tim][x1][y1][x2][y2] = get_var( x1, y1, x2, y2);
        return dp[tim][x1][y1][x2][y2] ;
    }
    int k, s, s1, s2, mm= INF;
    for( k= x1; k<x2; k++){
        mm =min( mm, dfs( tim-1, x1, y1, k, y2) + get_var( k+1, y1, x2, y2));
        mm =min( mm, dfs( tim-1, k+1, y1, x2, y2) + get_var( x1, y1, k, y2));
    }
    for( k= y1; k<y2; k++){
        mm =min( mm, dfs( tim-1, x1, y1, x2, k) + get_var( x1, k+1, x2, y2));
        mm =min( mm, dfs( tim-1, x1, k+1, x2, y2) + get_var( x1, y1, x2, k));
    }
    return ( dp[tim][x1][y1][x2][y2] = mm);
}
int main(){
  //  freopen("1.txt", "r", stdin);
    int i, j, k;
    int aa[2][2][2][2][2];
    double avg, ss;
    while( scanf("%d", &n) != EOF){
        memset( ans, 0, sizeof(ans));
        memset( area, -1, sizeof( area));
        for( i= 1; i< 9; i++){
            for( j= 1; j< 9; j++){
                scanf("%d", &mit[i][j]);
                ans[i][j]= ans[i-1][j] + ans[i][j-1] + mit[i][j] - ans[i-1][j-1];
            }
        }
        avg= ans[8][8]*1.0/n;
        memset(dp, -1, sizeof( dp));
        dfs(n-1, 1, 1, 8, 8);
        memset(aa, INF, sizeof( INF));
        ss= sqrt(fabs(dp[n-1][1][1][8][8]*1.0/n - avg*avg));
        printf("%.3f\n", ss);
    }
	return 0;
}