找到需要补充粉笔的学生编号（约瑟夫环+前缀和+二分）

最新推荐文章于 2022-05-05 18:24:55 发布

李sy同学

最新推荐文章于 2022-05-05 18:24:55 发布

阅读量118

点赞数

文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_47511190/article/details/120214101

版权

本文探讨了两种解决约瑟夫环问题的算法：直接计算和使用前缀和结合二分查找。这两种方法的时间复杂度均为O(n)，空间复杂度为O(1)。直接计算通过循环找到指定位置，而前缀和结合二分查找则先计算前缀和，再进行查找，提高了效率。

摘要由CSDN通过智能技术生成

在这里插入图片描述

约瑟夫环：

时间复杂度：O(n)
空间复杂度：O(1)

class Solution {
public:
    int chalkReplacer(vector<int>& chalk, int k) {

        int n=chalk.size(),i=-1;
        long long sum=0;
        for(int j=0;j<n;j++)
            sum+=chalk[j];
        k%=sum;
        while(1){
            i=(i+1)%n;
            if(k<chalk[i]){
                return i;
            }else{
                k-=chalk[i];
            }
        }
        
        return 0;
    }
};

前缀和+二分：

时间复杂度：O(n)
空间复杂度：O(1)

class Solution {
public:
    int chalkReplacer(vector<int>& chalk, int k) {

        int n=chalk.size();
        if(chalk[0]>k){
            return 0;
        }
        
        //求前缀和
        for(int i=1;i<n;i++){
            chalk[i]+=chalk[i-1];
            if(chalk[i]>k){
                return i;
            }
        }

        k%=chalk[n-1];

        //二分查找
        int i=0,j=n-1;
        while(i<=j){
            int mid=(i+j)/2;
            if(chalk[mid]>k){
                j=mid-1;
            }else if(chalk[mid]<=k){
                i=mid+1;
            }
        }
        return i;
       // return upper_bound(chalk.begin(), chalk.end(), k) - chalk.begin();
    }
};