递推方程:dp[i][j] = min{ dp[i][j - 1], dp[i - 1][j - 2] (if (n - j > 3 * (k - i)) 剩下的筷子能够凑满 };
用滚动数组。
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#define ll long long
#define ls rt<<1
#define rs ls1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define eps (1e-9)
#define type int
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define MOD 1000000007
#define inf 0x3f3f3f3f
#define pi acos(-1.0)
#define M 5000+5
int dp[2][M],a[M];
int inline dis(int i){
return (a[i]-a[i-1])*(a[i]-a[i-1]);
}
int main(){
int T,n,k,i,j,now,pre;
scanf("%d",&T);
while(T--){
scanf("%d %d",&k,&n);
for(i=1;i<=n;i++)scanf("%d",&a[i]);
k+=8;
memset(dp,0,sizeof(dp));
now=1;pre=0;
for(i=1;i<=k;i++){
dp[now][i*2-1]=-1;
for(j=2*i;j<=n;j++){
dp[now][j]=dp[now][j-1];
if(n-j>3*(k-i)&&dp[pre][j-2]!=-1&&(dp[now][j]==-1||dp[pre][j-2]+dis(j)<dp[now][j]))
dp[now][j]=dp[pre][j-2]+dis(j);
}
swap(now,pre);
}
printf("%d\n",dp[pre][n]);
}
return 0;
}