题解 | #能被多个质数整除的第K长子段#

能被多个质数整除的第K长子段

http://www.nowcoder.com/practice/162f79e68d6c4b9988a994344181d366

算法思想一:暴力法

解题思路:

根据题意:
1、首先我们准备了2到数组最大值中的所有质数,记录到primes中。
2、然后暴力遍历每一个数对,每次对于数对区间中的元素,对每一个primes中的质数,查看是否能整除区间中所有的元素,如果可以计数加1,当计数不小于时,就可以记录这对数对。
3、然后重载sort函数的比较,使其排序时按照的大小来排序
这样就可以直接在排好序的数组中找到第大的数对
图解:

代码展示:

C++版本
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 求满足条件的第K大长度
     * @param A int整型vector 牛牛的数组A
     * @param X int整型 至少要存在X个不同的质数
     * @param K int整型 牛牛希望找到第K大的长度
     * @return int整型
     */
    static bool comp(pair<int, int>& a, pair<int, int>& b){ //重载比较,按区间大小排序
        return a.second - a.first < b.second - b.first;
    }
    int GetKthLength(vector<int>& A, int X, int K) {
        // write code here
        int maxnum = *max_element(A.begin(), A.end()); //取数组最大数
        int n = A.size();
        vector<int> primes;  //记录小于最大值的所有质数
        vector<int> notPrime(maxnum + 1, 0); //辅助数组,运算质数的时候用
        notPrime[1] = 1;
        for(int i = 2; i <= maxnum; i++){ //遍历到最大值,找到所有质数
            if(!notPrime[i]) //非合数
                primes.push_back(i);
            for(int j = 0; j < primes.size() && i * primes[j] <= maxnum; j++)
                notPrime[i * primes[j]] = 1;  //这是合数
        }
        vector<pair<int, int> > leftRight;  //记录左右区间
        for(int i = 0; i < n; i++){ //遍历所有数对
            for(int j = i; j < n; j++){
                int count = 0; //统计能整除全部区间内元素的质数个数
                for(int k = 0; k < primes.size(); k++){
                    bool flag = true;
                    for(int x = i; x <= j; x++){
                        if(A[x] % primes[k] != 0){ //不能整除,这个质数不计算了
                            flag = false;
                            break;
                        }
                    }
                    if(flag)  //所有整除计数加1
                        count++;
                }
                if(count >= X) //质数个数不小于X
                    leftRight.push_back(make_pair(i, j));
            }
        }
        sort(leftRight.begin(), leftRight.end(), comp); //按照区间大小排序
        int m = leftRight.size();
        if(m < K)
            return -1;
        return leftRight[m - K].second - leftRight[m - K].first + 1; //返回第K大 
    }
};

复杂度分析

时间复杂度:,其中n为数组A的长度,m为质数数组的长度,找到所有的质数最坏遍历两层到数组A的最大值,后续嵌套了四层循环,排序和找最大值函数相对较小,可忽略。
空间复杂度:,多个辅助数组

算法思想二:方法一改进

解题思路:

方法一嵌套循环太多,复杂度较高
1、计算质数数组的时候就可以找到中每个数的质因数有多少个
2、然后排除质因数达不到的元素
3、然后可以根据子段的最大公约数来计算,最大公约能够被不小于个质数整除即可
4、可以从最长段开始找,利用二分思想找到第

代码展示:

C++版本
class Solution {
public:
    //依据长度,计算有多少个质因数,从x到结尾
    int CountMoreThanX(int x, vector<bool>& isOk, vector<vector<pair<int, int>>>& gcd){
        int s = 0, n = gcd.size();
        for(int i = 0; i < n; i++){  //遍历寻找共同的质因数个数
            for(int j = 0; j < gcd[i].size() - 1; j++){
                if(!isOk[gcd[i][j].second])
                    break;
                int len0 = i - gcd[i][j].first + 1, len1 = i - gcd[i][j + 1].first;
                if(len0 >= x)
                    s += (len1 - len0 + 1);
                else if(len1 >= x)
                    s += (len1 - x + 1);
            }
        }
        return s;
    }
    int GetKthLength(vector<int>& A, int X, int K) {
        int maxnum = *max_element(A.begin(), A.end()); //取数组最大数
        int n = A.size();
        vector <int> nums(maxnum + 1, 0); //统计每个数质因数个数
        vector <int> notPrime(maxnum + 1, 0); //记录某个数是否是质数
        vector <int> primes; //质数数组
        nums[1] = 0;
        nums[2] = 1;
        notPrime[1] = 1;
        for(int i = 2; i <= maxnum; i++){//遍历到最大值,找到所有质数
            if(!notPrime[i]){//非合数
                primes.push_back(i);
                nums[i] = 1;
            }
            for(int j = 0; j < primes.size() && i * primes[j] <= maxnum; j++){
                int p = primes[j];
                notPrime[i * p] = 1;  //这是合数
                if(i % p == 0){  //因数的个数
                    nums[i * p] = nums[i];
                    break;
                }else
                    nums[i * p] = nums[i] + 1;
            }
        }
        vector<bool> isOk(maxnum + 1, false);
        for(int i = 2; i <= maxnum; i++) //有不小于x个质因数的才标记为1
            isOk[i] = (nums[i] >= X);
        vector<vector<pair<int, int> > > gcd(n);
        for(int i = 0; i < n; i++){
            gcd[i].push_back(make_pair(i, A[i]));
            if(i > 0){
                for(auto pre : gcd[i - 1]){
                    int curgcd = __gcd(A[i], pre.second); //计算前后两个数的最大公约数
                    if(curgcd != gcd[i].back().second) //公约数不等于后面来的数
                        gcd[i].push_back(make_pair(pre.first, curgcd));  //加入前面的数
                }
            }
        }
        for(int i = 0; i < n; i++)
            gcd[i].push_back(make_pair(-1, 0));
        if(CountMoreThanX(1, isOk, gcd) < K) //数对数量小于K
            return -1;
        int l = 1, r = n;
        while (l < r) { //二分法找第k大
            int mid = l + ((r - l + 1) / 2);
            if (CountMoreThanX(mid, isOk, gcd) < K)
                r = mid - 1;
            else
                l = mid;
        }
        return l;
    }
};

复杂度分析

时间复杂度:,二分法这层为,二分法里面的函数是,找到所有的质数最坏遍历两层到数组A最大值,找最大值函数相对较小,可忽略。
空间复杂度:,多个辅助数组


全部评论

相关推荐

joe2333:怀念以前大家拿华为当保底的日子
点赞 评论 收藏
分享
整顿职场的柯基很威猛:这种不可怕,最可怕的是夹在一帮名校里的二本选手,人家才是最稳的。
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务