今天闲来无事,整理了一下参加蓝桥杯算法竞赛的题目,记得当时其中有一道是蚂蚁感冒的问题,原题如下:
//蓝桥杯 蚂蚁感冒问题:
/*
标题:蚂蚁感冒
长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
请严格按要求输出,不要画蛇添足地打印类似:“请您输入...” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准,不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include <xxx>, 不能通过工程设置而省略常用头文件。
提交时,注意选择所期望的编译器类型。
*/ 我整理的代码,跑例子没问题,具体答案我不知道,把研究的结果记录一下:
#include<iostream>
#include<algorithm>
using namespace std;
typedef struct Node
{
int flag; // 1 表示向右,0 表示向左
int width; // 表示蚂蚁距离左边的距离
bool ganmao; // true 表示蚂蚁感冒了,false 表示没有感冒
}Node;
Node* mayi;
// 求所有向左的蚂蚁中距离左端距离的最大值
// 和所有向右的蚂蚁中距离左端的最小值
bool operator <( Node a,Node b) // 根据width sort
{
if(a.width<=b.width)
return true;
else
return false;
}
int getMaxZuo(int n,int&min)
{
int max=0;
min=100;
for(int i=0;i<n;i++)
{
if(mayi[i].flag==0&& mayi[i].width>max)
max=mayi[i].width;
if(mayi[i].flag==1&& mayi[i].width<min)
min=mayi[i].width;
}
return max;
}
// 打印所有的蚂蚁情况
void Print(int n)
{
for(int i=0;i<n;i++)
{
cout<<"flag"<<mayi[i].flag<<" width:"<<mayi[i].width<<" ganmao:"<<mayi[i].ganmao<<endl;
}
}
// 得到所有感冒的蚂蚁的数量
int ALLGanmao(int n)
{
int max,min;
int countganmao=1;
max=getMaxZuo(n,min);
//cout<<"max:"<<max<< " min:"<<min<<endl;
while(max>min) // 只有当向左的蚂蚁中的距离左端最大值 小于所有向右的蚂蚁中的最小值时程序截止
{
//cout<<"max:"<<max<< " min:"<<min<<endl;
for(int i=0;i<n;i++)
{
if(mayi[i].flag==0) // 向左
mayi[i].width--;
else
mayi[i].width++;
}
for(int j=0;j<n-1;j++)
{
if(mayi[j].width==mayi[j+1].width) // 代表蚂蚁碰头
{
int t=mayi[j].flag;
mayi[j].flag=mayi[j+1].flag; // 首先方向对调
mayi[j+1].flag=t;
if(mayi[j].ganmao!=mayi[j+1].ganmao) // 如果两只蚂蚁的感冒状态不一致、
{
countganmao++;
mayi[j].ganmao=true;
mayi[j+1].ganmao=true; // 都要感冒!!!! 传染
}
}
}
cout<<"这次所有蚂蚁的状态是:***********"<<endl;
Print(n);
cout<<"***********************"<<endl;
max=getMaxZuo(n,min); // 状态改变后再次判断程序是否可以截止
}
return countganmao;
}
int main()
{
int n;
cin>>n;
mayi=new Node[n];
for(int i=0;i<n;i++)
{
int m;
cin>>m;
if(i==0)
mayi[i].ganmao=true;
else
mayi[i].ganmao=false;
if(m>0)
{
mayi[i].flag=1;
mayi[i].width=m;
}
else
{
mayi[i].flag=0;
mayi[i].width=-m;
}
}
sort(mayi,mayi+n);
Print(n);
cout<<"all ganmao is:"<<ALLGanmao(n)<<endl;
system("pause");
return 0;
}
运行例子的结果截图: