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

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=1615

1615: [Usaco2008 Mar]The Loathesome Hay Baler麻烦的干草打包机
Time Limit: 5 Sec Memory Limit: 64 MB
Submit: 741 Solved: 296
[Submit][Status][Discuss]
Description

Farmer John新买的干草打包机的内部结构大概算世界上最混乱的了，它不象普通的机器一样有明确的内部传动装置，而是，N (2 <= N <= 1050)个齿轮互相作用，每个齿轮都可能驱动着多个齿轮。 FJ记录了对于每个齿轮i，记录了它的3个参数：X_i，Y_i表示齿轮中心的位置坐标(-5000 <= X_i <= 5000; -5000 <= Y_i <= 5000)；R_i表示该齿轮的半径(3 <= R_i <= 800)。驱动齿轮的位置为0,0，并且FJ也知道最终的工作齿轮位于X_t,Y_t。 驱动齿轮顺时针转动，转速为10,000转/小时。你的任务是，确定传动序列中所有齿轮的转速。传动序列的定义为，能量由驱动齿轮传送到工作齿轮的过程中用到的所有齿轮的集合。对能量传送无意义的齿轮都应当被忽略。在一个半径为Rd，转速为S转/每小时的齿轮的带动下，与它相接的半径为Rx的齿轮的转速将为-S*Rd/Rx转/小时。S前的负号的意思是，一个齿轮带动的另一个齿轮的转向会与它的转向相反。 FJ只对整个传动序列中所有齿轮速度的绝对值之和感兴趣，你的任务也就相应转化成求这个值。机器中除了驱动齿轮以外的所有齿轮都被另外某个齿轮带动，并且不会出现2个不同的齿轮带动同一个齿轮的情况。 相信你能轻易地写个程序来完成这些计算:)

Input

• 第1行: 3个用空格隔开的整数：N，X_t，Y_t

• 第2..N+1行: 第i+1描述了齿轮i的位置及半径：X_i，Y_i，以及R_i

Output

• 第1行: 输出所有在传动中起到作用的齿轮转速的绝对值，包括驱动齿轮和 工作齿轮。只需要输出答案的整数部分

Sample Input

4 32 54

0 0 10

0 30 20

32 54 20

-40 30 20

机器里一共有4个齿轮，位于0,0的是半径为10的驱动齿轮，它带动了位于0,30的，半径为20的某个齿轮。这个齿轮又间接带动了位于32,54，半径为20的

工作齿轮，以及一个位于-40,30，半径同样为20的冗余的齿轮。

Sample Output

20000

HINT

输出说明:

齿轮 位置 半径 转速

1 (0,0) 10 10,000

2 (0,30) 20 -5,000

3 (32,54) 20 5,000

                ------

齿轮转速绝对值之和:20,000

思路：题目中有“不会出现2个不同的齿轮带动同一个齿轮的情况”，所以搜索可过，bfs时满足

sqr(x[i]-x[j])+sqr(y[i]-y[j])==sqr(r[i]+r[j]);

即可，速度公式如下：

v[i]=v[now]*r[now]/r[i];
ans[i]=ans[now]+v[i];

代码：

#include<iostream>
#include<stdio.h>
#include<queue>
#define N 1055
using namespace std;

int n;
int x[N],y[N],r[N];
double v[N],ans[N];
int tx,ty; 
int t;
queue<int> q;
int vis[N];
int sqr(int x)
{
    return x*x;
}
bool ok(int i,int j)
{
    return sqr(x[i]-x[j])+sqr(y[i]-y[j])==sqr(r[i]+r[j]);
}
void bfs()
{
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        for(int i=1;i<=n;i++)
        if(!ans[i]&&ok(i,now))
        {
                v[i]=v[now]*r[now]/r[i];
                ans[i]=ans[now]+v[i];
                q.push(i);
        }
    }
     printf("%d",(int)ans[t]);
}
int main()
{
    scanf("%d%d%d",&n,&tx,&ty);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d%d",&x[i],&y[i],&r[i]);
        if(x[i]==tx&&y[i]==ty)
        {
            t=i;
        }
        if(x[i]==0&&y[i]==0)
        {
            vis[i]=1;
            q.push(i);
            v[i]=10000;
            ans[i]=10000;
        }
    }

    bfs();
}