随机生成 Salary {name, baseSalary, bonus }的记录,如“wxxx,10,1”,每行一条记录,总共1000万记录,写入文本文件(UFT-8编码),
然后读取文件,name的前两个字符相同的,其年薪累加,比如wx,100万,3个人,最后做排序和分组,输出年薪总额最高的10组:
wx, 200万,10人
lt, 180万,8人
....
name 4位a-z随机, baseSalary [0,100]随机 bonus[0-5]随机 年薪总额 = baseSalary*13 + bonus
请努力将程序优化到5秒内执行完
/**
* © 武汉善德科技有限公司
*/
package com.tuodier.test.test;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Random;
/**
*
* @author lidehu
* @date 2018年11月28日上午10:06:54
*/
public class SalaryTest {
private static final char[] az = {'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h',
'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r',
's', 't', 'u', 'v', 'w', 'x', 'y', 'z'};
private static final String filePath = "F:"+File.separator+"salary.txt";
private static final String SPLIT = ",";
private static final Map<String, List<Integer>> map = new HashMap<String, List<Integer>>();
public static void main(String[] args) throws IOException {
String record = generateRecord();
writterFile(record);
readerFile();
}
private static void writterFile(String string) throws IOException {
long currentTimeMillis = System.currentTimeMillis();
BufferedWriter bufferedWriter = new BufferedWriter(new FileWriter(new File(filePath)));
bufferedWriter.write(string.trim());
bufferedWriter.flush();
bufferedWriter.close();
System.err.println("写入文件耗时:"+(System.currentTimeMillis()-currentTimeMillis)+"ms");
}
private static void readerFile() throws IOException {
long currentTimeMillis = System.currentTimeMillis();
String line = "";
try {
BufferedReader bReader = new BufferedReader(new FileReader(new File(filePath)));
while((line = bReader.readLine()) != null) {
String[] salary = line.trim().split(SPLIT);
String group = salary[0].substring(0, 2);
int annualSalary = Integer.parseInt(salary[1])*13+Integer.parseInt(salary[2]);
if(map.containsKey(group)) {
List<Integer> list = map.get(group);
list.set(0, list.get(0) + annualSalary);
list.set(1, list.get(1) + 1);
map.put(group, list);
} else {
map.put(group, Arrays.asList(annualSalary, 1));
}
}
bReader.close();
} catch (FileNotFoundException e) {
e.printStackTrace();
}
System.err.println("解析文件耗时:"+(System.currentTimeMillis()-currentTimeMillis)+"ms");
}
private static String generateRecord() {
long currentTimeMillis = System.currentTimeMillis();
Random random = new Random();
StringBuilder sb = new StringBuilder();
for(int i=0; i<10000000; i++) {
for(int j=0; j<4; j++) {
sb.append(az[random.nextInt(26)]);
}
sb.append(SPLIT);
sb.append(random.nextInt(101));
sb.append(SPLIT);
sb.append(random.nextInt(6));
sb.append("\n");
}
System.err.println("生成数据耗时:"+(System.currentTimeMillis()-currentTimeMillis)+"ms");
return sb.toString();
}
}
生成的String数据太大,trim()花了0.1s+
还没排序时间就超过5s了。
这种写法的排序我还得研究。