欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

键盘录入一个文件夹路径,统计该文件夹下的各种后缀名的文件的个数。例如:.txt有10个,.java有30个......

程序员文章站 2022-07-11 09:47:05
...

键盘录入一个文件夹路径,统计该文件夹下的各种后缀名的文件的个数。例如:.txt有10个,.java有30个…

代码并不多、思路很重要

import java.io.File;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Scanner;

public class HomeWork_day18_03_01 {
 public static void main(String[] args) {
  test_03();
 }
 private static void test_03() {
  /*
   * 键盘录入一个文件夹路径,统计该文件夹下的各种后缀名的文件的个数
	例如:.txt有10个,.java有30个......
   */
  File dir = getDir();//获取录入的文件夹路径
  List list = new ArrayList();//创建一个存文件后缀的list集合
  List overlist = getFileExtension(dir,list);//拿到返回的list(存有所有的文件后缀)
  System.out.println(overlist);
  bl(overlist);//遍历overlist,获取各个文件后缀的个数
 }

/**
  * 获取到文件下及子文件夹下的所有文件的后缀,并存入List中
  * @param dir   传入输入的文件夹路径
  * @param list  用于存所有文件的后缀名的list
  * @return      返回存所有文件的后缀名的list
  */
 private static List getFileExtension(File dir,List list) {
  // TODO Auto-generated method stub
  File[] f = dir.listFiles();
  for (File file : f) {
   if (file.isFile()) {//如果是文件就统计各种后缀名的文件的个数(使用LinkedHashMap)    
    String str = file.getName().substring(file.getName().lastIndexOf(".") + 1);
    list.add(str);  
   } else {//如果是文件夹就继续调用printFile()函数
    getFileExtension(file, list);
   }
  }
  return list;
 }

/**
  * 遍历得到的list(已存入文件夹及子文件夹中的文件的后缀名),获取各个文件后缀的个数
  * @param list  传入存所有文件的后缀名的list
  */
 private static void bl(List list) {
  LinkedHashMap<String, Integer> map = new LinkedHashMap<String, Integer>();   //准备用于统计与存储的容器,双列集合、用于统计对应关系
  for(int i = 0; i < list.size(); i ++) {
   if (map.containsKey(list.get(i))) {
    int count = map.get(list.get(i));  //遍历map集合,拼接字符串成要求格式  
    count += 1;
    map.put((String) list.get(i), count);
   } else {//如果集合中没有当前字符,说明以前没有统计过,说明是第一次
    map.put((String) list.get(i), 1);
   }
  }
  System.out.println(map);
  blMap(map);    //遍历map集合,拼接字符串成要求格式  
 }

/**
  * 遍历map集合,拼接字符串成要求格式
  * @param map   传入存有文件后缀及相应个数的map集合
  */
 private static void blMap(LinkedHashMap<String, Integer> map) {
  StringBuilder sb = new StringBuilder();
  for(Map.Entry<String, Integer> en : map.entrySet()) {
   String s = en.getKey();
   int count = en.getValue();
   sb.append(".").append(s).append("有").append(count).append("个, ");
  }
  System.out.println(sb.substring(0, sb.length() - 2));
 }
 
private static File getDir() { //从键盘录入一个文件夹路径,并判断是否是文件夹
  Scanner sc = new Scanner(System.in);
  while(true) {
   System.out.println("请录入一个文件夹路径:");
   String str = sc.nextLine();
   File f = new File(str);
   if (f.isDirectory()) {
    return f;
   } else {
    System.out.println("请重新录入...");
   }
  }
 }