Not so Mobile UVA - 839

题目链接

题意

有若干个天平，每个天平是否满足力矩原则，即 \(wl\times dl = wr\times dr\) 。

思路

读入有些麻烦，还得递归，注意细节问题。
我们建树，把每个天平抽象成节点，为了好算我们把一个点的左边或右边是一个子天平，那么这个天平（不是子天平！）的 \(wl\) 是负数，并且子天平在数组里的编号为 \(-wl\) ，右边同理。
最后在判断是否合法时直接深度优先遍历一下就好了，这里我用的返回值数天平的总重量，方便判断，答案是否合法就开一个全局变量即可。
注意为空行细节！！！（我就被坑了）

代码

#include <iostream>
#include <cstdio>
using namespace std;
const int N = 100010;
int h,x;
struct mobile {
	int dl,dr,wl,wr;
}tr[N];
int idx = 0;
bool ans;
int read () {
	int w1,d1,w2,d2;
	cin >> w1 >> d1 >> w2 >> d2;
	tr[idx] = {d1,d2,w1,w2};
	int t = idx++;
	if (w1 == 0) tr[t].wl = -read ();
	if (w2 == 0) tr[t].wr = -read ();
	return t;
}
int dfs (int x) {
	if (!ans) return 0;
	int wl = tr[x].wl,wr = tr[x].wr;
	if (tr[x].wl < 0) {
		wl = dfs (-tr[x].wl);
		if (!ans) return 0;
	}
	if (tr[x].wr < 0) {
		wr = dfs (-tr[x].wr);
		if (!ans) return 0;
	}
	if (wl * tr[x].dl != wr * tr[x].dr) {
		ans = false;
		return 0;
	}
	return wl + wr;
}
int main () {
	int T;
	cin >> T;
	while (T--) {
		ans = true,idx = 0;
		read ();
		dfs (0);
		if (ans) puts ("YES");
		else puts ("NO");
		if (T) cout << endl;
	}
    return 0;
}