有N条鱼每条鱼的位置及大小均不同,他们沿着X轴游动,有的向左,有的向右。游动的速度是一样的,两条鱼相遇大鱼会吃掉小鱼。从左到右给出每条鱼的大小和游动的方向(0表示向左,1表示向右)。问足够长的时间之后,能剩下多少条鱼?Input第1行:1个数N,表示鱼的数量(1 <= N <= 100000)。
第2 - N + 1行:每行两个数A , B ,中间用空格分隔,分别表示鱼的大小及游动的方向(1 <= A <= 10^9,B = 0 或 1,0表示向左,1表示向右)。Output输出1个数,表示最终剩下的鱼的数量。Sample Input
5
4 0
3 1
2 0
1 0
5 0
Sample Output
2
思路:把向右移动的鱼加入栈中,栈中只存向右游动的鱼,再拿向左的鱼与栈顶中的鱼进行比较,如果比向右的鱼大,就将向右的鱼从栈中pop出来,具体代码实现(Java)
//大鱼吃小鱼的编程(利用栈,首先将向右游的加入栈,然后在拿向左的和栈顶的鱼进行比较)
public static void main(String[] args) throws ExecutionException, InterruptedException {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
ArrayList<Integer> list = new ArrayList<>();
while (scanner.hasNextInt()){
list.add(scanner.nextInt());
}
int sum = n;
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < list.size() - 1; i = i+2) {
int size = list.get(i);
int f = list.get(i+1);
if(f == 1){
stack.push(size);
}else{
while(!stack.isEmpty()){
if(size > stack.peek()){
//向右的鱼被吃掉
stack.pop();
sum--;
}else{
//向左的鱼被吃掉
sum--;
break;
}
}
}
}
System.out.println("Sapmle Output:->"+sum);
}```