lintcode 1657 · 操作时间【simple vip 有序表】

题目

https://www.lintcode.com/problem/1657/

给出日志文件的操作，输出所有日志文件的最小和最大运行时间差的最大值是多少。


保证操作数不超过100000。
文件编号不超过100000。
单次文件运行时间不超过100000。

样例
样例 1

输入：
[1,2,1,2]
[100,300,400,500]
输出：
300

解释：
给出`op=[1,2,1,2]`代表每次的操作对应的日志文件，给出`a=[100,300,400,500]`,代表每次操作对应的时间，返回`300`。
操作1的最短运行时间为100，操作2的最短运行时间为300。
操作1的最长运行时间为400，操作2的最长运行时间为500。
操作1的差值为300，操作2的差值为200，所以返回300。

思路

针对每个文件的操作时间记录到有序表中
有序表的第一个和最后一个值的差值就是该文件的最大操作时间差

答案

public class Solution {
    /**
     * @param op: the operation
     * @param a: time of each operation
     * @return: the max sum
     */
    public int operationTime(int[] op, int[] a) {
        Map<Integer,TreeSet<Integer>> map = new HashMap<>();
        int n = op.length;
        int ans = 0;
        for (int i = 0; i < n; i++) {
            int f = op[i];
            int t = a[i];
            if(!map.containsKey(f)){
                TreeSet<Integer> ts=  new  TreeSet<Integer>();
                ts.add(t);
                map.put(f,ts);
            }else{
                map.get(f).add(t);
                ans =Math.max(ans,Math.abs(map.get(f).first()-map.get(f).last()));
                
            }
        }

        return ans;
    }
}