Leetcode - Merge Intervals?

Paste_Image.png

My code:

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

/**
 * Definition for an interval.
 * public class Interval {
 *     int start;
 *     int end;
 *     Interval() { start = 0; end = 0; }
 *     Interval(int s, int e) { start = s; end = e; }
 * }
 */
public class Solution {
    public List<Interval> merge(List<Interval> intervals) {
        ArrayList<Interval> result = new ArrayList<Interval>();
        if (intervals == null || intervals.size() == 0)
            return result;
        if (intervals.size() == 1) {
            result.add(intervals.get(0));
            return result;
        }
        
        int[] begin = new int[intervals.size()];
        HashMap<Integer, Integer> range = new HashMap<Integer, Integer>();
        range.put(intervals.get(0).start, intervals.get(0).end);
        begin[0] = intervals.get(0).start;
        for (int i = 1; i < intervals.size(); i++) {
            begin[i] = intervals.get(i).start;
            if (range.containsKey(intervals.get(i).start) && intervals.get(i).end <= range.get(begin[i]))
                continue;
            else
                range.put(intervals.get(i).start, intervals.get(i).end);
        }
        Arrays.sort(begin);
        Interval interval = new Interval(begin[0], range.get(begin[0]));
        
        for (int i = 1; i <= begin.length; i++) {
            if (i == begin.length)
                result.add(interval);
            else if (begin[i] > interval.end) {
                result.add(interval);
                interval = new Interval(begin[i], range.get(begin[i]));
            }
            else if (begin[i] < interval.start) {
                if (range.get(begin[i]) <= range.get(begin[i - 1]))
                    interval = new Interval(begin[i], range.get(begin[i - 1]));
                else
                    interval = new Interval(begin[i], range.get(begin[i]));
            }
            else if (range.get(begin[i]) > interval.end)
                interval = new Interval(interval.start, range.get(begin[i]));
            else
                continue;
        }
        return result;
    }
    
    public static void main(String[] args) {
        Solution test = new Solution();
        Interval a0 = new Interval(1, 4);
        Interval a1 = new Interval(1, 4);
        ArrayList<Interval> result = new ArrayList<Interval>();
        result.add(a0);
        result.add(a1);
        System.out.println(test.merge(result));
    }
}

My test result:

Paste_Image.png

这道题目不算很难。一直在思考一个问题,如何保存这种范围的映射,这种start和end的映射。最后还是决定用HashMap, key -> start, value -> end.
将所有的数据保存在哈希表中,
同时,将所有的开头数字,保存在begin数组中。注意保存哈希表的时候,如果当前头在哈希表中已经存在的话,需要判断下其尾是否大于哈希表对应的val。如果不是,则跳过,如果是,则更新哈希表。
然后对 begin[] 排序。
然后按照一定的规则,处理这个数组,不断地插入新的interval到结果中去。
差不多就这样了。

**
总结: Array, sort, hash table
今天总算把三道题目的量完成了。中午和初中同学吃饭。突然明白了那么一句话。
我努力了这么久,就是为了和你喝一杯咖啡。
感觉他们的世界,和我的不同。他们对知识的掌握程度,也远高于我。但同样的,他们也需要考虑汽车在哪里买便宜。租的房子干不干净。这些他们也不能幸免。
然后,我要潜下心来,默默地,变强。
最后,希望选课顺利!
**

Anyway, Good luck, Richardo!

My code:

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

/**
 * Definition for an interval.
 * public class Interval {
 *     int start;
 *     int end;
 *     Interval() { start = 0; end = 0; }
 *     Interval(int s, int e) { start = s; end = e; }
 * }
 */
public class Solution {
    public List<Interval> merge(List<Interval> intervals) {
        if (intervals == null || intervals.size() == 0)
            return intervals;
        Collections.sort(intervals, new Comparator<Interval>() {
            public int compare(Interval o1, Interval o2) {
//                int start1 = o1.start;
//                int start2 = o2.start;
//                int end1 = o1.end;
//                int end2 = o2.end;
//                if (start1 == start2 && end1 == end2)
//                  return 0;
//                else if (start1 <= start2 && end1 <= end2)
//                    return -1;
//                else
//                    return 1;
                return o1.start - o2.start;
            }        
        });
        
        for (int i = 1; i < intervals.size(); i++) {
            int frontStart = intervals.get(i - 1).start;
            int frontEnd = intervals.get(i - 1).end;
            int backStart = intervals.get(i).start;
            int backEnd = intervals.get(i).end;
            if (backStart > frontEnd)
                continue;
            else {
                intervals.get(i - 1).start = Math.min(frontStart, backStart);
                intervals.get(i - 1).end = Math.max(frontEnd, backEnd);
                intervals.remove(i);
                i--;
            }
        }
        return intervals;
    }
    
    public static void main(String[] args) {
        Interval a1 = new Interval(1, 3);
        Interval a2 = new Interval(2, 5);
        Interval a3 = new Interval(6, 7);
        Interval a4 = new Interval(3, 4);
        ArrayList<Interval> intervals = new ArrayList<Interval>();
        intervals.add(a1);
        intervals.add(a2);
        intervals.add(a3);
        intervals.add(a4);
        Solution test = new Solution();
        System.out.println(test.merge(intervals));
    }
}

这是我自己写的,直接在里面改元素。
但是速度较慢。
为什么呢?
因为我有一个删除操作。链表的删除操作复杂度是O(n),所以如果测试案例的重合元素很多,我的速度会很慢很慢。
于是有了一个改进方法,就是新建一个ArrayList<...> ret,
然后把需要塞进去的,已经更新好了的interval放进去,这样就不会再有删除操作了。同样也是遍历整个链表。链表的.add()并不是很消耗时间的操作。我就懒得写了,但是速度应该会更快。
然后就出现了一个问题,以前从来没有碰到过:

Line 19: java.lang.IllegalArgumentException: Comparison method violates its general contract!

我查了下,大概是,
如果 A = B, 那么 B = A, 而我的comparator方法写得可能违背了这一点。

Collections.sort(intervals, new Comparator() {
            public int compare(Object o1, Object o2) {
                int start1 = ((Interval) o1).start;
                int start2 = ((Interval) o2).start;
                if (start1 <= start2)
                    return -1;
                else
                    return 1;
            }        
        });

这是那段错误的代码。我改了下,

Collections.sort(intervals, new Comparator<Interval>() {
            public int compare(Interval o1, Interval o2) {
//                int start1 = o1.start;
//                int start2 = o2.start;
//                int end1 = o1.end;
//                int end2 = o2.end;
//                if (start1 == start2 && end1 == end2)
//                  return 0;
//                else if (start1 <= start2 && end1 <= end2)
//                    return -1;
//                else
//                    return 1;
                return o1.start - o2.start;
            }        
        });

如果直接用注释的部分,是对的。用当前部分也是对的。
我还是不能很好地理解这个错误。等回了学校再仔细研究下。

然后我看了以前的做法,竟然可以用HashMap来做,也果真是巧妙!
我又自己重写了下,错了好几次才AC。思路和代码比之前要简洁很多。
但其实和我上面的版本差不多的。就是一个排序。一个用Comparator实现,一个用HashMap实现。

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

/**
 * Definition for an interval.
 * public class Interval {
 *     int start;
 *     int end;
 *     Interval() { start = 0; end = 0; }
 *     Interval(int s, int e) { start = s; end = e; }
 * }
 */
public class Solution {
    public List<Interval> merge(List<Interval> intervals) {
        ArrayList<Interval> result = new ArrayList<Interval>();
        if (intervals == null || intervals.size() == 0)
            return result;
            
        int[] begin = new int[intervals.size()];
        HashMap<Integer, Integer> range = new HashMap<Integer, Integer>();
        for (int i = 0; i < intervals.size(); i++) {
            int start = intervals.get(i).start;
            int end = intervals.get(i).end;
            begin[i] = start;
            if (range.containsKey(start) && range.get(start) >= end)
                continue;
            else
                range.put(start, end);
        }
        Arrays.sort(begin);
        result.add(new Interval(begin[0], range.get(begin[0])));
        for (int i = 1; i < begin.length; i++) {
            if (begin[i] == begin[i - 1])
                continue;
            else {
                if (begin[i] > result.get(result.size() - 1).end)
                    result.add(new Interval(begin[i], range.get(begin[i])));
                else
                    result.get(result.size() - 1).end = Math.max(result.get(result.size() - 1).end, range.get(begin[i]));
            }
        }
        return result;
    }

    public static void main(String[] args) {
        Solution test = new Solution();
        Interval a0 = new Interval(2, 3);
        Interval a1 = new Interval(4, 5);
        Interval a2 = new Interval(6, 7);
        Interval a3 = new Interval(8, 9);
        Interval a4 = new Interval(1, 10);
        ArrayList<Interval> result = new ArrayList<Interval>();
        result.add(a0);
        result.add(a1);
        result.add(a2);
        result.add(a3);
        result.add(a4);
        System.out.println(test.merge(result));
    }
}

**
记住了,链表的remove操作很费时间,决定性浪费时间!
能不用就不用把!
还有一个comparator 问题没有搞清楚!
**

Anyway, Good luck, Richardo!

More efficient solution:
My code:

/**
 * Definition for an interval.
 * public class Interval {
 *     int start;
 *     int end;
 *     Interval() { start = 0; end = 0; }
 *     Interval(int s, int e) { start = s; end = e; }
 * }
 */
public class Solution {
    public List<Interval> merge(List<Interval> intervals) {
        ArrayList<Interval> ret = new ArrayList<Interval>();
        if (intervals == null || intervals.size() == 0)
            return ret;
        Collections.sort(intervals, new Comparator<Interval>() {
                public int compare(Interval o1, Interval o2) {
                    return o1.start - o2.start;
                }            
        });    
        Interval preInterval = intervals.get(0);
        for (int i = 1; i < intervals.size(); i++) {
            Interval curr = intervals.get(i);
            if (curr.start > preInterval.end) {
                ret.add(preInterval);
                preInterval = curr;
            }
            else {
                Interval merge = new Interval(preInterval.start, Math.max(preInterval.end, curr.end));
                preInterval = merge;
            }
        }
        ret.add(preInterval);
        return ret;
    }
}

Anyway, Good luck, Richardo!

My code:

/**
 * Definition for an interval.
 * public class Interval {
 *     int start;
 *     int end;
 *     Interval() { start = 0; end = 0; }
 *     Interval(int s, int e) { start = s; end = e; }
 * }
 */
public class Solution {
    public List<Interval> merge(List<Interval> intervals) {
        List<Interval> ret = new ArrayList<Interval>();
        if (intervals == null || intervals.size() == 0) {
            return ret;
        }
        
        ret.add(intervals.get(0));
        
        for (int i = 1; i < intervals.size(); i++) {
            Interval curr = intervals.get(i);
            int j = ret.size() - 1;
            for (; j >= 0; j--) {
                Interval cmp = ret.get(j);
                if (cmp.end < curr.start) {
                    ret.add(j + 1, curr);
                    break;
                }
                else if (cmp.start <= curr.start) {
                    cmp.end = Math.max(cmp.end, curr.end);
                    break;
                }
                else if (cmp.start <= curr.end) {
                    cmp.start = curr.start;
                    cmp.end = Math.max(cmp.end, curr.end);
                    curr = cmp;
                    ret.remove(j);
                }
                else {
                    continue;
                }
            }
            if (j < 0) {
                ret.add(0, curr);
            }
        }
        
        return ret;
    }
}

自己写了出来。应该是 Insert sort 的一种变形。
注意地方是,如果在向左遍历途中,更新了一个Interval导致他的start变小了,那么很有可能这个interval会覆盖他前面一个interval。这个时候就要把当前Interval从链表里面删除掉。然后更新curr指针。让他和下一个interval再次进行比较。

list remove 操作。一般情况下,时间复杂度是O(n)
如果删的是头部或者尾部,时间复杂度是O(1)

以前的那个用HashMap实现sort的方法也是很有意思的。

Anyway, Good luck, Richardo! -- 09/02/2016

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 195,980评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,422评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,130评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,553评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,408评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,326评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,720评论 3 386
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,373评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,678评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,722评论 2 312
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,486评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,335评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,738评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,009评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,283评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,692评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,893评论 2 335

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,712评论 0 33
  • My code: My test result: 首先说下测试结果为什么这么慢。我的理解是,这道题目是8.9才改题...
    Richardo92阅读 1,234评论 1 1
  • My code: My test result: 这个做法不是我一开始的做法。但是这个做法速度更快更高效。我也是看...
    Richardo92阅读 244评论 0 1
  • My code: 这道题目还是挺有意思的。一开始写出来超出时间了,没有怎么仔细分析。两个操作,get,set,最理...
    Richardo92阅读 400评论 0 0
  • My code: 最近比较忙。之前写的题目一直没有更新。现在正好有点时间,更新一下。这道题木之前听学长说过,他在面...
    Richardo92阅读 282评论 0 0