HDU - 5492 Find a path（方差公式+dp）

Find a path

Frog fell into a maze. This maze is a rectangle containing NN rows and MM columns. Each grid in this maze contains a number, which is called the magic value. Frog now stays at grid (1, 1), and he wants to go to grid (N, M). For each step, he can go to either the grid right to his current location or the grid below his location. Formally, he can move from grid (x, y) to (x + 1, y) or (x, y +1), if the grid he wants to go exists. 
Frog is a perfectionist, so he'd like to find the most beautiful path. He defines the beauty of a path in the following way. Let’s denote the magic values along a path from (1, 1) to (n, m) as A1,A2,…AN+M−1A1,A2,…AN+M−1, and AavgAavg is the average value of all AiAi. The beauty of the path is (N+M–1)(N+M–1) multiplies the variance of the values:(N+M−1)∑N+M−1i=1(Ai−Aavg)2(N+M−1)∑i=1N+M−1(Ai−Aavg)2 
In Frog's opinion, the smaller, the better. A path with smaller beauty value is more beautiful. He asks you to help him find the most beautiful path. 

InputThe first line of input contains a number TT indicating the number of test cases (T≤50T≤50). 
Each test case starts with a line containing two integers NN and MM (1≤N,M≤301≤N,M≤30). Each of the next NN lines contains MM non-negative integers, indicating the magic values. The magic values are no greater than 30. 
OutputFor each test case, output a single line consisting of “Case #X: Y”. XX is the test case number starting from 1. YY is the minimum beauty value.Sample Input

1
2 2
1 2
3 4

Sample Output

Case #1: 14

将公式变形得：（n+m-1）*ΣAi^2-(ΣAi)^2
dp求出每种和的最小的平方和，最后找出满足公式的最小解。

#include<bits/stdc++.h>
#define MAX 31
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

int a[MAX][MAX];
int dp[MAX][MAX][];

int main()
{
    int tt=,t,n,m,i,j,k;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(i=;i<=n;i++){
            for(j=;j<=m;j++){
                scanf("%d",&a[i][j]);
            }
        }
        memset(dp,INF,sizeof(dp));
        dp[][][]=;dp[][][]=;
        for(i=;i<=n;i++){
            for(j=;j<=m;j++){
                for(k=;k<=;k++){
                    if(k+a[i][j]<=) dp[i][j][k+a[i][j]]=min(dp[i][j][k+a[i][j]],min(dp[i-][j][k],dp[i][j-][k])+a[i][j]*a[i][j]);
                }
            }
        }
        int ans=INF;
        for(i=;i<=;i++){
            if(dp[n][m][i]!=INF){
                ans=min(ans,(n+m-)*dp[n][m][i]-i*i);
            }
        }
        printf("Case #%d: %d\n",++tt,ans);
    }
    return ;
}