UVA1598

思路:每个优先级队列分别存储sell和buy,每个指令后都判断是否产生交易

package test;

import java.util.Comparator;
import java.util.Iterator;
import java.util.PriorityQueue;
import java.util.Queue;
import java.util.Scanner;
import java.util.TreeSet;



public class Test{	
	static class Com{
		static int id = -1;
		static {
			id++;
		}
		int q = 0;		
	}
	
	static class Sell extends Com{
		int p,q;
		int sid = id; 
		public Sell(int p, int q) {
			super();
			this.p = p;
			this.q = q;
		}		
	}
	
	static class Buy extends Com{
		int p,q;
		int bid = id; 
		public Buy(int p, int q) {
			super();
			this.p = p;
			this.q = q;
		}		
	}

	public static void main(String[] args) {
		
		Comparator<Com> OrderIsdn =  new Comparator<Com>(){

			@Override
			public int compare(Com o1, Com o2) {
				if(o1.q < o2.q) return 1;
				return -1;
			}
			  			             
        };  
        Queue<Sell> priorityQueue =  new PriorityQueue<Sell>(100,OrderIsdn);  
        Queue<Buy> priorityQueue2 =  new PriorityQueue<Buy>(100,OrderIsdn);  
		
		Scanner sc = new Scanner(System.in);
		String line = null;
		System.out.println("input:");
		while(!"".equals((line=sc.nextLine().toLowerCase()))){
			if(line.startsWith("buy")){
				int p = Integer.valueOf(line.substring(3, line.indexOf(" ")));
				int q = Integer.valueOf(line.substring(line.indexOf(" ")+1));
				priorityQueue2.add(new Buy(p, q));
			}else if(line.startsWith("sell")){
				int p = Integer.valueOf(line.substring(4, line.indexOf(" ")));
				int q = Integer.valueOf(line.substring(line.indexOf(" ")+1));
				priorityQueue.add(new Sell(p, q));
			}else if(line.startsWith("cancel")){
				int i = Integer.valueOf(line.substring(line.indexOf(" ")+1));
				boolean flag = false;
				for(Sell s:priorityQueue){
					if(s.sid==i){
						priorityQueue.remove(s);
						flag = true;
						break;
					}
				}
				if(!flag){
					for(Sell s:priorityQueue){
						if(s.sid==i){
							priorityQueue.remove(s);
							break;
						}
					}
				}
			}
						
			for(Iterator<Buy> buys = priorityQueue2.iterator();buys.hasNext();){
				Buy temp = buys.next();
				for(Iterator<Sell> sells = priorityQueue.iterator();sells.hasNext();){
					Sell s = sells.next();
					if(temp.q>=s.q){
						if(s.p >= temp.p){
							s.p -= temp.p;							
							buys.remove();
							break;
						}else if(s.p < temp.p){
							temp.p -= s.p;
							sells.remove();					
						}
					}else{
						break;
					}
				}
			}						
		}	
		sc.close();
		
		
	}
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值