面试题—三个数之和小于某个值的组合个数

对着背影说爱祢 提交于 2020-03-05 13:59:07

本题是阿里蚂蚁金服面试题:

给定一个数组s,和一个数a,在数组s里面找到3个数使这3个数的和小于a,即s[i]+s[j]+s[k]<a,请你设计一个函数返回数组s里这样的组合有多少个?

   int f(double s[], double a) {
        int cnt = 0;
        Arrays.sort(s);
        int i = 0;
        int j = s.length;
        while (i < j - 1) {
            int low = i;
            int high = j;
            while (low < high) {
                int k = (low + high) / 2;

                if (s[i] + s[j] + s[k] >= a) {
                    high = k - 1;
                } else if (s[i] + s[j] + s[k] < a) {
                    // 下面区间都是满足的
                    cnt += k - low;
                    // 二分查找
                    low = k + 1;
                }
            }

            // 扫描结束,根据情况移动上下限
            if (s[i] + s[j] > a) {
                j--;
            } else {
                i++;
            }
        }

        return cnt;
    }
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!