通过万岁!!!
- 题目:给你一个数n,表示有n个人,你需要将这n个人分成2组。然后还给你了一个数组,每一行有两个数,表示第一个人和第二个人不能在在一组。问能不能分配成功。
- 思路:我的思路是遍历所有的条件(也就是不能在一起的情况),然后我们假设条件中第一行的第一个人在第一组,然后那么第二个人在第二组,然后找包含第二个人的条件,将与之对立的放在第一组。依次类推,如果遇到冲突的情况,就返回false即可了。否则都安排好了以后,就返回true。这里用到了递归,每一次都是找到与这个人相关的条件。
- 技巧:深度优先
java代码
class Solution {
public boolean possibleBipartition(int n, int[][] dislikes) {
int[] group = new int[n + 1];
int len = dislikes.length;
for (int i = 1; i < len; i++) {// 遍历所有的条件
if (group[dislikes[i][0]] == 0) {
group[dislikes[i][0]] = 1;
if (!get(dislikes, group, dislikes[i][0], group[dislikes[i][0]])) {
return false;
}
}
}
return true;
}
/**
* 每次判断当前这个的对应的人的他条件
*
* @param dis 条件数组
* @param group 分组情况
* @param person 当前这个人
* @param perGroup 当前这个人的分组
*/
public boolean get(int[][] dis, int[] group, int person, int perGroup) {
for (int i = 0; i < dis.length; i++) {
if (dis[i][0] == person) {
if (group[dis[i][1]] == 0) {// 这个人没有分过组
group[dis[i][1]] = perGroup == 1 ? 2 : 1;
if (!get(dis, group, dis[i][1], group[dis[i][1]])) {
return false;
}
continue;
}
if (group[dis[i][1]] != (perGroup == 1 ? 2 : 1)) {// 这个人分过组了
return false;
}
}
if (dis[i][1] == person) {
if (group[dis[i][0]] == 0) {
group[dis[i][0]] = perGroup == 1 ? 2 : 1;
if (!get(dis, group, dis[i][0], group[dis[i][0]])) {
return false;
}
continue;
}
if (group[dis[i][0]] != (perGroup == 1 ? 2 : 1)) {
return false;
}
}
}
return true;
}
}
- 总结:我觉得题目有点难,可能是我好久没有做深度学习的题目了。而且感觉写的代码也比较乱,时间复杂度也很高。但是不知道怎么进行优化了。只能弱弱的说一句通过万岁。