算法提高 双十一抢购 java

试题 算法提高 双十一抢购

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　一年一度的双十一又来了，某网购网站又开始了半价销售的活动。
　　小G打算在今年的双十一里尽情地购物，以享受购买的极度快感，她已经列好了她想买的物品的列表。
　　当然小G并不是出身富贵家庭，所以她网银里的钱只是一个有限的整数S（单位：元）。
　　这次抢购她打算遵循这三个原则选择每一个物品：
　　1.先买能“赚”最多的；
　　2.在“赚”一样多的情况下，先买最便宜的（这样买的东西就可能更多了）；
　　3.在前两条里都判断不了购买顺序的话，先购买在列表里靠前的。
　　（由于网站里还是有一部分商品并没有打五折，所以2的情况（“赚”的钱数为0）是完全可能发生的）
　　现在，在双十一的这一天，你要帮小G编写一个程序，来看看她应该去买她列表里的哪些物品。（总价格不要超过S哦）
　　要是帮她写好这个程序的话，或许你能在光棍节这一天里赢得她的芳心哦~
输入格式
　　输入共N+1行。
　　第一行包含两个整数S和N，S表示小G的可用金额，N表示她看上的物品个数。
　　接下来N行，对应每一个物品，每行有两个整数a和b，a是物品的原价（单位：元），
b为0或1，若b为0，则此物品不半价，若b为1，则此物品半价销售。
输出格式
　　输出共一行，为小G要买的物品序号（从1开始），用空格隔开，注意按序号从小到大输出。
　　若小G一件都买不了，则输出0.
样例输入
30 3
5 0
10 1
10 1
样例输出
2 3
样例输入
10 3
11 0
21 1
100 1
样例输出
0
数据规模和约定
　　0<S<=10000,0<N<=1000,每一个a和b满足0<a<=1000且b=0或1。

import java.util.*;

public class ADV_289双十一购物 {

	public static void main(String[] args) {
		Scanner sc  = new Scanner(System.in);
        int s =sc.nextInt();
        int n=sc.nextInt();
        Shop shop[] = new Shop[n];
        for (int i = 1; i <=n; i++) {
			int a=sc.nextInt();
			int b=sc.nextInt();
			shop[i-1]=new Shop(i,a,b);
		}
        Arrays.sort(shop);
        TreeSet<Integer> set = new TreeSet<Integer>();
       for(Shop sho:shop) {
    	   if(sho.d==1&&s>=sho.price*0.5) {
    		   s-=sho.price*0.5;
    		   set.add(sho.i);
    	   }else if(sho.d==0&&s>=sho.price) {
    		   s-=sho.price;
    		   set.add(sho.i);
    	   }
       }
       
        for(Integer tt:set) {
        	System.out.print(tt+" ");
        }
        if(set.size()==0)System.out.println(0);
	}

	
  static class Shop implements Comparable<Shop>{
	  int i;//商品序号
	  double price;//商品价格
	  int d;//是否打折
	
	public int compareTo(Shop o) {
	//判断Shop和this.shop是否是同一种类型商品(打折和不打折)
		if(this.d==o.d) {
			//比较谁赚的多
			if(this.price==o.price)return this.i-o.i;
			if(this.d==1)return (int) (o.price-this.price);
			return (int) (this.price-o.price);
		}
		if(this.d>o.d)return -1;
		return 1;
	}

	public Shop(int i, double price, int d) {
		super();
		this.i = i;
		this.price = price;
		this.d = d;
	}
	  
	  
  }	
}