SW练习_P1090_合并果子

原创于 2020-06-20 17:27:00 发布 · 93 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

96 篇文章

订阅专栏

本文介绍了一种利用PriorityQueue数据结构解决特定问题的高效算法实现。通过实例演示了如何读取输入，使用优先队列进行数据处理，并计算最终结果，展示了PriorityQueue在算法优化中的重要作用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用PriorityQueue解决这种问题，简直不要太爽



import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws Exception{
        BufferedReader reader=new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st=new StringTokenizer(reader.readLine());

        int n=Integer.parseInt(st.nextToken());
        PriorityQueue<Integer> pq=new PriorityQueue();
        st=new StringTokenizer(reader.readLine());
        for (int i = 0; i <n ; i++) {
            pq.add(Integer.parseInt(st.nextToken()));
        }
        int cout=0;
        while (pq.size()>1){
            int a=pq.poll();
            int b=pq.poll();
            int c=a+b;
            pq.add(c);
            cout+=c;
        }

        System.out.println(cout);

        reader.close();
    }
}