CCF201712-4行车路线

思路：用vector进行存储图。

dis节点，里面有之前选择道路（大道还是小道）的情况，每一条道路的长度，当前的总的花费。

计算花费的时候，用longlong形式，虽然结果不会超过1e6，但中间计算的时候会超int

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e3+10;
typedef long long ll;
struct Node{
	int y;
	string flag;//0大道 1 小道 
	int len;
	Node(){}
	Node(int a,string b,int c){
		y = a;
		flag = b;
		len = c;
	}
};
struct node{
	ll cost;
	vector<int>len;//这个是这条路上的点 
	string s;//大道小道的情况 
}dis[maxn];//这个是每个顶点的情况 
vector<Node>v[maxn];
int N,M,vis[maxn]={0};
vector<int>tmp;//记录当时的情况 
ll cost_(string s);//看看当前情况疲劳度是多少 
void spfa(int S); 
int main()
{
	int t,a,b,c;
	string s ="";
	cin>>N>>M;
	for(int i=0;i<M;i++){
		cin>>t>>a>>b>>c;
		if(t==0){
			s="0";
		}else{
			s="1";
		}
		v[a].push_back(Node(b,s,c));
		v[b].push_back(Node(a,s,c));
	} //输入好了
	spfa(1);//从节点1开始 
	cout<<dis[N].cost<<endl;
	return 0;
}
void spfa(int S)
{
	queue<int>q;
	for(int i=1;i<=N;i++){
		dis[i].cost = 1e18;
		dis[i].len.clear();
	}
	string s;
	dis[S].cost = 0;
	vis[S] = 1;
	q.push(S);
	while(!q.empty()){
		int t = q.front();
		q.pop();
		vis[t] = 0;//不在队列里面了
		for(int i=0;i<v[t].size();i++){//顶点t链接的所有的点 
			int y = v[t][i].y;
 			tmp.clear(); 
			tmp = dis[t].len;//长度的情况 
			tmp.push_back(v[t][i].len);
			s = dis[t].s + v[t][i].flag;			 
			ll cost = cost_(s);
			if(dis[y].cost > cost){//如果说 选择这条路要更好的话 
				dis[y].cost = cost;
				dis[y].s = s;
				dis[y].len = tmp;
 				if(vis[y] == 0){//如果不在队列里面就入队 
					vis[y] = 1;
					q.push(y);
				}
			} 
		} 
	}
	return ;
}
ll cost_(string s)
{	
	ll re = 0,cnt=tmp[0];
	char last = s[0];
 	for(int i=1;i<s.size();i++){
		if(s[i]!=last){//如果说不是连续的了 
			if(last == '0'){//如果说是大道 
				re += cnt;
 			}else{//前面一直走的是小道 
				re += (cnt*cnt);
 			}
			last = s[i];
			cnt = tmp[i];
		}else{
			cnt += tmp[i];
		}
	}
	if(last == '0'){//如果说是大道 最后一次会没有判断
		re += cnt;
 	}else{//前面一直走的是小道 
		re += (cnt*cnt); 
 	}
	return re;
}