1. 程式人生 > >演算法競賽入門經典例題-螞蟻

演算法競賽入門經典例題-螞蟻

題目描述:一根長度為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。