蓝桥杯Java组备赛--算法训练 kAc给糖果你吃

题目

问题描述

kAc有n堆糖果，每堆有A[i]个。
kAc说你只能拿m次糖果，聪明的你当然想要拿最多的糖果来吃啦啦啦~
//第二天，kAc问你还想吃糖果么？（嘿嘿嘿）说着眼角路出奇怪的微笑...

输入格式

第一行两个数字n和m，第二行有n个数字A[i]。

输出格式

输出一行表示最多能拿几个糖果。
样例输入
2 2
1 2
样例输出
3
数据规模和约定
0<n≤1000
其余数字都是不超过1,000,000,000的非负整数。

题目分析

一道简单的贪心问题。
就如题主之前讲过的《背包问题（贪心）& 二维01背包问题》。
按从大到小的顺序读取m次，就可以得到最多能拿的糖果。

静态数组

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        if(m == 0) {
            System.out.println("0");
            return;
        }
        long[] A = new long[n];
        for(int i = 0; i < n; i++) A[i] = scanner.nextLong();
        Arrays.sort(A);
        long sum = 0;
        for(int i = A.length-1; i >= 0; i--) {
            m--;
            sum += A[i];
            if(m == 0) break;
        }
        System.out.println(sum);
    }
}

动态数组ArrayList

动态数组的用法在我上一文Java中List接口两个实现，ArrayList类和LinkedList类的常用方法（一），有兴趣可以看一看。

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

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        long n = scanner.nextLong();
        long m = scanner.nextLong();
        if(m == 0){
        	System.out.println("0");
        	return;
        }//这一步其实不用，但为了代码规范
        ArrayList<Integer> A = new ArrayList<>();
        for(int i = 0; i < n; i++) {
            int num = scanner.nextInt();
            A.add(num);
        }
        A.sort(Comparator.reverseOrder());
        long sum = 0;
        for(int i = 0; i < m; i++){
            sum += A.get(i);
        }
        System.out.println(sum);
    }
}

注意：

• 两种方法思想并没有区别，不过静态数组的方法会繁琐一点。
• 静态数组用的Array.sort()是升序，而动态数组用的A.sort(Comparator.reverseOrder());是降序，因此直接从索引0开始往上加就行了。
• 还有就是需要注意数据类型溢出的问题，可以直接定义为long型。