蓝桥杯— Huffuman树

题目:在这里插入图片描述
在这里插入图片描述
法一:Arrays.copyOfRange(T[ ] original,int from,int to)
将一个原始的数组original,从下标from开始复制,复制到上标to,生成一个新的数组。
注意这里包括下标from,不包括上标to。
这个方法在一些处理数组的编程题里很好用,效率和clone基本一致,都是native method,比利用循环复制数组效率要高得多。

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

public class Main{
	public static void main(String[] args) {
		Scanner in=new Scanner(System.in);
		int n=in.nextInt();
		int a[]=new int [n];
		for(int i=0;i<n;i++) {
			a[i]=in.nextInt();
		}
		int i=0;
		int cost=0,tmp=0;
		while(i+1<a.length) {
			Arrays.sort(a);
			tmp=a[i]+a[i+1];
			a[i+1]=tmp;//替换掉第i+1个值
			cost=cost+tmp;
			//删除第i个元素
			a=Arrays.copyOfRange(a, i+1, a.length);
		}
		System.out.println(cost);
	}
}

法二:
用ArrayList来做

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

public class Main {
	public static void main(String[] args) {
		Scanner in=new Scanner(System.in);
		int n=in.nextInt();
		ArrayList a=new ArrayList();
		for(int i=0;i<n;i++)
			a.add(in.nextInt());
		while(a.size()>=2) {
			f(a);
		}
		System.out.println(cost);
	}
	
	static int cost=0;
	static ArrayList f(ArrayList a) {
		//Collections是一个工具类,sort是其中的静态方法,是用来对List类型进行排序的
		Collections.sort(a);
		Object m1=a.get(0);
		a.remove(m1);
		Object m2=a.get(0);
		a.remove(m2);
		cost=cost+(int)m1+(int)m2;
		a.add((int)m1+(int)m2);
		return a; 
	}
}

发布了35 篇原创文章 · 获赞 0 · 访问量 682

猜你喜欢

转载自blog.csdn.net/fendouzhilu666/article/details/104711147