複試上機----poj 1003墜落的螞蟻

題目描述 

https://www.nowcoder.com/practice/fdd6698014c340178a8b1f28ea5fadf8?tpId=40&tqId=21420&tPage=5&rp=5&ru=/ta/kaoyan&qru=/ta/kaoyan/question-ranking

    一根長度爲1米的木棒上有若干只螞蟻在爬動。它們的速度爲每秒一釐米或靜止不動，方向只有兩種，向左或者向右。如果兩隻螞蟻碰頭，則它們立即交換速度並繼續爬動。三隻螞蟻碰頭，則兩邊的螞蟻交換速度，中間的螞蟻仍然靜止。如果它們爬到了木棒的邊緣（0或100釐米處）則會從木棒上墜落下去。在某一時刻螞蟻的位置各不相同且均在整數釐米處（即1，2，3，…99釐米），有且只有一隻螞蟻A速度爲0，其他螞蟻均在向左或向右爬動。給出該時刻木棒上的所有螞蟻位置和初始速度，找出螞蟻A從此時刻到墜落所需要的時間。

輸入描述:

    第一行包含一個整數表示螞蟻的個數N（2<=N<=99），之後共有N行，每一行描述一隻螞蟻的初始狀態。每個初始狀態由兩個整數組成，中間用空格隔開，第一個數字表示初始位置釐米數P（1<=P<=99），第二個數字表示初始方向,-1表示向左，1表示向右，0表示靜止。

輸出描述:

    螞蟻A從開始到墜落的時間。若不會墜落，輸出“Cannot fall!”

示例1

輸入

4
10 1
90 0
95 -1
98 -1

輸出

複製

98

規律很重要，就像打乒乓球！

#include <iostream>
#include<queue>
#include<vector>
#include<string.h>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef struct node {
    int pos;
    int dir;
    bool flag;
} node;

static bool cmp(node a,node b)
{
	return a.pos<b.pos;
}

node a[102];
int n;
int main()
{
    cin>>n;
    int posa,indexa;
    for(int i=0; i<n; i++)
    {
        cin>>a[i].pos>>a[i].dir;
        a[i].flag=true;
    }
    sort(a,a+n,cmp);
    for(int i=0; i<n; i++)
    {
        if(a[i].dir==0)
        {
         	posa=a[i].pos;
         	indexa=i;
         	break;
        }   
    }
    int n1=0,n2=0;
    for(int i=indexa-1; i>=0; i--)
    {
        if(a[i].dir==-1)
            a[i].flag=false;
        else
            n1++;
    }

    for(int i=indexa+1; i<n; i++)
    {
        if(a[i].dir==1)
            a[i].flag=false;
        else
            n2++;
    }
    int count=0;
    int res=0;
    if(n1==n2) cout<<"Cannot fall!"<<endl;
    else
    {
        if(n1>n2)
            for(int i=indexa-1; i>=0; i--)
            {
                if(a[i].flag)
                    count++;
                if(count==n2+1) //這裏差點搞錯了
                {
                    res=100-a[i].pos;
                    break;
                }
            }
        else
            for(int i=indexa+1; i<n; i++)
            {
                if(a[i].flag)
                    count++;
                if(count==n1+1)
                {
                    res=a[i].pos;
                    break;
                }
            }
        cout<<res<<endl;
    }
}