2022-8-25 剑指offer-字典树-每日一题-二分/排序

剑指 Offer II 063. 替换单词

难度中等25收藏分享切换为英文接收动态反馈

在英语中，有一个叫做 词根(root) 的概念，它可以跟着其他一些词组成另一个较长的单词——我们称这个词为 继承词(successor)。例如，词根an，跟随着单词 other(其他)，可以形成新的单词 another(另一个)。

现在，给定一个由许多词根组成的词典和一个句子，需要将句子中的所有继承词用词根替换掉。如果继承词有许多可以形成它的词根，则用最短的词根替换它。

需要输出替换之后的句子。

class Solution {
    public String replaceWords(List<String> dictionary, String sentence) {
        Set<String> prefix=new HashSet<>();
        for (String s:dictionary){
            prefix.add(s);
        }  
        String[] arr=sentence.split(" ");
        StringBuilder sb=new StringBuilder();
        for (int i=0;i<arr.length;i++){
            boolean f=false;
            for (int j=0;j<arr[i].length();j++){
                if (prefix.contains(arr[i].substring(0,j+1))){
                    sb.append(arr[i].substring(0,j+1));
                    f=true;
                    break;
                }
            }
            if (!f) sb.append(arr[i]);
            if (i!=arr.length-1) sb.append(" ");
        }
        return sb.toString();
    }
}

思路：字典树也可以。前缀特殊标记方便判断是否有前缀。

658. 找到 K 个最接近的元素

难度中等395收藏分享切换为英文接收动态反馈

给定一个 排序好 的数组 arr ，两个整数 k 和 x ，从数组中找到最靠近 x（两数之差最小）的 k 个数。返回的结果必须要是按升序排好的。

整数 a 比整数 b 更接近 x 需要满足：

• |a - x| < |b - x| 或者
• |a - x| == |b - x| 且 a < b

class Solution {
    public List<Integer> findClosestElements(int[] arr, int k, int x) {
        PriorityQueue<Integer> q=new PriorityQueue<>(
            (a,b)->{
                if (Math.abs(a-x)==Math.abs(b-x)){
                    return a-b;
                }else{
                    return Math.abs(a-x)-Math.abs(b-x);
                }
            }
        );
        for (int num:arr){
            q.offer(num);
        }
        int[] nums=new int[k];
        for (int i=0;i<k;i++){
            nums[i]=q.poll();
        }
        Arrays.sort(nums);
        List<Integer> list=new ArrayList<>();
        for (int y:nums) list.add(y);
        return list;
    }
}

思路：可以array直接排序，或者二分查找。