树状数组练习

先看一下最后一题,这是一个树状数组的题目,那就水一下吧,但是由于没有注意问题,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;
        }
    }
};
相关推荐
电子_咸鱼2 分钟前
LeetCode——Hot 100【电话号码的字母组合】
数据结构·算法·leetcode·链表·职场和发展·贪心算法·深度优先
仰泳的熊猫3 分钟前
LeetCode:785. 判断二分图
数据结构·c++·算法·leetcode
rit84324998 分钟前
基于MATLAB实现基于距离的离群点检测算法
人工智能·算法·matlab
my rainy days2 小时前
C++:友元
开发语言·c++·算法
haoly19892 小时前
数据结构和算法篇-归并排序的两个视角-迭代和递归
数据结构·算法·归并排序
微笑尅乐2 小时前
中点为根——力扣108.讲有序数组转换为二叉搜索树
算法·leetcode·职场和发展
im_AMBER3 小时前
算法笔记 05
笔记·算法·哈希算法
夏鹏今天学习了吗3 小时前
【LeetCode热题100(46/100)】从前序与中序遍历序列构造二叉树
算法·leetcode·职场和发展
吃着火锅x唱着歌3 小时前
LeetCode 2389.和有限的最长子序列
算法·leetcode·职场和发展
嶔某3 小时前
二叉树的前中后序遍历(迭代)
算法