POJ 1062 昂贵的聘礼

题目连接http://poj.org/problem?id=1062

AC代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include <cstdio>
#include <cstring>
#define MAXN 1000
#define INF 0x3f3f3f3f
int lev[MAXN],pri[MAXN],g[MAXN][MAXN],G[MAXN][MAXN],d[MAXN],vis[MAXN];
using namespace std;
int m,n,k,a,b;
void dijkstra(int u)//一点到其他点的最短路,邻接矩阵
{
    memset(vis,0,sizeof(vis));
	for(int i=1;i<=m;i++)d[i]=(i==u?0:INF);
	for(int i=1;i<=m;i++)
	{
		int x,l=INF;
		for(int y=1;y<=m;y++)if(!vis[y]&&d[y]<=l)l=d[x=y];
		vis[x]=1;
		for(int y=1;y<=m;y++)
			d[y]=min(d[y],d[x]+g[x][y]);
	}
}
void sift(int l,int r)//区间筛选，若一条路的两端的等级不在[l,r]内,则把g[i][j]置为INF
{
    for(int i=1;i<=m;i++)
	   for(int j=1;j<=m;j++)
	   {
		   if(lev[i]>=l&&lev[i]<=r&&lev[j]>=l&&lev[j]<=r)g[i][j]=G[i][j];
		   else g[i][j]=INF;
	   }
}
int pro()
{
	int ans=INF;
	for(int i=lev[1]-n;i<=lev[1];i++)
	{
		sift(i,i+n);//依次扫描区间
		dijkstra(1);
		for(int i=1;i<=m;i++)
		{
			d[i]+=pri[i];
			ans=min(ans,d[i]);
		}
	}
	return ans;
}
int main()
{

   scanf("%d%d",&n,&m);
   for(int i=0;i<=m;i++)
	   for(int j=0;j<=m;j++)
	   {
		   G[i][j]=INF;
		   g[i][j]=INF;
	   }
   for(int i=1;i<=m;i++)
   {
       scanf("%d%d%d",&pri[i],&lev[i],&k);
	   for(int j=1;j<=k;j++)
	   {
           scanf("%d%d",&a,&b);
		   G[i][a]=b;
	   }
   }
   printf("%d\n",pro());
}