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;
}