SGU 145.Strange People(无环K短路)

时间:0.25s
空间:4m

题意:

     其实就是求无环第K短路。
     

输入:

     给出n,m,k,分别代表,n个点,m条边,第k长路。
     接下来m行,三个整数x,y,z,分别代表x,y之间有条费用为x的双向路。保证没有重边。

输出:

    第一行两个数a,b,第k小费用a,和经过的点的个数b。
    接下来b个数,代表第k短的路径。


Sample Input

 

 5 10 3
 1 2 6
 1 3 13
 1 4 18
 1 5 35
 2 3 14
 2 4 34
 2 5 17
 3 4 22
 3 5 15
 4 5 34
 1 5

Sample Output

 

 35 2 
 1 5




Solution:
求k短路的算法,基本都是A*,这里的数据量比较小,可以采用二分答案。
二分路径的长度,DFS求出有多少条路径的长度小于它,如果是(k-1)的话直接输出。


PS:sgu在这一题上数据似乎出了问题,很多人都PE没法AC,我试着提交别人AC过的代码还是pe。
因此代码没有AC,但对程序的正确性有把握。

参考代码:
#include <cstdio>
const int INF = 111;
int g[INF][INF], vis[INF], path[INF];
int n, m, k, x, y, z, l, r, mid, leSum;
int S, T, pd, len, tol;
void dfs (int x, int dis) {
	vis[x] = 1;
	if (x == T) {
		if (dis < mid)  leSum++;
		if (!pd && dis == mid) leSum++, pd = 1;
	}
	else
		for (int i = 1; i <= n; i++)
			if (!vis[i] && g[x][i] && dis + g[x][i] <= mid)
				dfs (i, dis + g[x][i]);
	vis[x] = 0;
}
int check (int x) {
	pd = leSum = 0;
	dfs (S, 0);
	return leSum;
}
int getPath (int x, int dis) {
	vis[x] = 1;
	if (x == T && dis == len) {
		path[++tol] = x;
		return pd = 1;
	}
	else
		for (int i = 1; i <= n; i++) {
			if (!vis[i] && g[x][i] && g[x][i] + dis <= len) {
				if (getPath (i, dis + g[x][i]) ) path[++tol] = x;
				if (pd) return 1;
			}
		}
	vis[x] = 0;
}
int Search () {
	while (l <= r) {
		mid = l + (r - l >> 1);
		int tem = check (mid);
		if (tem == k)
			return mid;
		else if (tem > k)
			r = mid - 1;
		else l = mid + 1;
	}
	return -1;
}
int main() {
	scanf ("%d %d %d", &n, &m, &k);
	for (int i = 1; i <= m; i++) {
		scanf ("%d %d %d", &x, &y, &z);
		g[x][y] = g[y][x] = z, r += z;
	}
	scanf ("%d %d", &S, &T);
	len = Search();
	pd = tol = 0;
	getPath (S, 0);
	printf ("%d %d\n", len, tol);
	for (int i = tol; i > 1; i--)
		printf ("%d ", path[i]);
		printf("%d\n",path[1]);
	return 0;
}

  


  


posted @ 2014-07-13 16:45  keambar  阅读(455)  评论(0编辑  收藏  举报