CTO 让我们老人做做公司现在招聘的面试题，一个二维数组的题目， orz

/**

• 有若 7 个球框，每个球筐最多能放 3 个球，可以不放球。
• 7 个球筐上依次写着[红球，橙球，黄球，蓝球，铁球，木球，玻璃球]，
• 表示这个球筐里对球的限定。现在有 20 个球，
• 规格如下：
• 红铁球 15g，红铁球 26g，
• 橙铁球 25g，
• 黄铁球 18g，黄铁球 13g，
• 蓝铁球 33g,
• 红木球 31g，红木球 36g，
• 橙木球 11g，橙木球 25g，
• 黄木球 7g，黄木球 18g，
• 蓝木球 10g，
• 红玻璃球 3g，
• 橙玻璃球 21g，橙玻璃球 16g，
• 黄玻璃球 45g，黄玻璃球 29g，
• 蓝玻璃球 3g，蓝玻璃球 7g
• 请问组成最大重量的 N 个球筐限定的组合是什么？
• 请写出伪代码。 */

我写了一半的代码，不知如何继续下去，求大佬

public class Ball {
	private Color color;
	private Meterial meterial;
	private int weight;
}

public enum Color {
	YELLOW("YELLOW","黄色"),
	ORANGE("ORANGE","橙色"),
	RED("RED","红色"),
	BLUE("BLUE","蓝色");
}

public enum Meterial {
	IRON("IRON","铁"),
	WOOD("WOOD","木"),
	GLASS("GLASS","玻璃");
}
    
public void calculate(Ball[] balls) {
    int totalWeight = 0;
    List<Ball> red = new ArrayList<>();
    List<Ball> orange = new ArrayList<>();
    List<Ball> yellow = new ArrayList<>();
    List<Ball> blue = new ArrayList<>();
    List<Ball> iron = new ArrayList<>();
    List<Ball> wood = new ArrayList<>();
    List<Ball> glass = new ArrayList<>();

    HashMap<String,List<Ball>> basketsMap = new HashMap();
    basketsMap.put(Color.RED.getKey(),red);
    basketsMap.put(Color.ORANGE.getKey(),orange);
    basketsMap.put(Color.YELLOW.getKey(),yellow);
    basketsMap.put(Color.BLUE.getKey(),blue);
    basketsMap.put(Meterial.IRON.getKey(),iron);
    basketsMap.put(Meterial.WOOD.getKey(),wood);
    basketsMap.put(Meterial.GLASS.getKey(),glass);

    for (int i=0; i<balls.length; i++) {
        if (basketsMap.containsKey(balls[i].getColor().getKey())) {
            List<Ball> color = basketsMap.get(balls[i].getColor().getKey());
            if (color.size()<3) {
                color.add(balls[i]);
                totalWeight+=balls[i].getWeight();
            }
            else {
                System.out.println(basketsMap.get(balls[i].getColor().getKey()).toArray().toString());
                basketsMap.remove(balls[i].getColor().getKey());
            }
        }

        if (basketsMap.containsKey(balls[i].getMeterial().getKey())) {
            List<Ball> material = basketsMap.get(balls[i].getMeterial().getKey());
            if (material.size()<3) {
                material.add(balls[i]);
                totalWeight+=balls[i].getWeight();
            }
            else {
                System.out.println(basketsMap.get(balls[i].getMeterial().getKey()).toArray().toString());
                basketsMap.remove(balls[i].getColor().getKey());
            }
        }
    }