1. 程式人生 > >CCF認證 2017-12 行車路線

CCF認證 2017-12 行車路線

  小明和小芳出去鄉村玩,小明負責開車,小芳來導航。
  小芳將可能的道路分為大道和小道。大道比較好走,每走1公里小明會增加1的疲勞度。小道不好走,如果連續走小道,小明的疲勞值會快速增加,連續走 s公里小明會增加 s 2的疲勞度。
  例如:有5個路口,1號路口到2號路口為小道,2號路口到3號路口為小道,3號路口到4號路口為大道,4號路口到5號路口為小道,相鄰路口之間的距離都是2公里。如果小明從1號路口到5號路口,則總疲勞值為(2+2) 2+2+2 2=16+2+4=22。
  現在小芳拿到了地圖,請幫助她規劃一個開車的路線,使得按這個路線開車小明的疲勞度最小。 輸入格式   輸入的第一行包含兩個整數 n
m,分別表示路口的數量和道路的數量。路口由1至 n編號,小明需要開車從1號路口到 n號路口。
  接下來 m行描述道路,每行包含四個整數 tabc,表示一條型別為 t,連線 ab兩個路口,長度為 c公里的雙向道路。其中 t為0表示大道, t為1表示小道。保證1號路口和 n號路口是連通的。 輸出格式   輸出一個整數,表示最優路線下小明的疲勞度。 樣例輸入 6 7
1 1 2 3
1 2 3 2
0 1 3 30
0 3 4 20
0 4 5 30
1 3 5 6
1 5 6 1 樣例輸出 76 樣例說明   從1走小道到2,再走小道到3,疲勞度為5 2
=25;然後從3走大道經過4到達5,疲勞度為20+30=50;最後從5走小道到6,疲勞度為1。總共為76。 資料規模和約定   對於30%的評測用例,1 ≤  n ≤ 8,1 ≤  m ≤ 10;
  對於另外20%的評測用例,不存在小道;
  對於另外20%的評測用例,所有的小道不相交;

  對於所有評測用例,1 ≤ n ≤ 500,1 ≤ m ≤ 105,1 ≤ ab ≤ nt是0或1,c ≤ 105。保證答案不超過106

Dij演算法變形 

在模板算權值的時候 加入小道判斷 如果是小道 則diss = dist[u] - small*small + (small + len)*(small + len)

small為 通過small小道長度走到點u,len為 u到v的距離

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include<cmath>
using namespace std;
const int maxn = 20005;
const int maxm = 200000;
const int inf = 0x3f3f3f3f;
int tot, head[maxn], n;
struct edge
{
	int v, len, next,size;
}e[maxm * 2];

void init()
{
	tot = 0;
	memset(head, -1, sizeof(head));
}

void addedge(int u, int v, int len,int size)
{
	e[tot] = { v,len,head[u],size };
	head[u] = tot++;
}

struct node
{
	int p, num;
}a[20005];

struct dnode
{
	long long p, len, small;
	bool operator<(const dnode &T) const
	{
		return len>T.len;
	}
};
bool vis[maxn];
long long dist[maxn];
long long dij(int st, int en)
{
	memset(vis, 0, sizeof(vis));
	for (int i = 0; i <= n; i++)
		dist[i] = inf;
	priority_queue<dnode> que;
	while (!que.empty())
		que.pop();
	que.push({ st,0,0 });
	dist[st] = 0;
	int ans = -1;
	while (!que.empty())
	{
		long long u = que.top().p,small = que.top().small;
		if (u == en) {
			ans = en;
			break;
		}
		que.pop();
		if (vis[u])
			continue;
		vis[u] = 1;
		for (int i = head[u]; i != -1; i = e[i].next)
		{
			long long v = e[i].v;
			long long len = e[i].len;
			if (e[i].size == 0) {
				if (!vis[v] && dist[v] > dist[u] + len)
				{
					dist[v] = dist[u] + len;
					que.push({ v,dist[v],0 });
				}
			}
			//和模板不同在於多了 else中的判斷小道還是大路
			else {
				long long diss = dist[u] - small*small + (small + len)*(small + len);
				if (!vis[v] && dist[v] > diss) {
					dist[v] = diss;
					que.push({ v,dist[v],small+len });
				}
			}
		}
	}
	if (ans == en)return dist[ans];
	else return -1;
}
int main()
{
	int  m, u, v, d, a;
	while (cin >> n >> m)
	{
		init();
		for (int i = 1; i <= m; i++)
		{
			scanf("%d%d%d%d", &a, &u, &v, &d);
			addedge(u, v, d, a);
			addedge(v, u, d, a);
		}
		cout << dij(1, n) << endl;
	}
	return 0;
}