UVa 10271 - Chopsticks（DP）

发表于 分类于 ， Disqus：

给出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博客，格式可能有所偏差。 **

相关文章