广工校赛——DP——悦动达人

Description

一个游戏，在屏幕上有5个格子形成一行，每一秒都会有一个格子闪烁，格子闪烁时你需要保证至少有一只手指在格子上面， 现在我们已经知道第i秒时，第xi个格子会闪烁，我们假设手指的移动不花费时间，你现在用两根手指玩这个游戏， 设初始两根手指都在0处位置，算出n秒过后手指需要移动的最小距离。（允许手指交叉）

注：手指移动的距离的计算是，假设你的一根从x，移动到y格，那么移动的距离是|x-y|

Input

第一行一个数T，表示有T组测试数据（T<=50） 第二行，n，表示进行n秒(1<=n<=10^4) 下一行 n个数，xi（0<=xi<=4）

Output

输出n秒过后手指需要移动的最小距离.

Sample Input

1 2 0 2

Sample Output

2

HINT

大意：两根手指总共走n次，每次走i-num[k]次，下标弄成num[i]是为了记录状态。最后遍历找走了n步之后的最小值，很经典的DP

PS：abs和fabs区别，abs头文件stdlib fabs头文件math，abs返回与原来参数相同的绝对值，fabs返回浮点型的绝对值。

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<cstdlib>

using namespace std;

const int MAX = 10010;

int dp[5][5][MAX],num[MAX];

int main()

{

    int T;

    scanf("%d",&T);

    while(T--){

            int n;

            scanf("%d",&n);

        for(int i = 1; i <= n ; i++)

        scanf("%d",&num[i]);

        memset(dp,-1,sizeof(dp));

        dp[0][0][0] = 0;

        for(int k = 1; k <= n ; k++){

            for(int i = 0 ; i <= 4; i++){

                for(int j = 0; j <= 4; j++){

           if(dp[i][j][k-1] == -1) continue;

                int ans = abs(i-num[k]);

                if(dp[num[k]][j][k] == -1) dp[num[k]][j][k] = dp[i][j][k-1]+ans;

                else dp[num[k]][j][k] = min(dp[num[k]][j][k],dp[i][j][k-1]+ans);

                ans = abs(j-num[k]);

                if(dp[i][num[j]][k] == -1) dp[i][num[j]][k] = dp[i][j][k-1]+ans;

                else dp[i][num[j]][k] = min(dp[i][num[j]][k],dp[i][j][k-1]+ans);

                }

            }

        }

        int min1 = 99999999;

        for(int i = 0; i <= 4; i++){

            for(int j = 0; j <= 4; j++){

                if(dp[i][j][n]!=-1&&dp[i][j][n] < min1)

                min1 = dp[i][j][n];

            }

        }

        printf("%d\n",min1);

    }

    return 0;

}

View Code