华为机试题：实现一个简易的银行叫号系统

排号机

描述:

实现一个简易的银行排号叫号系统

get 取号 示例:”get”或”get vip”

call 叫号 示例:”call”

delete 删除号码 示例:”delete 5”

count 获取当前排队总人数 示例:”count”

countN 获取号码N以前的排队人数 示例:”countN”

reset 重置排号机 示例:”reset”

quit 退出排号机 示例:”quit”

运行时间限制: 无限制

内存限制: 无限制

输入:

每行只会有一条输入（比如：C语言可使用gets函数获取一行输入命令的字符串）。

1、若输入不符合要求（如：命令字非法，或其他认为输入的错误）均需输出”error”

2、每条输出后使用换行符隔开（如后面示例）

输出:

1）取号。可获取普通号和vip号码。如初始状态，输入”get”,则获取普通号码,执行结果为”1”,如再次输入”get vip”,则获取VIP号码,执行结果为”vip 2”。如果末尾的2号被删除，则再次调用”get”时应输出”2”

VIP号码有绝对的优先级。普通号和vip号码统一编号，取号均为连续号码。号码从1开始编号，最大为100000.

2）叫号。获取当前应该处理用户的号码。例如当前排队号码为1 2 3 4 5 7，当输入”call”，执行结果为”1”,如1为vip号码，则为”vip 1”.如果再连续调用6次，第六次执行结果应为”error”

3）删除号码。客户不想办理时可删除号码，叫号时则跳过此号码。例如当前排队号码为1 2 3 4 5,输入”delete 5”，执行结果为”5”，如果5为vip则显示”vip 5”。再次输出”delete 5”，执行结果为”error”

4）获取当前排队总人数。获取当前排队人数。例如当前排队号码为1 2 3 4 5 6,执行结果为”6”

5）获取在某个号码之前排队的总人数。例如当前排队号码为1 2 3 4 5 7，输入”countN 7”,执行结果为”5”

6、重置排号机。例如输入”reset”,则重置排号机，进入初始状态，无需输出。

7、退出排号机。例如输入”quit”,则退出排号机，无需输出。

样例输入:

get

get

get

get vip

count

countN 1

call

quit

样例输出:

1

2

3

vip 4

4

1

vip 4

我的答案（java）：

import java.util.Scanner;
import java.util.TreeSet;

public class Main{

private final String VIP="vip";

private long num=1;// 号码 1~100000
private TreeSet<Long> vipNums=new TreeSet<Long>();// 保存vip号码
private TreeSet<Long> queue=new TreeSet<Long>();// 队列
private boolean quit;// 是否退出

public String[] input(Scanner sc){

String line=null;
String[] comd=null;
try{
line=sc.nextLine();
comd=line.split(" ");// 拆分命令
}catch(Exception e){
//          e.printStackTrace();
comd=null;
printError();
}finally{
//          sc.close();
}
return comd;
}

public void handle(String[] comd) throws Exception{
if(comd==null||comd.length==0){
printError();
}
String str=comd[0];// 提取命令

if(str.equals("get")){

if(num<1||num>100000){// ERROR
printError();
}

if(comd.length==2){
String str2=comd[1];
if(str2.equals(VIP)){// get vip num
System.out.println(VIP+" "+num);
vipNums.add(num);
}else{
printError();
}
}else{
System.out.println(num);// get common num
}
queue.add(num);// 把取的号加入队列
num++;
}else if(str.equals("call")){
Long first=null;
// vip first
if(vipNums.size()>0){
first=vipNums.first();
System.out.println(VIP+" "+first);
vipNums.remove(first);
queue.remove(first);
}else if(queue.size()>0){
first=queue.first();
System.out.println(first);
queue.remove(first);
}else{
printError();
}
}else if(str.equals("delete")){
if(comd.length==2){
String str2=comd[1];
Long numToDel=Long.parseLong(str2);
if(queue.contains(numToDel)){
if(vipNums.contains(numToDel)){
System.out.println(VIP+" "+numToDel);
vipNums.remove(numToDel);
}else{
System.out.println(numToDel);
}
if(queue.last().equals(numToDel)){
num--;
}
queue.remove(numToDel);

}else{
printError();
}
}else{
printError();
}
}else if(str.equals("count")){
System.out.println(queue.size());
}else if(str.equals("countN")){
if(comd.length==2){
String str2=comd[1];
Long numToCount=Long.parseLong(str2);
if(queue.contains(numToCount)){
int count=0;
for(Long number : queue){
if(number<numToCount){
count++;
}
}
System.out.println(count);
}else{
printError();
}
}else{
printError();
}
}else if(str.equals("reset")){
num=1;
vipNums.clear();
queue.clear();
}else if(str.equals("quit")){
quit=true;
}else{
printError();
}
}

public void printError(){
System.out.println("error");
}
public static void main(String[] args){

Scanner sc=new Scanner(System.in);

Main m=new Main();

while(!m.quit){
String[] comd=m.input(sc);
try {
if(comd==null){
throw new Exception("comd is null");
}
m.handle(comd);
} catch (Exception e) {
m.printError();
//              e.printStackTrace();
}
}
sc.close();
}
}