滴滴笔试题 就餐问题

package cn.edu.algorithm.prototype;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        while (scan.hasNext()) {
            int n = scan.nextInt();
            int m = scan.nextInt();

            ArrayList<Integer> desk = new ArrayList<Integer>();
            for (int i = 0; i < n; i++) {
                desk.add(scan.nextInt());
            }
            Collections.sort(desk);

            System.out.println(desk);

            ArrayList<Group> customer = new ArrayList<Group>();
            for (int i = 0; i < m; i++) {
                Group tem = new Group();
                tem.setN(scan.nextInt());
                tem.setC(scan.nextInt());
                customer.add(tem);
            }
            Collections.sort(customer, new Comp().comparator);

            System.out.println(customer);

            int res = 0;
            for (int i = 0; i < m; i++) {//用户批次
                int loc = n;
                if (desk.isEmpty()) {
                    break;
                }

                for (int j = desk.size() - 1; j >= 0; j--) {
                    int k = desk.get(j);
                    int k1 = customer.get(i).num;

                    if (desk.get(j) < customer.get(i).num) {
                        break;
                    } else {
                        loc = j;
                    }
                }

                if (loc != n) {
                    res += customer.get(i).cost;
//                  System.out.println(i+ "     " + customer.get(i).cost + "......" + loc);
                    desk.remove(loc);
                }

            }
            System.out.println(res);
        }
        scan.close();
    }
}

class Group {
    int num;
    int cost;

    public void setN(int n) {
        this.num = n;
    }

    public void setC(int c) {
        this.cost = c;
    }

    @Override
    public String toString() {
        return "num:" + num + " cost:" + cost;
    }
}

class Comp {
    Comparator<Group> comparator = new Comparator<Group>() {
        public int compare(Group g1, Group g2) {
            if (g1.cost < g2.cost) {
                return 1;
            } else if (g1.cost > g2.cost) {
                return -1;
            } else {
                return 0;
            }
        }
    };
}