题目如下:你有足够数量的天秤和砝码。每个天秤有10磅。天秤的左右两边可以放砝码,也可以放天秤。题目要求是,在给定的初始组合情况下,如何添加砝码,让整体保证平衡。输入是一串数字,第一行是一个整数,代表当前初始状态一共有多少个天秤,每个天秤都有一个序号接下来往下,每两行分别代表一个天秤左右两边所包含的天秤序号和包含的砝码重量每一组的格式如下:左半的砝码重量 <天秤i,天秤j,...>右半的砝码重量 <天秤k,天秤m,...>其中,<>表示数组输入样例如下:4 //4个天秤0 1 //0号天秤左边放置着1号天秤0 2 //0号天秤右边放置着2号天秤0 //1号天秤左边啥都没有0 3 //1号天秤右边放置着3号天秤3 //2号天秤左边有三磅重的砝码0 //2号天秤右边啥都没有0 //3号天秤左边啥都没有0 //3号天秤右边啥都没有输出,输出N行。第i行代表第i个天秤的左右两边需要添加多少重量的砝码输出样例如下:0: 0 141: 10 02: 0 33: 0 0大家可以试试看。当时给我的时间是1小时,当时做完这题就可以进入phone interview,可惜挂在了phone interview那- -。
2 回答
BIG阳
TA贡献1859条经验 获得超6个赞
下面来简单的说一下思想:
针对input来看,其实就是有一棵或多棵现成的树,目的就是把每个节点的左右两个分支配平。
由于子节点的配平会影响到父节点的配平,所以很自然的就想到了用递归的方法来解这个问题。
整个递归过程就是一个树的后序遍历过程。
调整了代码缩进,加入必要注释
package PhoneInterview;import java.io.BufferedReader;import java.io.File;import java.io.FileNotFoundException;import java.io.FileReader;import java.io.IOException;public class Balance { // 每个天秤的属性 public boolean balanced = false;// 判断是否已经配平 public int nodeID;// 天秤id public Balance[] leftChild = null;// 左子树的天秤数组 public Balance[] rightChild = null;// 右子树的天秤数组 public int balanceWeight = 10;// 自身重量 public int leftWeight = 0;// 左子树总重量 public int rightWeight = 0;// 右子树总重量 public int adding = 0;// 为了配平,还需添加多少重量的砝码 // 递归函数 public static int balancing(Balance node) { // 判断天秤左边是否有子天秤 if (node.leftChild != null) { for (int i = 0; i < node.leftChild.length; i++) node.leftWeight += balancing(node.leftChild[i]); } // 判断天秤右边是否有紫天秤 if (node.rightChild != null) { for (int i = 0; i < node.rightChild.length; i++) { node.rightWeight += balancing(node.rightChild[i]); } } // 天秤左右子树都计算完毕,来计算当前天秤左右重量差 node.adding = Math.abs(node.leftWeight - node.rightWeight); // 标记该天秤已经配平 node.balanced = true; return node.balanceWeight + node.leftWeight + node.rightWeight + node.adding; } // 主函数 public static void main(String[] args) { int N = 0; Balance[] Balances; BufferedReader br; try { // 载入数据,初始化过程,无视这个file name吧 br = new BufferedReader(new FileReader("input00.txt")); // load第一行,得到天秤总数 String string = br.readLine(); N = Integer.parseInt(string); // 初始化Balances数组 Balances = new Balance[N]; int i = 0; for (int k = 0; k < N; k++) { Balances[k] = new Balance(); Balances[k].nodeID = k; } /* * 开始一行一行的读入数据,每两行一个循环 */ while ((string = br.readLine()) != null) { // 天秤左半部分初始化 String[] strs = string.split(" "); Balances[i].leftWeight = Integer.parseInt(strs[0]); /* * 这里是根据数据格式来的 如果length超过1,那就代表除了自重的值以外 还有放置的子天秤 * 遂初始化子树上的数组,添加对子天平的引用,方便日后调用 */ if (strs.length != 1) Balances[i].leftChild = new Balance[strs.length - 1]; for (int j = 1; j < strs.length; j++) { Balances[i].leftChild[j - 1] = Balances[Integer .parseInt(strs[j])]; } // 天秤右半部分初始化 string = br.readLine(); String[] strs1 = string.split(" "); Balances[i].rightWeight = Integer.parseInt(strs1[0]); // 同理左半部分的操作 if (strs1.length != 1) Balances[i].rightChild = new Balance[strs1.length - 1]; for (int j = 1; j < strs1.length; j++) { Balances[i].rightChild[j - 1] = Balances[Integer .parseInt(strs1[j])]; } i++; } // 初始化完毕,开始配平 for (i = 0; i < Balances.length; i++) { if (!Balances[i].balanced) Balance.balancing(Balances[i]); } // 输出结果 for (int m = 0; m < Balances.length; m++) { if (Balances[m].leftWeight < Balances[m].rightWeight) System.out.println(m + ": " + Balances[m].adding + " 0"); else if (Balances[m].leftWeight >= Balances[m].rightWeight) System.out.println(m + ": 0 " + Balances[m].adding); } } catch (FileNotFoundException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } }
倚天杖
TA贡献1828条经验 获得超3个赞
个人解法如下:
假设天平之间没有嵌套关系(即不会出现1在2左边,2在1左边这种情况)
为了使一个天平平衡,首先得把放在其左边或右边的天平配平
假设力矩为0,天平左右两边重量不等时在少的一边用砝码补上重量
如果符合以上,题目综合性不错,涉及了简单树的遍历以及贪心算法
下面是代码(脑袋不好使状态,不知道弄对了没——反正代码风格成渣渣了)
import sys def set_balance(id): if not balanced[id]: for i in lcld[id]: lw[id] += set_balance(i) for i in rcld[id]: rw[id] += set_balance(i) if lw[id] < rw[id]: ladd[id] = rw[id] - lw[id] lw[id] += ladd[id] elif lw[id] > rw[id]: radd[id] = lw[id] - rw[id] rw[id] += radd[id] balanced[id] = True return lw[id] + rw[id]n = int(sys.stdin.readline()) lw = [5 for i in range(n)]rw = [5 for i in range(n)]lcld = [[] for i in range(n)]rcld = [[] for i in range(n)]for i in range(n): row = [int(v) for v in sys.stdin.readline().split()] lw[i] += row[0] for v in row[1:]: lcld[i].append(v) row = [int(v) for v in sys.stdin.readline().split()] rw[i] += row[0] for v in row[1:]: rcld[i].append(v) balanced = [False for i in range(n)]ladd = [0 for i in range(n)]radd = [0 for i in range(n)]for i in range(n): get_balance(i) for i in xrange(n): print "%d:%d %d" % (i, ladd[i], radd[i])
添加回答
举报
0/150
提交
取消