Java程序:统计颜色在相邻行号距离小于m的次数
这是一个Java程序,主要实现了以下功能:
-
从命令行读入三个整数n,m,c。
-
从命令行读入n行整数,每行第一个整数表示该行颜色数量,后面是这些颜色的编号。
-
统计颜色编号在所有行中出现过的位置,将其存入一个Map<Integer, List
>中,其中Map的key为颜色编号,value为该颜色出现的行号列表。 -
遍历Map中的每个颜色,对于该颜色出现的行号列表,判断其中相邻行号之间的距离是否小于m,如果是则计数器加1。
-
输出计数器的值。
该程序的功能是统计所有颜色中,相邻行号之间距离小于m的颜色数量。
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Scanner;
import java.util.Set;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int m = scanner.nextInt();
int c = scanner.nextInt();
int [][]colors = new int[n][51];
for (int i = 0; i < n; i++) {
colors[i][0] = scanner.nextInt();
for (int j = 1; j <= colors[i][0]; j++) {
colors[i][j]= scanner.nextInt();
}
}
scanner.close();
Map<Integer, List<Integer>> posMap = new HashMap<>();
for (int i = 0; i < n; i++) {
for (int j = 1; j <= colors[i][0]; j++) {
int colorType = colors[i][j];
if (colorType != 0) {
if (!posMap.containsKey(colorType)) {
List<Integer> list = new ArrayList<>();
list.add(i);
posMap.put(colorType, list);
} else {
List<Integer> list = posMap.get(colorType);
list.add(i);
}
}
}
}
Set<Integer> set = posMap.keySet();
Iterator<Integer> iterator = set.iterator();
int count = 0;
while (iterator.hasNext()) {
Integer integer = iterator.next();
List<Integer> list = posMap.get(integer);
for (int i = 0; i < list.size(); i++) {
if (i < list.size() - 1) {
if (list.get(i + 1) - list.get(i) < m) {
count++;
break;
}
} else {
if (n - list.get(i) + list.get(0) < m) {
count++;
break;
}
}
}
}
System.out.println(count);
}
}
原文地址: https://www.cveoy.top/t/topic/lTWA 著作权归作者所有。请勿转载和采集!