HDU-4362 Dragon Ball DP+优化

这题如果采用普通的DP方程的话果断TLE。所以需要对DP方程进行优化。

由于这里龙珠可以随意选取，所以龙珠的编号也就没有了什么意义了，所以直接先对龙珠进行排序，我们只要保证其位置和花费同步排序就可以了。

接下来就是优化了，这个部分可以见代码，在上一题中使用的同步滑动指针，这里则不然，需要根据数值来异步滑动指针。

代码如下：

#include <cstdlib>

#include <cstring>

#include <cstdio>

#include <algorithm>

#define INF 0x7f7f7f7f

using namespace std;



int N, M, x, dp[55][1005];

// dp[i][j] 代表第i时刻选择j号物品的最小值 

struct Node

{

    int pos, ti;

    bool operator < (Node temp) const

    {

        return pos < temp.pos;    

    }

}e[55][1005];



// dp[i][j] 表示i时刻拿j个龙珠的最少花费

// 完整的dp方程是 dp[i][j] = min( dp[i-1][k] - abs(loc[i][j] - loc[i-1][k]) + ti[i][j] )

// 当 loc[i][j] < loc[i-1][k]

// 方程变为 dp[i][j] = min( dp[i-1][k] + loc[i][j] - loc[i-1][k] + ti[i][j])

// 显然此时只要求得 dp[i-1][k] - loc[i-1][k] 的最小值来递推便可了

// 同理当 loc[i][j] > loc[i-1][k]

// 方程变为 dp[i][j] = min( dp[i-1][k] - loc[i][j] + loc[i-1][k] + ti[i][j])

// 显然此时只要求得 dp[i-1][k] + loc[i-1][k] 的最小值来递推便可了



void read()

{

    for (int i = 1; i <= M; ++i) {

        for (int j = 1; j <= N; ++j) {

            scanf("%d", &e[i][j].pos);

        }

    }

    for (int i = 1; i <= M; ++i) {

        for (int j = 1; j <= N; ++j) {

            scanf("%d", &e[i][j].ti);

        }

        sort(e[i]+1, e[i]+N+1);  // 对所有的坐标进行排序

    }

}



void DP()

{

    int Min, k;

    memset(dp, 0x7f, sizeof (dp));

    for (int i = 1; i <= N; ++i) {

        dp[0][i] = 0;

        e[0][i].pos = x;

    }

    for (int i = 1; i <= M; ++i) {

        k = -1;

        Min = INF; // 假设是从该点的左边的递推过来

        // 由于给定的点不一定满足 ptr < j，那么前面的点的坐标一定小于后面的点 

        // 所以要进行一次排序，来使得其满足上述的性质，这样才能够使得dp[i][j]能够取到最优值

        for (int j = 1, ptr = 1; j <= N; ++j) {

            while (ptr <= N && e[i-1][ptr].pos <= e[i][j].pos) {

                if (Min > dp[i-1][ptr] - e[i-1][ptr].pos) {

                    Min = dp[i-1][ptr] - e[i-1][ptr].pos;

                    k = j;

                }

                ++ptr;

            }

            if (k != -1) {

                dp[i][j] = Min + e[i][j].pos + e[i][j].ti;

            }

        }

        k = -1;

        Min = INF;

        for (int j = N, ptr = N; j >= 1; --j) {

            while (ptr >= 1 && e[i-1][ptr].pos > e[i][j].pos) {

                if (Min > dp[i-1][ptr] + e[i-1][ptr].pos) { 

                    Min = dp[i-1][ptr] + e[i-1][ptr].pos;

                    k = j;

                }

                --ptr;

            }

            if (k != -1) {

                dp[i][j] = min(dp[i][j], Min - e[i][j].pos + e[i][j].ti);

            }

        }

    }

    Min = INF;

    for (int i = 1; i <= N; ++i) {

        Min = min(Min, dp[M][i]);

    }

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

}



int main()

{

    int T;

    scanf("%d", &T);

    while (T--) {

        scanf("%d %d %d", &M, &N, &x);

        read();

        DP();

    }

    return 0;

}