演算法競賽入門經典例題-螞蟻
阿新 • • 發佈:2019-01-29
題目描述:一根長度為L的木棍上有n只螞蟻,每隻螞蟻要麼朝左爬,要麼朝右爬,速度為1釐米/秒。二者同時掉頭,給出每隻螞蟻的初始位置和朝向,計算T秒之後每隻螞蟻的位置。
輸入格式:
輸入的第一行為資料組數。每組資料的第一行為3個正整數L,T,n(0<=n<=10000);以下n行每行描述一隻螞蟻的初始位置。其中,整數x為螞蟻距離左端的距離(單位:釐米),字母表示朝向(L表示朝左,R表示朝右)
輸出格式:
對於每組資料,輸出n行,按輸入順序輸出每隻螞蟻的位置和朝向(Turning 表示正在碰撞)。在第T秒之前已經掉下木棍的螞蟻(正好爬到木棍邊緣的不算)輸出Fell off
樣例輸入:
2
10 1 4
1 R
5 R
3 L
10 R
10 2 3
4 R
5 L
8 R
樣例輸出:
Case #1:
2 Turning
6 R
2 Turning
Fell off
Case #2:
3 L
6 R
10 R
*分析:每隻螞蟻在木棍上都有自己的相對位置, 如果兩隻螞蟻碰頭就會改變自己的朝向,所以他們在木棍上的相對位置是不會改變的,給所有螞蟻的位置標記大小並從小到大排序,設定初始陣列和終態陣列表示螞蟻的狀態,設定一個order陣列表示螞蟻被放置進去的順序,在輸出結果的時候根據這個順序輸出,
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=10000+5;
const char dirName[][10]={"L","Turning","R"};
struct Ant{
int id; //輸入順序
int p; //位置
int d; //朝向 -1 0 1
bool operator <(const Ant& a) const{
return p<a.p;
}
}before[maxn],after[maxn];
int order[maxn];
int main()
{
int k=0;
scanf("%d",&k);
//開始狀態和結束狀態
for(int kase=1;kase<=k;kase++)
{
int L,T,n;
scanf("%d%d%d",&L,&T,&n);
for(int i=0;i<n;i++)
{
int p,d;
char c;
scanf("%d%c",&p,&c);
d=(c=='L'?-1:1);
before[i]=(Ant){i,p,d};
after[i]=(Ant){0,p+T*d,d};
}
//計算order陣列
sort(before,before+n);
for(int i=0;i<n;i++)
order[before[i].id]=i;
//計算終態
sort(after,after+n);
for(int i=0;i<n-1;i++)
if(after[i].p==after[i+1].p)
after[i].d=after[i+1].d=0;
printf("Case #%d:\n",kase);
//輸出結果
for(int i=0;i<n;i++)
{
int a=order[i];
if(after[a].p<0||after[a].p>L)
printf("Fell off\n");
else
printf("%d %s\n",after[a].p,dirName[after[a].d+1]);
}
printf("\n");
}
return 0;
}
要注意的是,為什麼要用到order陣列,因為終態陣列沒有輸入順序,在最後輸出結果的時候需要按照這個order把結果順序輸出。
碰撞的時候記為0,向左為-1,向右為1。