AcWing 1211. 蚂蚁感冒
长 100 厘米的细长直杆子上有 n 只蚂蚁。
它们的头有的朝左,有的朝右。
每只蚂蚁都只能沿着杆子向前爬,速度是 1 厘米/秒。
当两只蚂蚁碰面时,它们会同时掉头往相反的方向爬行。
这些蚂蚁中,有 1 只蚂蚁感冒了。
并且在和其它蚂蚁碰面时,会把感冒传染给碰到的蚂蚁。
请你计算,当所有蚂蚁都爬离杆子时,有多少只蚂蚁患上了感冒。
输入格式
第一行输入一个整数 n, 表示蚂蚁的总数。
接着的一行是 n 个用空格分开的整数 Xi, Xi 的绝对值表示蚂蚁离开杆子左边端点的距离。
正值表示头朝右,负值表示头朝左,数据中不会出现 0 值,也不会出现两只蚂蚁占用同一位置。
其中,第一个数据代表的蚂蚁感冒了。
输出格式
输出1个整数,表示最后感冒蚂蚁的数目。
数据范围
1<n<50,
0<|Xi|<100
输入样例1:
3
5 -2 8
输出样例1:
1
输入样例2:
5
-10 8 -20 12 25
输出样例2:
3
题意:在一根杆子上有若干蚂蚁,以数字代表蚂蚁坐标位置,正负代表移动方向,其中一只蚂蚁为感冒状态,触碰到的其他蚂蚁也会被感染为感冒状态,同时两只蚂蚁的移动方向对撞改变,直至全部蚂蚁移动出杆,计算感冒蚂蚁数量
上次做到蚂蚁的题还是堆和队列,这题主要考察的是思维模式的转换,如果真按照题中所说的每当对撞时方向改变感冒状态也改变,计算是十分复杂的
由于蚂蚁并没有特殊性,每只蚂蚁间的区别无非是感冒与否而已,“对撞”这个动作可以看似为“穿过”,碰撞后蚂蚁更新状态即可,继续按照原有方向运动也不会影响到结果,由此可以分别计数初始感冒蚂蚁左右两侧的蚂蚁数量
设res=1为感冒蚂蚁数,设 l 为初始蚂蚁左侧向右移动的蚂蚁数,设 r 为右侧向左移动的蚂蚁数(由于速度相同,所以无需在意右侧向右或左侧向左移动的蚂蚁,它们在移动过程中会直接出杆,即使发生碰撞也是两只未感冒的蚂蚁,不对结果产生影响)。
当初始蚂蚁移动方向为左时,则记录其左侧向右的蚂蚁数 l ,他们必将会与初始蚂蚁碰撞,因此直接在res中加上 l ,同时由于会有向右移动的感冒蚂蚁,会与右侧向左移动的蚂蚁碰撞,所以再将res加上 r ,即是本题结果,反之亦然
若初始蚂蚁移动方向上没有将会与之发生碰撞的蚂蚁,如 l=0 ,则它将直接出杆,res为1不变
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
import java.util.Vector;
public class Main {
static Scanner tab = new Scanner(System.in);
static BufferedWriter tabb = new BufferedWriter(new OutputStreamWriter(System.out));
static int N = 110000;
public static void main(String[] args) throws IOException {
int n=tab.nextInt();
int l=0,r=0;
int x=tab.nextInt();
n--;
while(n-->0) {
int t=tab.nextInt();
if(Math.abs(t)>Math.abs(x)&&t<0)
r++;
if(Math.abs(t)<Math.abs(x)&&t>0)
l++;
}
if((x<0&&l==0)||(x>0&&r==0))
System.out.println("1");
else
System.out.println(l+r+1);
}
}