POJ 1062 昂贵的聘礼 (最短路径)

最短路径的一点变形，我的做法是这样的：

首先构造一个点0，表示探险家的编号，那么最后求的就是0到1的最短路径，其中根据题意知1是酋长的物品。然后对于每一件物品x（1<=x<=N），首先判断它跟1的等级差距是否小于等于M，否则直接continue，因为由题知它是无法到达1的。然后对于该物品所能替代的物品k,要先判断k与1的等级差距以及k与x的等级差距，如果都满足小于等于M，则对k与x连边（这里应是单向边）。最后做dijkstra或者spfa，注意松弛的时候要记录一下前驱，因为整个松弛过程都要满足等级差距不大于M，刚才对边的处理只是保证了每个点与1和它能替换的点的等级不大于M。

比如这组数据

1 4

1000 3 2

2 1

3 3

100 2

3 1

4 3

100 3 1

4 1

50 4 0

ans: 54 (4->3->1)

如果不记录前驱，则会出现4->3->2->1 = 53，由于M==1，此时4，2的等级差距大于M。

 

/*
dijkstra
296K 16MS 
*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define SIZE 105
#define inf 0xfffffff
using namespace std;

struct node
{
    int val,lev,ct[SIZE],vv[SIZE],num;
}ps[SIZE];

int N,M,ans;
int cnt[SIZE][SIZE],lowcost[SIZE],pre[SIZE];
bool vis[SIZE];

int Abs(int a)
{
    return a > 0? a:(-a);
}

void dijkstra()
{
    for(int i=0; i<=N; i++)
        lowcost[i] = inf;
    for(int i=1; i<=N; i++)
        lowcost[i] = cnt[0][i];
    vis[0] = true;
    for(int i=1; i<=N; i++)
    {
        int Min = inf,k = 0;
        for(int j=0; j<=N; j++)
        {
            if(!vis[j] && lowcost[j] < Min)
            {
                Min = lowcost[j];
                k = j;
            }
        }
        vis[k] = true;
        for(int j=0; j<=N; j++)
        {
            if(!vis[j] && lowcost[j] > Min + cnt[k][j])
            {
                if(pre[k] == 0 || Abs(ps[pre[k]].lev - ps[j].lev) <= M)
                {
                    lowcost[j] = Min + cnt[k][j];
                    pre[j] = k;
                }
            }
        }
    }
}

int main()
{
    while(~scanf("%d%d",&M,&N))
    {
        for(int i=1; i<=N; i++)
        {
            scanf("%d%d%d",&ps[i].val,&ps[i].lev,&ps[i].num);
            for(int j=1; j<=ps[i].num; j++)
                scanf("%d%d",&ps[i].ct[j],&ps[i].vv[j]);
        }
        for(int i=1; i<=N; i++)
            for(int j=1; j<=N; j++)
                cnt[i][j] = inf;
        for(int i=1; i<=N; i++)
            cnt[0][i] = ps[i].val;
        for(int i=1; i<=N; i++)
        {
            if(Abs(ps[i].lev - ps[1].lev) > M)
                continue;
            for(int j=1; j<=ps[i].num; j++)
            {
                int to = ps[i].ct[j];
                if(Abs(ps[to].lev - ps[1].lev) <= M && Abs(ps[to].lev - ps[i].lev) <= M)
                    cnt[to][i] = ps[i].vv[j];
            }
        }
        memset(vis,0,sizeof(vis));
        memset(pre,0,sizeof(pre));
        dijkstra();
        printf("%d\n",lowcost[1]);
    }
    return 0;
}

/*
spfa
304K 16MS 
*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#define SIZE 105
#define inf 0xfffffff

using namespace std;

struct node
{
    int val,lev,num,to[SIZE],dec[SIZE];
}ps[SIZE];

int cnt[SIZE][SIZE],lowcost[SIZE],pre[SIZE];
bool vis[SIZE];
int M,N;

int Abs(int a)
{
    return a > 0? a:(-a);
}

void spfa()
{
    queue <int> Q;
    for(int i=1; i<=N; i++)
    {
        lowcost[i] = inf;
        vis[i] = false;
        pre[i] = 0;
    }
    pre[0] = 0;
    lowcost[0] = 0;
    vis[0] = true;
    Q.push(0);
    while(!Q.empty())
    {
        int cur = Q.front();
        Q.pop();
        vis[cur] = false;
        for(int i=1; i<=N; i++)
        {
            if(cnt[cur][i] != inf && lowcost[i] > lowcost[cur] + cnt[cur][i])
            {
                if(pre[cur] == 0 || Abs(ps[pre[cur]].lev - ps[i].lev) <= M)
                {
                    lowcost[i] = lowcost[cur] + cnt[cur][i];
                    pre[i] = cur;
                    if(!vis[i])
                    {
                        Q.push(i);
                        vis[i] = true;
                    }
                }
            }
        }
    }
}

int main()
{
    while(~scanf("%d%d",&M,&N))
    {
        for(int i=1; i<=N; i++)
        {
            scanf("%d%d%d",&ps[i].val,&ps[i].lev,&ps[i].num);
            for(int j=1; j<=ps[i].num; j++)
                scanf("%d%d",&ps[i].to[j],&ps[i].dec[j]);
        }
        for(int i=1; i<=N; i++)
            for(int j=1; j<=N; j++)
                cnt[i][j] = inf;
        for(int i=1; i<=N; i++)
            cnt[0][i] = ps[i].val;
        for(int i=1; i<=N; i++)
        {
            if(Abs(ps[i].lev - ps[1].lev) > M)
                continue;
            for(int j=1; j<=ps[i].num; j++)
            {
                int to = ps[i].to[j];
                if(Abs(ps[i].lev-ps[to].lev <= M && Abs(ps[1].lev-ps[to].lev)<=M))
                    cnt[to][i] = ps[i].dec[j];
            }
        }
        spfa();
        printf("%d\n",lowcost[1]);
    }
    return 0;
}