TYVJ2032 「Poetize9」升降梯上

P2032 「Poetize9」升降梯上

时间: 1000ms / 空间: 131072KiB / Java类名: Main

描述

开启了升降梯的动力之后，探险队员们进入了升降梯运行的那条竖直的隧道，映入眼帘的是一条直通塔顶的轨道、一辆停在轨道底部的电梯、和电梯内一杆控制电梯升降的巨大手柄。
Nescafe之塔一共有N层，升降梯在每层都有一个停靠点。手柄有M个控制槽，第i个控制槽旁边标着一个数Ci，满足
C1<C2<C3<……<CM。如果Ci>0，表示手柄扳动到该槽时，电梯将上升Ci层；如果Ci<0，表示手柄扳
动到该槽时，电梯将下降-Ci层；并且一定存在一个Ci=0，手柄最初就位于此槽中。注意升降梯只能在1~N层间移动，因此扳动到使升降梯移动到1层以
下、N层以上的控制槽是不允许的。
电梯每移动一层，需要花费2秒钟时间，而手柄从一个控制槽扳到相邻的槽，需要花费1秒钟时间。探险队员现在在1层，并且想尽快到达N层，他们想知道从1层到N层至少需要多长时间？

输入格式

第一行两个正整数N、M。
第二行M个整数C1、C2……CM。

输出格式

输出一个整数表示答案，即至少需要多长时间。若不可能到达输出-1。

测试样例1

输入

6 3
-1 0 2

输出

19

备注

手柄从第二个槽扳到第三个槽（0扳到2），用时1秒，电梯上升到3层，用时4秒。
手柄在第三个槽不动，电梯再上升到5层，用时4秒。
手柄扳动到第一个槽（2扳到-1），用时2秒，电梯下降到4层，用时2秒。
手柄扳动到第三个槽（-1扳倒2），用时2秒，电梯上升到6层，用时4秒。
总用时为(1+4)+4+(2+2)+(2+4)=19秒。

对于30% 的数据，满足1≤N≤10，2<=M<=5。
对于 100% 的数据，满足1≤N≤1000，2<=M<=20，-N<C1<C2<……<CM<N。

 

【题解】

(i - 1) * m + j - 1表示在第i层j槽处

连边Dijstra即可

 #include <iostream>
 #include <cstdio>
 #include <vector>
 #include <cstdlib>
 #include <cstring>
 #include <queue>
 #define max(a, b) ((a) > (b) ? (a) : (b))
 #define min(a, b) ((a) < (b) ? (a) : (b))
 #define abs(a) ((a) < 0 ? (-1 * (a)) : (a)) 

 const int INF = 0x3f3f3f3f;
 const int MAXN =  + ;
 const int MAXM =  + ;

 inline void read(int &x)
 {
     x = ;char ch = getchar(), c = ch;
     while(ch < '' || ch > '') c = ch, ch = getchar();
     while(ch <= '' && ch >= '') x = x *  + ch - '', ch = getchar();
     if(c == '-')x = -x;
 }

 int n,m;

 inline int number(int a, int b)
 {
     return (a - ) * m + b - ;
 }

 struct Edge
 {
     int u,v,w,next;
     Edge(int _u, int _v, int _w, int _next){u = _u;v = _v;w = _w;next = _next;}
     Edge(){}
 }edge[MAXN * MAXM * MAXM * ];
 int head[MAXN * MAXM * ], cnt, c[MAXN];

 inline void insert(int a, int b, int c)
 {
     edge[++cnt] = Edge(a,b,c,head[a]);
     head[a] = cnt;
 }

 struct Node
 {
     int node,w;
     Node(int a, int b){node = a;w = b;}
     Node(){}
 };

 struct cmp
 {
     bool operator()(Node a, Node b)
     {
         return a.w > b.w;
     }
 };

 std::priority_queue<Node, std::vector<Node>, cmp> q;
 int s, d[MAXN * MAXM * ], b[MAXN * MAXM * ];

 void dijstra()
 {
     memset(d, 0x3f, sizeof(d));
     d[s] = ;
     q.push(Node(s, ));
     Node tmp;
     int u,v;
     while(q.size())
     {
         tmp = q.top(), q.pop();
         if(b[tmp.node])continue;
         u = tmp.node;
         b[u] = ;
         for(register int pos = head[u];pos;pos = edge[pos].next)
         {
             v = edge[pos].v;
             if(b[v])continue;
             if(d[v] > d[u] + edge[pos].w)
             {
                 d[v] = d[u] + edge[pos].w;
                 q.push(Node(v, d[v]));
             }
         }
     }
 }

 int main()
 {
     read(n), read(m);
     for(register int i = ;i <= m;++ i)
     {
         read(c[i]);
         if(c[i] == )s = number(, i);
     }
     for(register int i = ;i < n;++ i)
     {
         for(int j = ;j <= m;++ j)
         {
             if(i + c[j] >  && i + c[j] <= n && c[j] != )
                 insert(number(i,j), number(i + c[j], j), abs(c[j]) * );
             for(int k = ;k <= m;++ k)
                 if(k != j)
                     insert(number(i, j), number(i, k), abs(j - k));
         }
     }
     dijstra();
     int ans = INF;
     for(register int i = ;i <= m;++ i)
         ans = min(ans, d[number(n, i)]);
     if(ans == INF)ans = -;
     printf("%d", ans);
     return ;
 }

TYVJ2032