NYOJ 1365 河南省第十一届ACM省赛 山区修路

NYOJ 1365 河南省第十一届ACM省赛 山区修路

题目描述：

SNJ位于HB省西部一片群峰耸立的高大山地，横亘于A江、B水之间，方圆数千平方公里，相传上古的神医在此搭架上山采药而得名。景区山峰均在海拔3000米以上，堪称"华中屋脊"。SNJ是以秀绿的亚高山自然风光，多样的动植物种，人与自然和谐共存为主题的森林生态区。
SNJ处于中国地势第二阶梯的东部边缘，由大巴山脉东延的余脉组成中高山地貌，区内山体高大，高低不平。 交通十分不便。
最近，HB省决定修一条从YC市通往SNJ风景区的高速公路。经过勘测分析，途中需要经过高度分别为H1，H2，……,Hn的N个山区。由于高低不平，除正常的修路开支外，每段还要多出高度差|Hi - Hi-1|*X万元的斜坡费用。Dr. Kong 决定通过填高一些区域的高度来降低总的费用。当然填高也是需要一些费用的。每填高Y单位，需要付出Y2万元费用。
你能否帮Dr. Kong做出一个规划，通过部分填高工程改造，使得总的费用降下来。

输入描述:

第一行： T 表示以下有T组测试数据（ 1≤ T ≤8 ）
对每组测试数据，
第一行：N X（2 ≤ N ≤100，000 1≤ X ≤100）
第二行：N个整数，分别表示N个区域的高度Hi( 1<=Hi<=100 , i=1…. n)

输出描述:

对每组测试数据，输出占一行，一个整数，即经过部分填高工程改造后的最少费用。

样例输入:

1
5 2
2 3 5 1 4

样例输出:

15

——听学长说，这个题目在比赛时数据比较水，直接三层循环就过了，郑州轻工业OJ上的数据也比较水，大家可以去试试。但是南阳理工的OJ上面的数据量被加大了，我本人提交了好多次时间超限，学长说不知道怎么优化，结果最后莫名其妙的过了。因为循环次数太多了，所以寄存器运行register起到了作用。register可以加快速度。

我一开始写的时候也超时好多次

不知道什么原因，我数组开的有点小了，OJ也显示时间超限，可是被害惨了。

代码如下：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int n,t,x;
int f(int a){
    return a*a;
}
int arr[100005];
int dp[100005][101];
#define re register//寄存器运行
int main()
{
    cin>>t;
    while(t--){
        cin>>n>>x;
        int _max=-1;
        memset(dp,0x3f,sizeof dp);
        for(re int i=1;i<=n;i++){
            scanf("%d",arr+i);
            _max=max(_max,arr[i]);//这个是判断下面的循环次数，因为没必要判断比最高的山还高的高度
            //当然也可以直接在下面的循环中循环100次
            //本人推荐循环100次因为上面的循环n次中判断次数也不少
        }
        for(re int j=arr[1];j<=_max;j++){
            dp[1][j]=f(j-arr[1]);
        }
        for(re int i=2;i<=n;i++){//寄存器运行
            for(re int j=arr[i];j<=_max;j++){//同上
                int temp=f(j-arr[i]);//计算垫高需要的费用，提前计算一下，因为下面的循环中要用到
                					//如果循环中再计算的话会更耗时
                for(re int k=arr[i-1];k<=_max;k++){
                    dp[i][j]=min(dp[i][j],dp[i-1][k]+temp+abs(k-j)*x);
                }
            }
        }
        int Min=0x3f3f3f3f;
        for(re int i=0;i<=100;i++){
            Min=min(Min,dp[n][i]);
        }
        cout<<Min<<endl;
    }
    return 0;
}

附上一张AC图片