在昨天参加了东哥的笔试,选择题做的还算可以,但是还有道编程题和关于jdk8的Stream特性难住了。鉴于此用博客总结一下这道编程题,并结合Stream特性来简化代码,熟悉Api。
题目描述
某校在积极推行无人监考制度,但是总有学生是不自觉的,如果将两个很熟的异性朋友放在同一个考场里,他们就会交流甚至作弊。因此一个考场中不能允许两个很熟的异性朋友存在,学校希望通过搬出一部分学生的方法来改善这一问题。但是又因为教室数量有限,因此希望一个教室中容下的学生尽可能多,即需要搬出教室的学生数量尽可能少,请你输出搬出教室人数最少,且字典序最小的方案。
输入
输入第一行有两个整数n和m,分别表示有n个男生和n个女生,有m个朋友关系。(1<=n<=500,1<=m<=100000)接下来m行,每行有两个整数,x和y,表示第x号男生和第y号女生是朋友。
男生的编号均为[1,n],女生的编号为[n+1,2n]。
输出
输出第一行包含一个整数a,表示最少需要搬出教室的人数。
输出第二行有a个整数,即a个需要搬出教室的人的编号,要求人数最少,且字典序最小。
意思是:一个教室内不能有亲密关系的男女,选择要移除的人满足尽可能的少、学号尽可能的小。
思路
按照题目描述可以明确:移除人数最少代表先移除不止和一个人有关系的同学。学号尽可能的小代表着优先移除男生。它的优先级是:人数>学号。
有了这个目标,再来确定算法的数据结构,我运用了以下的数据结构:
//维护每个学生的关系映射,一对多的关系
Map<Integer, List<Integer>> relationmap = new HashMap<>(16); //某个学生的关系具体学号
List templist = relationmap.getOrDefault(boynum, new ArrayList<>()); //存放男女生有关系的个数,男生与女生各n个,第一位不放数据。
int[] relation = new int[2*n+1];
然后利用伪代码来具体描述这个过程:
获得输入的男生人数n和关系m;
初始化relation数组,map;
for( m个关系){
接收男生与女生的关系; 将两个学号作key放入Map中; 把另一个学号作value放入map的List中; 以两个学号为下标的relation数组加一;
}
while(true){
从头到尾遍历relation数组,获得最大值的下标maxRelationIndex;
当maxRelationIndex==0,意味着当前没有亲密关系,break;
根据maxRelationIndex获得map中的关系结合list; for(list){
遍历list,将对应的relation数组的下标-1;
}
删除以学号为索引在relation数组的记录;
记录被删除的学号。
}
输出记录:
代码实现
import java.util.*; /**
* TODO
* @Author: HILL
* @date: 2019/8/25 9:50
*
**/
public class Main {
public static void main(String[] args) { //维护关系映射
Map<Integer, List<Integer>> relationmap = new HashMap<>(16);
//男女生人数
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int m = sc.nextInt(); //存放男女生有关系的个数
int[] relation = new int[2*n+1]; for (int i=0;i<m;i++){
int boynum = sc.nextInt();
int girlnum = sc.nextInt(); //添加男生关系映射
List templist = relationmap.getOrDefault(boynum, new ArrayList<>());
templist.add(girlnum);
relationmap.put(boynum,templist);
//添加女生关系映射
templist = relationmap.getOrDefault(girlnum, new ArrayList<>());
templist.add(boynum);
relationmap.put(girlnum,templist);
//维护每个人的关系度的权值,越大代表与越多人有关系
relation[girlnum]++;
relation[boynum]++; }
List<Integer> result = new ArrayList<>();
while (true){
int maxRelationIndex = 0; //从头到尾遍历,优先移除男生
for (int i=1 ;i<relation.length;i++ ){
if (relation[i]>maxRelationIndex){
maxRelationIndex = i;
}
}
//当教室里没有亲密关系时
if (maxRelationIndex == 0){
break;
} //优先移除与最多人有关系的学生
relation[maxRelationIndex] = 0;
//查出所有与被移除学生有关系的学生
List<Integer> list = relationmap.get(maxRelationIndex);
//将它们的关系计数-1
list.forEach(i-> relation[i]--); //将移除的学生加入到结果集
result.add(maxRelationIndex);
relationmap.remove(maxRelationIndex);
}
System.out.println(result.size());
result.forEach(num->System.out.print(num+" ")); }
}
提醒
以上代码仅供交流参考,用例不一定全部通过,因为当我想出来时已经没时间了。从总体上分析,性能估计不会很高,毕竟一个算法下来会有两次循环,同时空间复杂度也比较高。但能做一步是一步嘛,后面再看看能不能换个思路。