洛谷 P5661 公交換乘 & [NOIP2019普及組] （模擬）

傳送門

---

解題思路

先把所有的數據讀下來。

對於地鐵，答案直接加，然后把編號放入一個數組a內。

對於公交車，從前往后枚舉a數組，然后找到出現最早的且符合價錢大於等於公交車的價錢，然后把這個數刪除（變為0）。

然后再考慮有效期是45分鍾，為了優化時間，我們可以每一次把數組看做一個隊列，當a[first]是0或者時間超過了45分鍾時，first++。

這樣就保證了數組內的數不超過45個。

最后看一眼時間復雜度，O(NK)，K為不超過45。

輕松A掉。

AC代碼

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=100005;
int n,x,p[maxn],t[maxn],a[maxn],cnt,fir=1;
long long ans;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%d%d%d",&x,&p[i],&t[i]);
        while(fir<=cnt&&(a[fir]==0||t[i]-t[a[fir]]>45)) fir++;
        ans+=p[i];
        if(x==0){
            a[++cnt]=i;
        }else{
            for(int j=fir;j<=cnt;j++){
                if(a[j]==0) continue;
                if(p[i]<=p[a[j]]){
                    a[j]=0;
                    ans-=p[i];
                    break;
                }
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}

//CSP2019普及組 t2