洛谷 P5661 公交换乘 & [NOIP2019普及组] (模拟)
传送门
解题思路
先把所有的数据读下来。
对于地铁,答案直接加,然后把编号放入一个数组a内。
对于公交车,从前往后枚举a数组,然后找到出现最早的且符合价钱大于等于公交车的价钱,然后把这个数删除(变为0)。
然后再考虑有效期是45分钟,为了优化时间,我们可以每一次把数组看做一个队列,当a[first]是0或者时间超过了45分钟时,first++。
这样就保证了数组内的数不超过45个。
最后看一眼时间复杂度,O(NK),K为不超过45。
轻松A掉。
AC代码
1 #include<iostream> 2 #include<cstdio> 3 using namespace std; 4 const int maxn=100005; 5 int n,x,p[maxn],t[maxn],a[maxn],cnt,fir=1; 6 long long ans; 7 int main() 8 { 9 cin>>n; 10 for(int i=1;i<=n;i++){ 11 scanf("%d%d%d",&x,&p[i],&t[i]); 12 while(fir<=cnt&&(a[fir]==0||t[i]-t[a[fir]]>45)) fir++; 13 ans+=p[i]; 14 if(x==0){ 15 a[++cnt]=i; 16 }else{ 17 for(int j=fir;j<=cnt;j++){ 18 if(a[j]==0) continue; 19 if(p[i]<=p[a[j]]){ 20 a[j]=0; 21 ans-=p[i]; 22 break; 23 } 24 } 25 } 26 } 27 cout<<ans<<endl; 28 return 0; 29 }
//CSP2019普及组 t2