坠落的蚂蚁
一根长度为 1 米的木棒上有若干只蚂蚁在爬动。 它们的速度为每秒一厘米或静止不动,方向只有两种,向左或者向右。
如果两只蚂蚁碰头,则它们立即交换速度并继续爬动。 三只蚂蚁碰头,则两边的蚂蚁交换速度,中间的蚂蚁仍然静止。
如果它们爬到了木棒的边缘(0 或 100 厘米处)则会从木棒上坠落下去。 在某一时刻蚂蚁的位置各不相同且均在整数厘米处(即
1,2,3,…99 厘米),有且只有一只蚂蚁 A 速度为 0,其他蚂蚁均在向左或向右爬动。
给出该时刻木棒上的所有蚂蚁位置和初始速度,找出蚂蚁 A 从此时刻到坠落所需要的时间
输入格式
第一行包含一个整数表示蚂蚁的个数 N,之后共有 N 行,每一行描述一只蚂蚁的初始状态。
每个初始状态由两个整数组成,中间用空格隔开,第一个数字表示初始位置厘米数 P,第二个数字表示初始方向,−1 表示向左,1 表示向右,0 表示静止。
输出格式
蚂蚁 A 从开始到坠落的时间。若不会坠落,输出 Cannot fall!
。
数据范围:
2≤N≤99, 1≤P≤99
输入样例:
4
10 1
90 0
95 -1
98 -1
输出样例:
98
思路
最初的思路是模拟,但后来想了想太麻烦
参考了其他几位博主的解答,思路大体相同。以下是几个要点:
- 两只蚂蚁交换速度,相当于身份互换,二者对穿。所以,两只与A无关的蚂蚁碰面,并不影响A的最终结果。
- 由上一条推知,A的左边蚂蚁若向左走,右边蚂蚁若向右走,不会影响A,且最终会掉落下去,不用考虑。
- A的左边蚂蚁向右走,A的右边蚂蚁向左走,无论其与A相遇的先后顺序,最终A的位置不变。
(解释:想象有 B、A、C 3只蚂蚁,BA先碰面,B留在A的位置,A向右走,AC碰面,交换速度后,A向左走再次与B碰面,留在最初A的位置) - 由上一条可推知,若A左右两边相向而行的蚂蚁数量相等,则A留在原位,不会掉下去。
- 如果两边相向而行的的蚂蚁数量不同,根据上一条,最终会影响到A的蚂蚁应该是左边蚂蚁与右边蚂蚁抵消之后剩下的蚂蚁中,最靠近A的第一只,因此,A掉下去的时间,应该就是这个第一只蚂蚁从它的最初位置走到边缘的时间。
因此,做法如下:用pair将蚂蚁的位置和速度打包成一个数据,利用set自排序的特征,用一个set存储整个蚂蚁群,再分别用两个vector存储A左边向右走的蚂蚁和A右边向左走的蚂蚁,统计两个set的蚂蚁数量,若相同,则A不会坠落,若左多,则左边第L-R只 (注意:下标为L-R-1,下同) 蚂蚁为第一只;反之右边第L+1只为第一只。
代码
#include<iostream>
#include<cstdlib>
#include<string>
#include<set>
#include<vector>
using namespace std;
/*
题目描述:
坠落的蚂蚁
*/
typedef pair<int,int> ant;
set<ant> a;
vector<ant> l,r;
int pos_A;
int main(){
int n;
int x,y;
cin>>n;
while(n--){
cin>>x>>y;
if(!y) pos_A = x;
a.insert({x,y});//这种打包方式还不熟,借鉴了一下别人的操作
}
for(auto i : a){ //要注意,set插入为insert,vector插入为push_back,另外set不支持随机访问
if(i.first < pos_A && i.second == 1) l.push_back(i);
else if(i.first > pos_A && i.second == -1) r.push_back(i);
}
if(l.size() == r.size()) cout<<"Cannot fall!"<<endl;
else if(l.size() > r.size()) cout<<100 - l[l.size()-r.size()-1].first <<endl;
else cout<<r[l.size()].first<<endl;
system("pause");
return 0;
}