长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左,有的朝右。 每只蚂蚁都只能沿着杆子向前爬,速度是1厘米/秒。 当两只蚂蚁碰面时,它们会同时掉头往相反的方向爬行。 这些蚂蚁中,有1只蚂蚁感冒了。并且在和其它蚂蚁碰面时,会把感冒传染给碰到的蚂蚁。 请你计算,当所有蚂蚁都爬离杆子时,有多少只蚂蚁患上了感冒。
【数据格式】
第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。
接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100), Xi的绝对值,表示蚂蚁离开杆子左边端点的距离。正值表示头朝右,负值表示头朝左,数据中不会出现0值,也不会出现两只蚂蚁占用同一位置。其中,第一个数据代表的蚂蚁感冒了。
要求输出1个整数,表示最后感冒蚂蚁的数目。
例如,输入:
3
5 -2 8
程序应输出:
1
再例如,输入:
5
-10 8 -20 12 25
程序应输出:
3
资源约定:
峰值内存消耗 < 256M
CPU消耗 < 1000ms
解题思路
首先明白两只蚂蚁碰撞之后掉头和穿行过去是一样的,可以把穿行看做碰撞后掉头了,然后两个蚂蚁交换了,而是哪一只蚂蚁对结果不影响。
那么,假如第一只感冒蚂蚁向右走,那么碰到所有想左走的都会被感染,而感染后的蚂蚁必定是向左走的,那么他会把左边向右走的都感染了。
向左走的也是这样。
所以 ans = 左边向右走的 + 右边向左走的 + 1(本身)。
当然还有特殊情况,第一只感染的向右走,右边的都在向右走,那么速度一样的话它不会感染其他所有,所以 ans = 1,相反也是。
这是我个人思路,欢迎指正
nyoj题目:http://acm.nyist.net/JudgeOnline/problem.php?pid=990
我的代码:
#include <cstdio>
#include <algorithm>
using namespace std;
struct Node
{
int x;
int dis;
int num;
};
Node a[100];
int comp(Node a1,Node a2)
{
if(a1.x!=a2.x)
return a1.x<a2.x;
}
int main()
{
int n,x;
while(~scanf("%d",&n))
{
for(int i=0;i<n;i++)
{
scanf("%d",&x);
a[i].dis=x<0?-1:1;
a[i].x=abs(x);
a[i].num=i+1;
}
sort(a,a+n,comp); //排序
int tmp=0,left=0,right=0,ans=0;
for(int i=0;i<n;i++) //左边向右走的
{
if(a[i].num==1)
{
tmp=i;break;
}
if(a[i].dis==1)
left++;
}
for(int i=tmp+1;i<n;i++) //右边向左走的
{
if(a[i].dis==-1)
right++;
}
if(a[tmp].dis==1 && right==0 || a[tmp].dis==-1 && left==0)
ans=1;
else
ans=left+right+1;
printf("%d\n",ans);
}
return 0;
}