ZJU1234 Chopsticks - 动态规划

 题目描述：

有一个人发明了一种新的用筷子的方法，每次用三根筷子，筷子长度不一（A<=B<=C），规定三根筷子的“不合适度”用(A-B)^2来表示。现在有n根长度不一的筷子，问选出m副筷子的最小不合适度之和是多少。输入时按照筷子长度非降序输入。

分析：

这个动态规划的变形比较奇怪，三个筷子的长度，最长那根是没有用的，但是必须要有。

数组a[i]表示第i根筷子的长度，按照非升序排列，下标从1开始。用状态f[i][j]表示从前i根筷子选出j副，得到的最小不合适度。

可以证明，若用选第a[i]根筷子作为最短的，那么次短的筷子必然是a[i-1]。所以记bad(i)表示第i根筷子作为最短得到的不合适度。

显然，初始状态：

f[i][0]=0；f[3][1]=bad(3)。

状态转移方程 f[i][j] = min{ f[i-1][j], f[i-2][j-1]+bad(i) }。

要注意处理f[i][j]不合法的情况。

/*
ZJU1234 Chopsticks
*/

#include 
 
 
  
  
#include 
  
  
   
   

#define N 5005
#define K 1005
#define INF 99999999

#define clr(a) memset(a,0,sizeof(a))
#define MIN(a,b) ((a)>(b)?(b):(a))
#define POW(a) ((a)*(a))

int n,m;
int a[N];
int f[N][K];

int bad(int i){
    return POW(a[i]-a[i-1]);
}

int main()
{
    int i,j,k,T;
    int min;
    
    scanf("%d",&T);
    while(T--){
        //init
        clr(a);
        clr(f);
        min=INF;
        
        //input
        scanf("%d%d",&k,&n);
        m=k+8;
        for(i=n;i>=1;i--)
            scanf("%d",&a[i]);
        
        //DP
        f[3][1]=bad(3);
        for(i=4;i<=n;i++){
            k=MIN(i/3,m);
            for(j=1;j<=k;j++){
                if(j*3==i) f[i][j]=INF;
                else f[i][j]=f[i-1][j];
                f[i][j]=MIN(f[i][j],f[i-2][j-1]+bad(i));
                if(j==m) min=MIN(min,f[i][j]);
            }
        }
        
        //output
        printf("%d/n",min);
        
    }
    
    return 0;
}