道路网络

【题目描述】

X 王国准备建立城市道路网络，X 王国有一位伟大的巫师，这位巫师可以把其中某一条直接由两个城市连接的路用魔法建造，这样的话，这条路的花费可以减少为原来的一半（假

定100 的一半为50， 99 的一半为49）但是这样的魔法只能使用一次！给出预计道路网络的建造代价，国王想知道某两个城市之间建造一条通路的最小代价。由于X 王国很大，所以他的建造师求助你让你帮忙解决这个问题。

【标准输入】
第一行：测试数据组数T（T<=50）
对于每组测试数据第一行输入城市数N(2 <= N <= 1000)，和总的预计道路数M(0 <= M <= 50000)为了简化问题，城市用数字表示，接下来M 行输入预算的两个城市(u, v)，以及他们之间的道路花费(e)，(1<= u,v <= N, 0 <= e <= 1000)，两点之间可能存在多条路径，路都是双向的。接下来输入国王想知道建立通路的两个城市序号u，v，(1<= u,v <= N)
【标准输出】
对于每组测试数据，如果不可能建立一条通路，输出”The budget is wrong”

否则输出建立通路的最小代价。

看着别人的代码怎么感觉自己这么久来的努力简直就像是小孩子弄的东西，太受打击了啊。。。不多说了。。

#include <set>
#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <limits.h>
#include <string.h>
#include <string>
#include <algorithm>
#define MID(x,y) ( ( x + y ) >> 1 )
#define L(x) ( x << 1 )
#define R(x) ( x << 1 | 1 )
#define FOR(i,s,t) for(int i=(s); i<(t); i++)
#define BUG puts("here!!!")
#define STOP system("pause")
#define file_r(x) freopen(x, "r", stdin)
#define file_w(x) freopen(x, "w", stdout)

using namespace std;

const int MAX = 1010;
const int MAXM = 50010*2;
const int inf = INT_MAX;
struct NODE{
	int v, len;
	NODE *next;
};

NODE node[MAXM], *p[MAX];
int atoall[MAX], btoall[MAX];
int cnt;
void init()
{
	cnt = 0;
	memset(p, 0, sizeof(p));
}

void add(int u, int v, int len, NODE *p[], NODE node[])
{
	node[cnt].v = v;
	node[cnt].len = len;
	node[cnt].next = p[u];
	p[u] = &node[cnt++];
}

int Dijkstra(int n, int from, int to, int *dis)
{
	bool used[MAX];
	memset(used, false, sizeof(used));
	fill(dis, dis+n, inf);
	dis[from] = 0;
	used[from] = true;
	int now = from;
	
	FOR(i, 0, n)
	{
		NODE *head = p[now];
		while( head )
		{
			int v = head->v;
			int len = head->len;
			if( dis[now] != INT_MAX && dis[v] > dis[now] + len )
				dis[v] = dis[now] + len;
			head = head->next;
		}
		int min = inf;
		FOR(k, 0, n)
			if( !used[k] && dis[k] < min )
				min = dis[now = k];
		used[now] = true;
	}
	return dis[to];
}

int solve(int n, int from, int to)
{
	int len = Dijkstra(n, from, to, atoall);
	if( len == inf ) return -1;
	Dijkstra(n, to, from, btoall);
	int mmin = inf;
	FOR(i, 0, n)
	{
		NODE *head = p[i];
		while( head )
		{
			int u = i;
			int v = head->v;
			if( atoall[u] != inf && btoall[v] != inf )
				mmin = min(mmin, atoall[u] + btoall[v] + head->len/2);
			head = head->next;
		}
	}
	return mmin;
}
int main()
{
 	int n, m, u, v, ncases, len;

 	scanf("%d", &ncases);
 	
 	while( ncases-- )
 	{
		scanf("%d%d", &n, &m);
		init();
		while( m-- )
		{
			scanf("%d%d%d", &u, &v, &len);
			u--; v--;
			add(u, v, len, p, node);
			add(v, u, len, p, node);
		}
		
		scanf("%d%d", &u, &v);
		u--; v--;
		
		int ans = solve(n, u, v);
		if( ans == -1 )
		{
			printf("The budget is wrong\n");
			continue;
		}
		printf("%d\n", ans);
	}

return 0;
}