N个建筑，选择3个作为埋伏点，要求距离小于D

最新推荐文章于 2020-10-06 20:41:48 发布

青盏

最新推荐文章于 2020-10-06 20:41:48 发布

阅读量362

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/qq_16234613/article/details/89600915

LeetCode 专栏收录该内容

154 篇文章

订阅专栏

本文探讨了一道算法题目，给定N个已排序的建筑位置，选择3个作为埋伏点，要求这3个点的距离小于D。通过双指针技巧，高效地计算出所有符合条件的组合数量。

试题
N个建筑，选择3个作为埋伏点，要求3个埋伏点距离小于D。输入已排序好，求方案数量。
代码
以end表示当前建筑必须包含在方案里。使用前后两个指针维持所有与end距离小于D的建筑。

import java.util.*;

public class Main{
    public static void main(String[] args){
//        int[] A = {1,10,20,30,50};
//        int d = 19;
        int[] A = {1,2,3,4};
        int d = 3;

        if(A==null || A.length<3) System.out.println(0);
        int len = A.length;

        int total=0;
        int start=0,end=2;
        int count = 3;

        while(end<len){
            while(start<end && A[end]-A[start]>d){
                start ++;
                count--;
            }
            if(count>2){
                total += (count-2)*(count-1)/2;
            }
            end++;
            count++;
        }
        System.out.println(total);
    }
}