Estrella's Travel （最短路spfa）

Estrella's Travel

Time Limit : 1000 MS     Memory Limit : 65536 KB

题目描述

Estrella喜欢旅行，她准备去自己心仪的城市看风景。她精心选择了一条线路，准备自驾游。自驾线路是一条链路，上有 n+1 座城市，编号依次为 0∼n ，Estrella住在 0 号城市，目的地是 n 号城市。Estrella为了环保，驾驶了一辆纯电动车，所以最多只能一次行驶 m 个城市的距离。如果Estrella在某个城市（包括 n 号城市）停留，自然会需要一些费用，当然每个城市的花费是不一样，Estrella想知道，这个花费最小是多少？

输入

多组数据输入。
每组数据第一行两个整数 n(1<n<1000),m(1≤m<n) 。
第二行 n 个整数 ai(0<ai≤1000) ，表示编号为 1,2,⋯,n 的城市的停留花费。

输出

对于每个数据，输出一行，表示总的花费。

样例输入

5 2
1 2 3 4 5
6 2
6 5 4 3 2 1

样例输出

9
9

样例解释

第1个样例，依次到城市1,3,5，所以花费为1+3+5=9；
第2个样例，依次到城市2,4,6，所以花费为5+3+1=9。

题解：题意为1至n各个点，每个点有他独自的花费，一次最多可以走m个点，求1到n的最小花费……

我首先想的的是最短路，用最短路打出来后，发现可以用动归，可是，不会写，以后再补吧……

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<stdlib.h>
#include<time.h>
#include<string>
#include<math.h>
#include<map>
#include<queue>
#include<stack>
#define INF 0x3f3f3f3f
#define ll long long
#define For(i,a,b) for(int i=a;i<b;i++)
#define sf(a)  scanf("%d",&a)
#define sff(a,b)  scanf("%d%d",&a,&b)
#define pf(a) printf("%d\n",a)
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
int w[1005][1005],d[1005],a[1005]; // w存各边的距离，d存0到点I的距离，a存各点自己的花费
int n,m;
bool v[1005];                // 标记
void spfa()
{
    mem(v,0);
    mem(d,INF);
    queue<int>q;
    q.push(0);
    d[0]=0;
    v[0]=1;
    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        v[x]=0;
        for(int i=1; i<=n; i++)
        {
            if(d[x]+w[x][i]<d[i])
            {
                d[i]=d[x]+w[x][i];
                if(!v[i])
                {
                    q.push(i);
                    v[i]=1;
                }
            }
        }
    }
    cout<<d[n]<<endl;
}
int main()
{
    while(~sff(n,m))
    {
        mem(a,0);
        mem(w,INF);
        For(i,1,n+1)
        {
            sf(a[i]);
        }
        For(i,0,n+1)         // 预处理数据……
        {
            for(int j=i; j<=m+i&&j<n+1; j++)     // i 可以走的地方 
            {
                if(i==j)w[i][j]=a[i];
                else
                w[i][j]=w[j][i]=a[j];
            }
        }
        spfa();
    }
}