蚂蚁金服面试编码题 01:
读取一个txt文件,文件内容是n个网址,一行一条,请读取文件内容,计算出每个网址的数量,并以降序排列,输出网址和数量至新文件中。
要点: 熟悉nio api,包括FileChannel,ByteBuffer,RandomAccessFile等,stream 分组以及排序。
package priv.yi.yzim;
import java.io.RandomAccessFile;
import java.net.URL;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
public class LoadAndCountFile {
public static void main(String[] args) {
try {
URL url = LoadAndCountFile.class.getClassLoader().getResource("file.txt");
if (url != null) {
RandomAccessFile file = new RandomAccessFile(url.getFile(), "rw");
FileChannel channel = file.getChannel();
ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
int readBytes = channel.read(byteBuffer);
StringBuilder stringBuilder = new StringBuilder();
System.out.println("begin read");
while (readBytes != -1) {
byteBuffer.flip();
while (byteBuffer.hasRemaining()) {
stringBuilder.append((char) byteBuffer.get());
}
byteBuffer.clear();
readBytes = channel.read(byteBuffer);
}
file.close();
String[] strArray = stringBuilder.toString().split(System.getProperty("line.separator"));
Map<String, Integer> stringIntegerMap = Arrays.stream(strArray).collect(Collectors.groupingBy((x -> x), Collectors.summingInt(t -> 1)));
List<Map.Entry<String, Integer>> entryList = stringIntegerMap.entrySet().stream()
.sorted((a, b) -> b.getValue().compareTo(a.getValue())).collect(Collectors.toList());
byteBuffer.clear();
RandomAccessFile outPutFile = new RandomAccessFile("D:\\outPutFile.txt", "rw");
FileChannel writeChannel = outPutFile.getChannel();
entryList.forEach(entry -> {
byteBuffer.put(entry.getKey().getBytes());
byteBuffer.put(System.getProperty("line.separator").getBytes());
byteBuffer.put(String.valueOf(entry.getValue()).getBytes());
byteBuffer.put(System.getProperty("line.separator").getBytes());
});
byteBuffer.flip();
while (byteBuffer.hasRemaining()) {
writeChannel.write(byteBuffer);
}
writeChannel.close();
System.out.println("write over");
}
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}
结果:
www.google.com
3
abc.com
2
www.baidu.com
2
ls.com
1
ps.com
1
c.com
1