UVa 10271 - Chopsticks(DP)

给出n只筷子,找出 k + 8 组,每组三根,输出每组最短两根长度差的平方和的最小值。
倒序读入,对每个可以选的点位有两种转移状态。
DP数组滚动使用,节约空间和初始化时间。
状态转移方程: d [ i ] [ j ] = m i n ( d [ i − 1 ] [ j ] , d [ i − 2 ] [ j − 1 ] + ( a [ i ] − a [ i − 1 ] ) 2 )

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=5010;
int a[maxn],d[3][maxn];
int main(){
int t;scanf("%d",&t);
while(t--){
int k,n;scanf("%d%d",&k,&n);
k+=8;
memset(d,0x3f,sizeof(d));
for(int i=n;i>=1;--i)
scanf("%d",&a[i]);
d[0][0]=d[1][0]=d[2][0]=0;
for(int i=3;i<=n;++i)
for(int j=1;j*3<=i;++j)
d[i%3][j]=min(d[(i-1)%3][j],d[(i-2)%3][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
printf("%d\n",d[n%3][k]);
}
return 0;
}

** 本文迁移自我的CSDN博客,格式可能有所偏差。 **