題目
題目鏈接:
https://www.nowcoder.com/practice/4d3151698e33454f98bce1284e553651
https://leetcode.cn/problems/maximum-number-of-events-that-can-be-attended/description/
思路
貪心+優先級隊列
Java代碼
import java.util.*;public class Solution {/*** 代碼中的類名、方法名、參數名已經指定,請勿修改,直接返回方法規定的值即可*** @param meetings int整型ArrayList<ArrayList<>>* @return int整型*/public int attendmeetings (ArrayList<ArrayList<Integer>> meetings) {//貪心+優先級隊列//力扣1353. 最多可以參加的會議數目 力扣上題目描述的更好int ans = 0, max = -1; //max為最后一個會議的結束時間//按會議結束時間排序PriorityQueue<Integer> q = new PriorityQueue<>(new Comparator<Integer>() {@Overridepublic int compare(Integer a, Integer b) {return a - b;}});// key -> 第 i 天, value -> 第 i 天開始的會議的結束時間Map<Integer, List<Integer>> map = new HashMap<>();for (ArrayList<Integer> e : meetings) {//將開始時間相同的所有會議的結束時間放一起if (!map.containsKey(e.get(0))) {map.put(e.get(0), new ArrayList<>());}map.get(e.get(0)).add(e.get(1));//max為最后一個會議的會議結束時間max = Math.max(e.get(1), max);}// 整體思路就是從1開始到最晚結束時間依次遍歷一遍,然后挑選此時間要進行的會議;// 而隊列的作用就是存儲未進行的會議;挑選的判斷條件就是挑選的時間不能小于此時的時間,// 因為隊列中存儲的是每個會議最晚進行時間,即,結束時間for (int i = 1; i <= max ; i++) {if (map.containsKey(i)) {for (Integer cur : map.get(i)) {q.add(cur);}}while (!q.isEmpty() && q.peek() < i) {q.poll();}if (!q.isEmpty()) {ans++;q.poll();}}return ans;}
}
Go代碼
package main/*** 代碼中的類名、方法名、參數名已經指定,請勿修改,直接返回方法規定的值即可*** @param meetings int整型二維數組* @return int整型*/
func attendmeetings(meetings [][]int) int {//貪心+優先級隊列//自己實現的升序堆,按meetings里的結束時間排序h := Heap{make([]int, 10), 0}lastday := -1 //最后會議結束時間//key: 會議開始是第key天,開始時間的相同的會議的結束時間放同個一list中daymap := map[int][]int{}for _, e := range meetings {start := e[0]end := e[1]_, ok := daymap[start]if !ok {daymap[start] = []int{}}daymap[start] = append(daymap[start], end)if lastday < end {lastday = end}}ans := 0// 整體思路就是從1開始到最晚結束時間依次遍歷一遍,然后挑選此時間要進行的會議;// 而隊列的作用就是存儲未進行的會議;挑選的判斷條件就是挑選的時間不能小于此時的時間,// 因為隊列中存儲的是每個會議最晚進行時間,即,結束時間for i := 1; i <= lastday; i++ {_, ok := daymap[i]if ok {list := daymap[i]for _, v := range list {h.add(v)}}for h.Size > 0 && h.peek() < i {h.remove()}if h.Size > 0 {ans++h.remove()}}return ans
}type Heap struct {Arr []intSize int
}func (h *Heap) ensure(length int) { //堆的擴容old := len(h.Arr)if old >= length {return}newSize := old + old>>1newArr := make([]int, newSize)for i := 0; i < old; i++ {newArr[i] = h.Arr[i]}h.Arr = newArr
}func (h *Heap) add(x int) { //往堆中添加元素idx := h.Sizeh.ensure(idx + 1)h.Arr[idx] = xh.shiftup(idx)h.Size++
}func (h *Heap) shiftup(idx int) { //堆的上濾base := h.Arr[idx]for idx > 0 {pid := (idx - 1) >> 1parent := h.Arr[pid]if base >= parent {break}h.Arr[idx] = parentidx = pid}h.Arr[idx] = base
}func (h *Heap) peek() int {return h.Arr[0]
}func (h *Heap) remove() int {ans := h.Arr[0]idx := h.Sizeh.Arr[0] = h.Arr[idx-1]h.shiftdown(0)h.Size--return ans
}func (h *Heap) shiftdown(idx int) { //下竄,其實可以不傳idx,默認是0,從0下標下竄base := h.Arr[idx]half := h.Size >> 1for idx < half {cidx := idx<<1 + 1right := cidx + 1child := h.Arr[cidx]if right < h.Size && h.Arr[right] < child {cidx = rightchild = h.Arr[right]}if base < child {break}h.Arr[idx] = childidx = cidx}h.Arr[idx] = base
}
PHP代碼
<?php/*** 代碼中的類名、方法名、參數名已經指定,請勿修改,直接返回方法規定的值即可** * @param meetings int整型二維數組 * @return int整型*/
function attendmeetings( $meetings )
{//貪心+優先級隊列$ans = 0;$max = -1; //會議日期最大的那個日期,從每場會議的結束日期里獲取$h = new Heap();//key: 會議的開始日期start,value:存結束時間的集合list,相同的start的集合$map = [];foreach ($meetings as $e){$start = $e[0];$end = $e[1];if(!isset($map[$start])){$map[$start] =[];}$map[$start][count($map[$start])] = $end;if($max < $end){$max = $end;}}// 整體思路就是從1開始到最晚結束時間依次遍歷一遍,然后挑選此時間要進行的會議;// 而隊列的作用就是存儲未進行的會議;挑選的判斷條件就是挑選的時間不能小于此時的時間,// 因為隊列中存儲的是每個會議最晚進行時間,即,結束時間for($i=1;$i<=$max;$i++){if(isset($map[$i])){$arr = $map[$i];foreach ($arr as $d){$h->add($d);}}while ($h->size >0 && $h->peek() <$i){$h->remove();}if ($h->size >0){$ans++;$h->remove();}}return $ans;
}class Heap{ //自己的實現的升序堆public $arr;public $size;public function __construct(){//初始化堆for($i=0;$i<10;$i++){$this->arr[$i] =0;}$this->size =0;}public function ensure($cap){ //擴容代碼$old = count($this->arr);if($old >=$cap) return;$newsize = $old+$old >>1;$newarr = [];for($i=0;$i<$newsize;$i++){$newarr[$i] =0;if($i<$old){$newarr[$i] = $this->arr[$i];}}$this->arr =$newarr;}public function add($x){$idx = $this->size;$this->ensure($idx+1);$this->arr[$idx] =$x;$this->shiftup($idx);$this->size++;}public function shiftup($idx){ //上濾$base = $this->arr[$idx];while ($idx >0){$pid = ($idx-1) >> 1; //父id$parent = $this->arr[$pid];if($base >$parent) break;$this->arr[$idx] = $parent;$idx = $pid;}$this->arr[$idx] = $base;}public function peek(){return $this->arr[0];}public function remove(){$ans =$this->arr[0];$curlen = $this->size;$this->arr[0] = $this->arr[$curlen-1];$this->size--;$this->shiftdown(0);return $ans;}//下竄,可以不用傳idx參數,一般是從0開始下竄,idx=0public function shiftdown($idx){$base = $this->arr[$idx];$half = $this->size >> 1;while ($idx<$half) {$cidx = ($idx<<1) +1;$right = $cidx+1;$child = $this->arr[$cidx];if($right < $this->size && $child > $this->arr[$right]){$cidx = $right;$child = $this->arr[$right];}if($child > $base) break;$this->arr[$idx] =$child;$idx= $cidx;}$this->arr[$idx] = $base;}
}