NYOJ_1276_机器设备

/*
机器设备
时间限制：1000 ms  |  内存限制：65535 KB
难度：2

描述

    Alpha 公司设计出一种节能的机器设备。它的内部结构是由 N 个齿轮组成。整个机器设备有 一个驱动齿轮，
    当启动它时，它立即按 10,000 圈/小时转速顺时针转动，然后它又带动与它相切 的齿轮反方向，即逆时针转动。
    齿轮之间互相作用，每个齿轮都可能驱动着多个齿轮，最终带动 一个工作齿轮完成相应的任务。 在这套设备中，
    记录了每个齿轮的圆心坐标和齿轮半径。已知驱动齿轮位于（0,0），最终的 工作齿轮位于（Xt, Yt）。
     Alpha 公司想知道传动序列中所有齿轮的转速。所谓传动序列，即能量由驱动齿轮传送，
     最 后到达工作齿轮的过程中用到的所有齿轮。能量传送过程是，在一个半径为 R，转速为 S 圈/每小 时的齿轮的带动下，
     与它相切的半径为 R’的齿轮的转速为-S*R/R’ 转/小时。负号的意思是， 表 示按反方向转动。

     

    已知，机器设备中除了驱动齿轮以外，所有齿轮都可能被另外某个齿轮带动，并且不会出现 2
    个不同的齿轮带动同一个齿轮的情况。 你的任务是计算整个传动序列中所有齿轮的能量之和。即所有齿轮转速的绝对值之和。

输入
    第一行： T 表示以下有 T 组测试数据（1≤T ≤8）
    对每组测试数据：
    第 1 行： N Xt Yt （2≤N ≤1100）
    接下来有 N 行， Xi Yi Ri 表示 N 个齿轮的坐标和半径 i=1,2,….,N
    （ -5000 ≤Xi ，Yi ≤ 5000 3 ≤ Ri ≤ 1000 ）
    坐标以及半径是整数
输出
    每组测试数据，输出占一行，即所有齿轮转速的绝对值之和 在double范围内，输出整数部分
样例输入

    1
    4 32 54
    0 30 20
    0 0 10
    32 54 20
    -40 30 20

样例输出

    20000

*/

#include <cstdio>
#include <cmath>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
struct Wheel
{
	int x;
	int y;
	int r;
}edge[1110];

bool flag[1110];
double s[1110];

bool f;
int n,begin,end;

bool judge(int a,int b)//判断是否相切 
{
	double R = pow((edge[a].x-edge[b].x),2)+pow((edge[a].y-edge[b].y),2);
	if(R == pow((edge[a].r+edge[b].r),2))
	{
		return true;
	}
	return false;
}
void DFS(int con,double ans)
{
	int i;
	if(f==true)
	{
		return ;
	}
	if(con == end)
	{
		f = true;
		printf("%d\n",(int)ans);		
		return ;
	}
	for(i=1;i<=n;i++)
	{
		if(flag[i]==false)
		{
			if(judge(i,con))//递归 + 回溯 
			{
				s[i] = s[con] * edge[con].r/edge[i].r;
				flag[i] = true;	
				ans += s[i];			
				DFS(i,ans);			
				ans -= s[i];
				flag[i] = false;
			}
		}
	}	
}
int main()
{
	int T;
	cin >> T;
	while(T--)
	{
		f = false;
		int end_x,end_y,i,j;
		scanf("%d%d%d",&n,&end_x,&end_y);
		memset(flag,false,sizeof(flag));
		for(i=1;i<=n;i++)
		{
			scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].r);
			if(edge[i].x == 0 && edge[i].y == 0)//找到起始位置编号 
			{
				begin = i;
				s[begin] = 10000;
			}
			else if(edge[i].x == end_x && edge[i].y == end_y)//找到终点位置编号 
			{
				end = i;
			}
		}
		flag[begin] = true;
		DFS(begin,s[begin]);
	}
	return 0;
}