树状数组练习

先看一下最后一题,这是一个树状数组的题目,那就水一下吧,但是由于没有注意问题,wa了很多次

cpp 复制代码
const int N = (int)1e5 + 5;
int n;
int flag[N];
int dp[N];
class Solution {
public:
    vector<int> countOfPeaks(vector<int>& nums, vector<vector<int>>& queries) {
        n = nums.size();
        memset(dp,0,sizeof dp);
        memset(flag,0,sizeof flag);
        vector<int> ans;
        for (int i = 1; i < n - 1; i++) {
            if (nums[i] > nums[i + 1] && nums[i] > nums[i - 1]) {
                flag[i + 1] = 1;
                add(i + 1, 1);
            }
        }
        //         for (int i = 1; i <= n; i++) {
        //     cout << i << " " << find(i) - find(i - 1) << endl;
        // }
        for (int i = 0; i < queries.size(); i++) {
            if (queries[i][0] == 2) {
                int index = queries[i][1], val = queries[i][2];
                nums[index] = val;
                fun(index, nums); fun(index - 1, nums); fun(index + 1, nums);
        //                 for (int i = 1; i <= n; i++) {
        //     cout << i << " " << find(i) - find(i - 1) << endl;
        // }
            }
            else {

                 //ans.push_back(find(queries[i][2]) - find(queries[i][1] + 2));
                 int temp = find(queries[i][2] + 1) - find(queries[i][1]);
                 if (flag[queries[i][2] + 1]) temp--;
                 if (flag[queries[i][1] + 1]) temp--;
                 ans.push_back(temp>0?temp:0);
            }
        }
        return ans;
    }

    int lowbit(int x) {
        return x & (-x);
    }
    void add(int x, int p) {
        for (int i = x; i <= n; i += lowbit(i)) {
            dp[i] += p;
        }
    }
    int find(int x) {
        if (x == 0) return 0;
        return dp[x] + find(x - lowbit(x));
    }
    void fun(int index, vector<int>& nums) {
        if (index <= 0 || index >= (n - 1)) return;
        if (nums[index] > nums[index + 1] && nums[index] > nums[index - 1]) {
            if (flag[index + 1]) return;
            add(index + 1, 1), flag[index + 1] = 1;
        }
        else {
            if (!flag[index + 1]) return;
            add(index + 1, -1), flag[index + 1] = 0;
        }
    }
};
相关推荐
IT小番茄31 分钟前
Kubernetes云平台管理实战:自动加载到负载均衡(七)
算法
笑口常开xpr36 分钟前
【C++继承】深入浅出C++继承机制
开发语言·数据结构·c++·算法
让我们一起加油好吗1 小时前
【基础算法】DFS
算法·深度优先
爱学习的小鱼gogo3 小时前
python 矩阵中寻找就接近的目标值 (矩阵-中等)含源码(八)
开发语言·经验分享·python·算法·职场和发展·矩阵
红纸2813 小时前
Subword算法之WordPiece、Unigram与SentencePiece
人工智能·python·深度学习·神经网络·算法·机器学习·自然语言处理
CUMT_DJ4 小时前
从零复现论文(1)——通感一体化实现协作基站分配与资源分配(CBARA)策略
算法·通感一体化
tt5555555555554 小时前
CSDN 教程:C++ 经典字符串与栈算法题逐行详解
c++·算法·哈希算法
_dindong4 小时前
基础算法:滑动窗口
数据结构·学习·算法·leetcode·力扣
Voyager_45 小时前
图像处理踩坑:浮点数误差导致的缩放尺寸异常与解决办法
数据结构·图像处理·人工智能·python·算法
文艺倾年5 小时前
【八股消消乐】手撕分布式协议和算法(基础篇)
分布式·算法