POJ 1191 棋盘分割 （DP）

题目大意

• 输入一个8*8的矩阵，将其分解成n个矩阵，使各矩阵总分的均方差最小。
• 均方差，其中平均值，xi为第i块矩形棋盘的总分。

分析

• 将方差公式化简成s^2 = (x1^2+…+xn^2) / n - x^2，x为平均值
• dp[i][x1][y1][x2][y2]表示对于(x1,y1)(x2,y2)的矩阵，第i次分割后得到的平方和的最小值
• sum[x1][y1][x2][y2]表示矩形(x1,y1)(x2,y2)中每个方块的平方和
• 每次要么横向分割，要么竖向分割，且每次分割，必须一分到底，否则就得不到矩形。每次只能选择其中一个矩形继续分割，另一个将不能再分割。

• 状态转移方程：

  dp[i][x1][y1][x2][y2] = min{
   dp[i-1][x1][y1][a][y2] + sum[a+1][y1][x2][y2] , // 选上半部分继续分割
   dp[i-1][a][y1][x2][y2] + sum[x1][y1][a+1][y2] , //选下半部分继续分割
   dp[i-1][x1][y1][x2][b] + sum[b+1][y1][x2][y2] , //选左半部分继续分割
   dp[i-1][x1][b][x2][y2] + sum[x1][y1][x2][b+1] , //选右半部分继续分割
       }
  

代码

#include <iostream>
#include <cmath>
#include <iomanip>
#include <cstring>
#include <cstdio>

using namespace std;
#define INF 0x3fffffff //不能用0x7fffffff，因为tmp + sum 会超
#define REP(i , k , n) for(int i = (k); i < (n); i++)

const int maxn = 20;
const int N = 10;
/*dp[i][x1][y1][x2][y2]表示对于(x1,y1)(x2,y2)的矩阵，
 *第i次分割后得到的平方和的最小值
 */
int dp[maxn][N][N][N][N];
//sum[x1][y1][x2][y2]表示矩形(x1,y1)(x2,y2)中每个方块的平方和
int cb[N][N] , sum[N][N][N][N];


int DFS(int i , int x1 , int y1 , int x2 , int y2)
{
    if(dp[i][x1][y1][x2][y2] != -1) return dp[i][x1][y1][x2][y2];
    dp[i][x1][y1][x2][y2] = INF;
    int tmp1 , tmp2 , tmp;
    //横着切
    if(x2 > x1) for(int a = x1; a < x2; a++) {
        tmp1 = DFS(i-1 , x1 , y1 , a , y2);   //取上半部分
        tmp2 = DFS(i-1 , a+1 , y1 , x2 , y2); //取下半部分
        tmp = min(tmp1+sum[a+1][y1][x2][y2],tmp2+sum[x1][y1][a][y2]);
        dp[i][x1][y1][x2][y2] = min(dp[i][x1][y1][x2][y2] , tmp);
    }
    //竖着切
    if(y2 > y1) for(int b = y1; b < y2; b++) {
        tmp1 = DFS(i-1 , x1 , y1 , x2 , b); //取左边部分
        tmp2 = DFS(i-1 , x1 , b+1 , x2 , y2); //取右边部分
        tmp = min(tmp1+sum[x1][b+1][x2][y2],tmp2+sum[x1][y1][x2][b]);
        dp[i][x1][y1][x2][y2] = min(dp[i][x1][y1][x2][y2] , tmp);
    }
    return dp[i][x1][y1][x2][y2];
}
/* 每次只能横着切或者竖着切，且每次只能选择一部分作为可以再切割的
 * 另一部分将不能再切割
 * dp[i][x1][y1][x2][y2] = min{
 * dp[i-1][x1][y1][a][y2] + sum[a+1][y1][x2][y2] , 选上半部分
 * dp[i-1][a][y1][x2][y2] + sum[x1][y1][a+1][y2] , 选下半部分
 * dp[i-1][x1][y1][x2][b] + sum[b+1][y1][x2][y2] , 选左半部分
 * dp[i-1][x1][b][x2][y2] + sum[x1][y1][x2][b+1] , 选右半部分
 */
int main()
{
    int n;
    while(cin >> n)
    {
        int num = 0;
        REP(i , 0 , 8) REP(j , 0 , 8) {
            cin >> cb[i][j];
            num += cb[i][j];
        }
        memset(dp , -1 , sizeof(dp));

        REP(x1,0,8) REP(x2,x1,8) REP(y1,0,8) REP(y2,y1,8)
        {
            int tmp = 0;
            REP(i,x1,x2+1) REP(j,y1,y2+1) tmp += cb[i][j];
            dp[0][x1][y1][x2][y2] = sum[x1][y1][x2][y2] = tmp * tmp;
        }

        double rel = DFS(n-1 , 0 , 0 , 7 , 7) , avg = (double)num / n;
        rel = rel / n - avg * avg;
        cout << fixed << setprecision(3) << sqrt(rel) << endl;
    }
    return 0;
}