- 设计RandomPool结构
- 并查集
- 岛问题
1.设计RandomPool结构
题目要求: 设计一种结构,在该结构中有如下三个功能 insert(key):将某个key加入到该结构,做到不重复加入。 delete(key):将原本在结构中的某个key移除。 getRandom(): 等概率随机返回结构中的任何一个key。
public static class RandomPool<T>{
private final HashMap<T,Integer> hashMap1;
private final HashMap<Integer,T> hashMap2;
private int size;
public RandomPool() {
this.hashMap1 = new HashMap<>();
this.hashMap2 = new HashMap<>();
this.size = 0;
}
public void insert(T key){
if (hashMap1.containsKey(key)){
return;
}else {
hashMap1.put(key,size);
hashMap2.put(size++,key);
}
}
public void delete(T key){
if (hashMap1.containsKey(key)){
T last = hashMap2.get(--size);
Integer keyIndex = hashMap1.get(key);
hashMap1.put(last,keyIndex);
hashMap2.put(keyIndex,last);
hashMap1.remove(key);
hashMap2.remove(size);
}
}
public T getRandom(){
if (size == 0){
return null;
}else {
return hashMap2.get((int)(Math.random() * size));
}
}
}
2.并查集 题目: 功能: (1)查两个元素是否属于一个集合; (2)将两个元素各自所在的集合合并为一个集合 (3)优化 在查询两个元素是否属于一个集合 将路过元素扁平化
public static class UnionFindSet{
private HashMap<Node,Node> fatherSet;
private HashMap<Node,Integer> sizeSet;
public UnionFindSet(List<Node> lists){
fatherSet = new HashMap<>();
sizeSet = new HashMap<>();
for (Node node:
lists) {
fatherSet.put(node,node);
sizeSet.put(node,1);
}
}
private Node findHead_noRecursion(Node node) {
Node cur = node;
Node curFather = fatherSet.get(cur);
Node father;
while (curFather != cur){
cur = curFather;
curFather = fatherSet.get(cur);
}
father = curFather;
cur = node;
curFather = fatherSet.get(cur);
while (curFather != cur){
fatherSet.put(cur,father);
cur = curFather;
curFather = fatherSet.get(cur);
}
return father;
}
private Node findHead_recursion(Node node) {
Node father = fatherSet.get(node);
if (father != node){
findHead_recursion(father);
}
fatherSet.put(node,father);
return father;
}
public boolean isSameSet(Node a, Node b) {
return findHead_recursion(a) == findHead_recursion(b);
}
public void union(Node a, Node b){
if (a == null || b == null){
return;
}
Node aHead = findHead_recursion(a);
Node bHead = findHead_recursion(b);
if (aHead != bHead){
int aSize = sizeSet.get(aHead);
int bSize = sizeSet.get(bHead);
if (aSize > bSize){
fatherSet.put(bHead,aHead);
sizeSet.put(aHead,aSize + bSize);
sizeSet.remove(bHead);
}else {
fatherSet.put(bHead,aHead);
sizeSet.put(bHead,aSize + bSize);
sizeSet.remove(aHead);
}
}
}
}
3.岛问题
题目: 一个矩阵中只有0和1两种值,每个位置都可以和自己的上、下、左、右 四个位置相连,如果有一片1连在一起,这个部分叫做一个岛,求一个 矩阵中有多少个岛?
private static int countIslands(int[][] arr) {
if (arr == null || arr[0] == null) {
return 0;
}
int num = 0;
for (int i = 0;i < arr.length;i++){
for (int y = 0;y < arr[0].length;y++){
if (arr[i][y] == 1){
infect(arr,i,y);
num++;
}
}
}
return num;
}
private static void infect(int[][] arr,int row, int column) {
if (row < 0 || row >= arr.length || column < 0 || column >= arr[0].length || arr[row][column] != 1){
return;
}
arr[row][column] = 2;
infect(arr,row - 1,column);
infect(arr,row + 1,column);
infect(arr,row,column - 1);
infect(arr,row,column + 1);
}
使用并查集计算大数据
(1)先将原矩阵 分成多块 (2)每块调用singleCPU求岛数量 -》其中每一个感染源 为并查集头结点 (3)看边界条件 -》如果边界两点不来自同一个岛屿 -》 合并 岛数量-1 -》如果边界两个点来自同一个岛屿 -》 不管
|