算法训练 纪念品分组 Java

问题描述

元旦快到了，校学生会让乐乐负责新年晚会的纪念品发放工作。为使得参加晚会的同学所获得的纪念品价值 相对均衡，他要把购来的纪念品根据价格进行分组，但每组最多只能包括两件纪念品，并且每组纪念品的价格之和不能超过一个给定的整数。为了保证在尽量短的时 间内发完所有纪念品，乐乐希望分组的数目最少。
　　你的任务是写一个程序，找出所有分组方案中分组数最少的一种，输出最少的分组数目。

输入格式

输入包含n+2行：
　　第1行包括一个整数w，为每组纪念品价格之和的上限。
　　第2行为一个整数n，表示购来的纪念品的总件数。
　　第3~n+2行每行包含一个正整数pi (5 <= pi <= w)，表示所对应纪念品的价格。

输出格式

输出仅一行，包含一个整数，即最少的分组数目。

样例输入

100
9
90
20
20
30
50
60
70
80
90

样例输出

6

数据规模和约定

50%的数据满足：1 <= n <= 15
　　100%的数据满足：1 <= n <= 30000, 80 <= w <= 200

思路

• 运用贪心算法，每次选择价格最大的和最小的分一组，看是否符合要求
• 将输入的价格存入数组并按从小到大排序
• 指针head和指针tail分别指向数组的头和尾（即指向最低价格和最高 价格）
• 如果两指针对应的价格之和超过了最大值，则说明tail指针对应的元素需独立分一组，指针前移
• 如果两指针对应的价格之和小于最大值，则可以分一组，head指针后移，tail指针前移
• 两指针位置交叉则结束

代码

import java.util.Arrays;
import java.util.Scanner;
public class Souviner {
	public static void main(String[] args) {
		Scanner input = new Scanner(System.in);
		int max = input.nextInt();
		int n = input.nextInt();
		int[] price = new int[n];
		for (int i = 0; i < price.length; i++) {
			price[i] = input.nextInt();
		}
		input.close();
		Arrays.sort(price);
		System.out.println(solution(max, price));
	}

	private static int solution(int max, int[] price) {
		int count = 0;
		int head = 0; //指向价格最低
		int tail = price.length - 1; //指向价格最高
		while(head < tail) {
			if(price[head] + price[tail] > max) {
				tail--;
				count++;
			}
			else {
				tail--;
				head++;
				count++;
			}
		}
		if(head == tail) {
			count++;
		}
		return count;
		
	}

}