VJ1061迎春舞会之三人组舞

题目链接

推了半个下午。。写的三重超时了  加了点单调队列的思想 优化了下过了

dp[i][j] 第二组的最右边的人选第J个人 那最左边肯定选第j-1个人 肯定是选相邻的 

dp[i][j] = min(o,dp[i-1][j-2]+(h[j]-h[j-1])*(h[j]-h[j-1])) 加个变量o保存 到j为止 最优的一个位置 j-2得满足条件 

#include <iostream>
#include<cstdio>
#include<cstring>
#include<stdlib.h>
#define N 5010
#define M 3010
#define INF 0xfffffff
using namespace std;
int dp[M][N],h[N];
int main()
{
    int i,j,n,m,g;
    scanf("%d%d",&m,&n);
    for(i = 1; i <= n ;i++)
    scanf("%d",&h[i]);
    for(i = 1; i <= m ;i++)
        for(j = 1; j <= n ;j++)
        dp[i][j] = INF;
    int o = INF;
    for(i = 2 ; i < n ;i++)
    {
        dp[1][i] = min(o,(h[i]-h[i-1])*(h[i]-h[i-1]));
        o = min(dp[1][i],o);
    }
    for(i = 2 ; i <= m ; i++)
    {
        int o = INF;
        for(j = 2*i ; j < n-(m-i)*3 ;j++)
        {
            if(j-2<(n-(m-i+1)*3))
            dp[i][j] = min(o,dp[i-1][j-2]+(h[j]-h[j-1])*(h[j]-h[j-1]));
            else
            dp[i][j] = min(o,dp[i-1][j-3]+(h[j]-h[j-1])*(h[j]-h[j-1]));
            o = min(o,dp[i][j]);
        }
    }
    printf("%d\n",dp[m][n-1]);
    return 0;
}