关闭

hdu 4171 Paper Route bfs

标签: struct算法ini
446人阅读 评论(0) 收藏 举报
分类:

        题目意思是给N个点,然后还有一个办公室。下面N+1行输入从办公室(即0号)开始每个到学校的距离,然后N行,正好是棵树,输入这棵树以及权值。要求遍历树上每一点然后到学校的最短的那个距离。算法是首先存下每个点到办公室的点,即把办公室的点看为根,然后通过bfs求得每个点到根的距离,然后把树的总距离求出,因为只要不是终点到根的路径,都是需要走两次的,所以取个min(total*2 - i->0 + i->school)就是答案了。用dfs做TLE了,因为好久没写bfs了,代码其搓无比,看来要多练练了=。=


#include<iostream>
#include<algorithm>
#include<cstdio>
#include<string>
#include<vector>
#include<cstring>
#include<queue>
using namespace std;

#define inf 2000000000
struct point
{
	vector<pair<int,int> > v;
};

int school[100005];
int zero[100005];
int biaoji[100005];

/*void dfs(int now,int val,vector<point> v)//TLE
{
	int i;
	for(i=0;i<v[now].v.size();i++)
	{
		if(biaoji[v[now].v[i].first]) continue;
		zero[v[now].v[i].first]=v[now].v[i].second+val;
		biaoji[v[now].v[i].first]=1;
		dfs(v[now].v[i].first,zero[v[now].v[i].first],v);
	}
}*/

int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		memset(biaoji,0,sizeof(biaoji));
		zero[0]=0;
		int i,j;
		for(i=0;i<n+1;i++)
			scanf("%d",&school[i]);
		vector<point> v(n+1);
		int total=0;
		for(i=0;i<n;i++)
		{
			int a,b,val;
			scanf("%d%d%d",&a,&b,&val);
			pair<int,int> t,tt;
			t.first=b;tt.first=a;
			t.second=val;tt.second=val;
			v[a].v.push_back(t);
			v[b].v.push_back(tt);
			total+=val;
		}
		queue<pair<int,int> > qu;
		int sum=0;
		biaoji[0]=1;
		while(1)
		{
			if(sum==0)
			{
				for(i=0;i<v[0].v.size();i++)
				{
					//v[0].v[i].second+=zero[0];
					qu.push(v[0].v[i]);
					sum=1;
				}
			}
			else
			{
				pair<int,int> t=qu.front();
				zero[t.first]=t.second;
				biaoji[t.first]=1;
				qu.pop();
				for(i=0;i<v[t.first].v.size();i++)
				{
					if(biaoji[v[t.first].v[i].first]) continue;
					v[t.first].v[i].second+=t.second;
					qu.push(v[t.first].v[i]);
				}
			}
			if(qu.empty()) break;
		}
		//dfs(0,0,v);
		total*=2;
		int min=inf;
		for(i=0;i<n+1;i++)
		{
			if(total-zero[i]+school[i]<min) min=total-zero[i]+school[i];
		}
		printf("%d\n",min);
	}
}

复杂度就是BFS的复杂度,其实按理讲就是O(N),查询复杂度会高一点,现在都不会算复杂度了。。。。

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:3773次
    • 积分:154
    • 等级:
    • 排名:千里之外
    • 原创:11篇
    • 转载:0篇
    • 译文:0篇
    • 评论:0条
    文章分类