Input示例
5 4 0 3 1 2 0 1 0 5 0
Output示例
2
死者:2 1 3
栈模拟
把向右的鱼看成左括号进栈,向左的鱼看成右括号出栈。答案为没被吃掉的向左的鱼和最后剩在栈里的向右的鱼的个数之和。
#include<iostream> #include<cstring> #include<cstdio> #include<cmath> #include<vector> #include<set> #include<stack> #include<algorithm> using namespace std; stack<int> s; int n; int a[100010],dir[100010]; int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d%d",a+i,dir+i); } int ans=0; for(int i=1;i<=n;i++) { if(dir[i]==0) { while(!s.empty() && s.top()<a[i]) s.pop(); if(s.empty()) ans++; } else { s.push(a[i]); } } cout<<ans+s.size()<<endl; }