POJ 1062 最短路 dijskra + 枚举

ZWJ 推荐的一道最短路神题

等级限制这个条件可以枚举所有符合条件的区间的点，从最高点到酋长的等级。

然后就是求最短路了。算是 dijstra 的模板吧。

#include <stdio.h>
#include <iostream>
#include <queue>
#include <algorithm>
#include <map>
#include <vector>
#include <cmath>
#include <string.h>
#include <stdlib.h>
#include <time.h>
#include <fstream>
#include <set>
#include <stack>
#include <list>
using namespace std;

#define READ freopen("acm.in","r",stdin)
#define WRITE freopen("acm.out","w",stdout)
#define ll long long
#define ull unsigned long long 
#define uint unsigned int
#define PII pair<int,int>
#define PDD pair<double,double>
#define fst first
#define sec second
#define MS(x,d) memset(x,d,sizeof(x))
#define INF 1000001000
#define ALL(x) x.begin(),x.end()
#define PB push_back
#define MOD 99991
#define MAX 11111

struct node
{
    int to,cost;
};

vector<node> G[1000];
int price[1000];
int level[1000];
vector<int> GL[1000];
vector<node> re[1000];
int m,n;
int LM;
int dijskra()
{
    int dist[1000];
    fill(dist,dist+1000,INF);
    dist[n+1]=0;
    priority_queue< PII,vector<PII>,greater<PII> > q;
    q.push(PII(0,n+1));
    while(!q.empty())
    {
        PII p=q.top();q.pop();
        if(p.fst>dist[p.sec])
            continue;
        for(int i=0;i<G[p.sec].size();i++)
        {
            node tn=G[p.sec][i];
            if(dist[tn.to]>dist[p.sec]+tn.cost)
            {
                dist[tn.to]=dist[p.sec]+tn.cost;
                q.push(PII(dist[tn.to],tn.to));
            }
        }
    }
    return dist[1];
}
int solve()
{
    int maxL=-1;
    for(int i=1;i<=n;i++)
        maxL=max(level[i],maxL);
    maxL=min(LM+m,maxL);
    int ans=INF;
    for(int l=maxL;l>=LM;l--)// 枚举区间最高等级 从最高可到等级到酋长
    {
        MS(G,0);
        for(int tl=l;tl>=0&&abs(tl-l)<=m;tl--)//枚举在区间中的物品
        {
            for(int k=0;k<GL[tl].size();k++)
            {
                int to=GL[tl][k];
                G[n+1].PB((node){to,price[to]});
                for(int j=0;j<re[to].size();j++)
                {
                    node tmp=re[to][j];
                    if(abs(LM-level[tmp.to])<=m&&level[tmp.to]<=l)
                        G[tmp.to].PB((node){to,tmp.cost});
                }
            }
        }
        ans=min(ans,dijskra());
    }
    return ans;
}
int main()
{
    READ;
    while(scanf("%d%d",&m,&n)!=EOF)
    {
        MS(re,0);
        MS(GL,0);
        for(int i=1;i<=n;i++)
        {
            int p,l,x;
            scanf("%d%d%d",&p,&l,&x);
            price[i]=p,level[i]=l;
            GL[l].PB(i);
            if(i==1)
                LM=l;    
            for(int j=0;j<x;j++)
            {
                int to,cost;
                scanf("%d%d",&to,&cost);
                re[i].PB((node){to,cost});
            }
        }
        cout<<solve()<<endl;
    }   
    return 0;
}