bzoj 1615: [Usaco2008 Mar]The Loathesome Hay Baler麻烦的干草打包机【bfs】

直接bfs即可，注意开double，还有驱动和终点的齿轮都在序列里，要把它们找出来= =

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
const int N=1505;
int n,sx,sy,s,t,q[N],fr[N];
bool v[N];
double ans,dis[N];
struct qwe
{
	int x,y,r;
}a[N];
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
inline int jl(int x,int y)
{
	return (a[x].x-a[y].x)*(a[x].x-a[y].x)+(a[x].y-a[y].y)*(a[x].y-a[y].y);
}
int main()
{
	n=read(),sx=read(),sy=read();
	for(int i=1;i<=n;i++)
		a[i].x=read(),a[i].y=read(),a[i].r=read();
	for(int i=1;i<=n;i++)
	{
		if(!a[i].x&&!a[i].y)
			s=i;
		if(a[i].x==sx&&a[i].y==sy)
			t=i;
	}
	queue<int>q;
	q.push(s),dis[s]=10000,v[s]=1;
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=1;i<=n;i++)
			if(!v[i]&&(a[u].r+a[i].r)*(a[u].r+a[i].r)==jl(u,i))
			{
				fr[i]=u;
				v[i]=1,q.push(i);
				dis[i]=dis[u]*a[u].r/a[i].r;
				if(i==t)
					break;
			}
	}
	for(int i=t;i;i=fr[i])
		ans+=dis[i];
	printf("%d\n",(int)ans);
	return 0;
}