539.最小时间差 & 382.链表随机节点

127 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

题目

539.最小时间差

题目大意

给定一个 24 小时制(小时:分钟 "HH:MM")的时间列表,找出列表中任意两个时间的最小时间差并以分钟数表示。

样例

img

数据规模

img

思路

为了处理数据方便,这里用python语言。一天最多就1440分钟,不妨考虑用数组标记时间,一旦出现一个时间(x,y)(x,y),就转换成x60+yx*60+y 分钟,这样标记vis[x+60+y]++vis[x+60+y]++,如果某一个vis[i]>=2(i[0,1440)vis[i]>=2(i∈[0,1440)那么直接返回0,因为有两个相同时间,可以让差值最小为0,其余情况可以考虑两个vis[x1]=vis[x2]=1vis[x1]=vis[x2]=1并且x1x2|x1-x2|尽可能小,线性枚举即可,为了解决环形,可以直接环形展开,即相当于[0,1440)[0,1440)在复制一份,然后再线性求解。

代码

class Solution:
    def findMinDifference(self, timePoints: List[str]) -> int:
        last=1e9
        a=[0 for i in range(2880+2)]
        for it in timePoints:
            x,y=map(int,it.split(':'))
            a[x*60+y]+=1
            a[x*60+y+1440]+=1
            if(a[x*60+y]>=2):
                return 0
            last=min(last,x*60+y)
        ans=1e9
        for i in range(2880+1):
            if(a[i] and i!=last):
                ans=min(ans,i-last)
                last=i
        return ans

题目

382.链表随机节点

题目大意

给你一个单链表,随机选择链表的一个节点,并返回相应的节点值。每个节点 被选中的概率一样

实现 Solution 类:

  • Solution(ListNode head) 使用整数数组初始化对象。
  • int getRandom() 从链表中随机选择一个节点并返回该节点的值。链表中所有节点被选中的概率相等。

样例

img

数据规模

img

思路1

可以在初始化时,用一个数组记录链表中的所有元素,这样随机选择链表的一个节点,就变成在数组中随机选择一个元素,相当于每次rand一个数∈[0,tot-1],然后返回a[rand值]。

代码1

class Solution {
public:
    int a[100000+50],tot;
    Solution(ListNode* head) {
        while(head){
            a[tot++]=head->val;
            head=head->next;
        }
    }
    
    int getRandom() {
        int x=rand()%tot;
        return a[x];
    }
};

思路2

img

方法一需要花费 O(n)O(n) 的空间存储链表中的所有元素,那么能否做到 O(1)O(1) 的空间复杂度呢?

可以设计算法:从链表头开始,遍历整个链表,对遍历到的第ii个节点,随机选择区间[0,i)[0,i)内的一个整数,如果其等于00,则将答案置为该节点值,否则答案不变。

该算法会保证每个节点的值成为最后被返回的值的概率均为1n\frac{1}{n}

代码2

class Solution {
public:
    ListNode* a;
    Solution(ListNode* head) {
        a=head;
    }
    
    int getRandom() {
        int i=1,ans=0;
        for(auto Node=a;Node;Node=Node->next){
            if(rand()%i==0){
                ans=Node->val;
            }
            i++;
        }
        return ans;
    }
};