今天遇到个题,题列如下:
火车站的列车调度铁轨的结构如下图所示。
两端分别是一条入口(Entrance)轨道和一条出口(Exit)轨道,它们之间有N
条平行的轨道。每趟列车从入口可以选择任意一条轨道进入,最后从出口离开。在图中有9趟列车,在入口处按照{8,4,2,5,3,9,1,6,7}的顺序排队等待进入。如果要求它们必须按序号递减的顺序从出口离开,则至少需要多少条平行铁轨用于调度?
输入格式:
输入第一行给出一个整数N
(2 ≤ N
≤105),下一行给出从1到N
的整数序号的一个重排列。数字间以空格分隔。
输出格式:
在一行中输出可以将输入的列车按序号递减的顺序调离所需要的最少的铁轨条数。
输入样例:
9
8 4 2 5 3 9 1 6 7
输出样例:
4
一开始我的想法是单调栈,后面发现并不是这个问题,而应该是一个模拟题,于是我傻乎乎的去模拟每个序号的火车的出入,并且使用了二维数组进行存储,后面发现并不太行,会超时。
代码如下:
from collections import deque
n = int(input())
array = list(map(int,input().split()))
stack = []
ans = 1
place = n
for a in array:
flag = False
for i,s in enumerate(stack):
if not s:
stack.pop(i)
continue
while s and s[0] == place:
s.popleft()
place -= 1
if s[-1] > a:
s.append(a)
flag = True
break
if not flag:
stack.append(deque([a]))
ans = max(ans,len(stack))
print(ans)
但是我们注意到,每个管道的情况其实只与最小值和最大值有关,最小值判断是否能够将新元素进栈,最大值判断目前的元素是否能够弹出。
于是我们可以只存储每个通道的最小值,最大值我们并不需要进行操作,因为最大值本身是可能弹出的,那干脆我们都弹出即可,反正也不影响我目前这个通道是否存在。如果目前新的值大于所有通道的最小值,那肯定需要新开一个通道,如果小于,那就跟在与当前值差距最小的通道交换即可。
具体代码如下:
#include<bits/stdc++.h>
using namespace std;
int main(){
int n;
cin >> n;
set<int>sc;
for(int i=0;i<n;i++){
int k;
cin >> k;
set<int>::iterator it = sc.lower_bound(k);
if(it != sc.end()){
sc.erase(it);
sc.insert(k);
}
else sc.insert(k);
}
cout << sc.size();
return 0;
}
sc.insert(k);
}
else sc.insert(k);
}
cout << sc.size();
return 0;
}