201712CSPT4行车路线

题意：一个无向连通图有n个点，m条边，每一条边有权值和一个编号id。id=0时，经过该边花费为其权值；当id=1时，经过该边花费为连续经过id为1的边的权值平方。求从1到n最小花费。

#include<bits/stdc++.h>
using namespace std;
struct Node
{
	int w;                      
	int v;
	int id;//边权,邻接点,大路小路 
	Node(int w,int v,int id):w(w),v(v),id(id){}
};
struct node
{
	long long u,j,d;//结点,小路长度,距离 
	bool operator <(const node y) const
	{
		return d>y.d;
	}
};
const long long INF=9999999999999999;
long long dis[2010][2010],vis[2010][2010];
vector <Node> V[510];
int n,m;
void Dij()
{
	memset(dis,0x3f,sizeof(dis));
	priority_queue <node> q;
	dis[1][0]=0;
	node t;t.d=0;t.u=1;t.j=0;q.push(t);
	while(q.size()!=0)
	{
		int x=q.top().u;int j=q.top().j;int D=q.top().d;
		q.pop();
		if(vis[x][j]==1)continue;
		vis[x][j]=1;
		for(int i=0;i<V[x].size();i++)
		{
			int y=V[x][i].v;int w=V[x][i].w;int id=V[x][i].id;//cout<<y<<w<<id<<endl;
			if(id==1)
			{
				if(j+w<=1000)
				if(dis[y][j+w]>D-j*j+(j+w)*(j+w))
				{
					dis[y][j+w]=D-j*j+(j+w)*(j+w);	//cout<<dis[y][j+w]<<endl;
					q.push({y,j+w,dis[y][j+w]});
				}
			}
			else 
			{
				if(dis[y][0]>D+w)
				{
					dis[y][0]=D+w;
					q.push({y,0,dis[y][0]});
				}
			}
		}
	}
}
int main()
{		
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int id,u,v,w;
		cin>>id>>u>>v>>w;
		V[u].push_back(Node(w,v,id));
		V[v].push_back(Node(w,u,id));
	}
	Dij();
	long long minn=INF;
	for(int i=0;i<=1000;i++)
	{
		minn=min(minn,dis[n][i]);
	}
	cout<<minn;
	return 0;
}