题目:阿里巴巴找黄金宝箱
前言
本人最近再练习算法,所以会发布一些解题思路,希望大家多指教
一、题目描述
一贫如洗的樵夫在去砍柴的路上,无意中发现了强盗集团的藏宝地,藏宝地有编号从0~N的箱子,每个箱子上面贴有一个数字,箱子中可能有一个黄金宝箱。黄金宝箱满足排在它之前的所有箱子数字和等于排在它之后的所有箱子数字和;第一个箱子左边部分的数字和定义为0;最后一个宝箱右边部分的数字和定义为0。请帮阿里巴巴找到黄金宝箱,输出第一个满足条件的黄金宝箱编号,如果不存在黄金宝箱,请返回-1。
二、输入描述
箱子上贴的数字列表,使用逗号分隔,例如1,-1,0。
宝箱的数量不小于1个,不超过10000。
宝箱上贴的数值范围不低于-1000,不超过1000。
三、输出描述
第一个黄金宝箱的编号。
四、java代码
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
List<Node> list = new ArrayList<>();
// 输入信号字符串
int[] signal = Arrays.stream(sc.nextLine().split(",")).mapToInt(Integer::parseInt).toArray();
//通过双向指针来保存每个节点,初始化前置节点
Node pre = new Node(1001);
for (int i=0; i< signal.length; i++){
Node node = new Node(signal[i]);
if(pre.getValue() == 1001){
node.setPre(null);
} else{
node.setPre(pre);
pre.setNext(node);
}
pre = node;
list.add(node);
}
//判断每个元素前面元素的和和后面元素的和是否相等
int i = 0;
boolean flag = false;
for(Node node : list){
//初始化前置节点的和
int preNum = 0;
//初始化后置节点的和
int nextNum = 0;
Node preNode = node;
if(preNode.getPre() != null){
preNode = preNode.getPre();
while (preNode != null){
preNum += preNode.getValue();
preNode = preNode.getPre();
}
}
Node nextNode = node;
if(nextNode.getNext() != null){
nextNode = nextNode.getNext();
while (nextNode != null){
nextNum += nextNode.getValue();
nextNode = nextNode.getNext();
}
}
//前后节点的和相等,则找到宝箱,输出对应下标
if(preNum == nextNum){
System.out.println(i);
flag = true;
break;
}
i++;
}
//没有找到宝箱,则输出-1;
if(!flag){
System.out.println(-1);
}
}
static class Node{
private Node pre;
private Node next;
private int value;
public Node getPre() {
return pre;
}
public void setPre(Node pre) {
this.pre = pre;
}
public Node getNext() {
return next;
}
public void setNext(Node next) {
this.next = next;
}
public int getValue() {
return value;
}
public void setValue(int value) {
this.value = value;
}
public Node(int value) {
this.value = value;
}
}
五、测试用例
输入:1,5,-1,7,-1
输出:
说明,1+5 = 7+(-1),所以-1满足宝箱条件,输出编号为下标值2