树状数组练习

先看一下最后一题,这是一个树状数组的题目,那就水一下吧,但是由于没有注意问题,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 分钟前
阿德莱德大学Nat. Commun.:盐模板策略实现废弃塑料到单原子催化剂的高值转化,推动环境与能源催化应用
人工智能·算法·机器学习·能源·材料工程
wan5555cn9 分钟前
多张图片生成视频模型技术深度解析
人工智能·笔记·深度学习·算法·音视频
u60637 分钟前
常用排序算法核心知识点梳理
算法·排序
蒋星熠3 小时前
Flutter跨平台工程实践与原理透视:从渲染引擎到高质产物
开发语言·python·算法·flutter·设计模式·性能优化·硬件工程
小欣加油3 小时前
leetcode 面试题01.02判定是否互为字符重排
数据结构·c++·算法·leetcode·职场和发展
3Cloudream3 小时前
LeetCode 003. 无重复字符的最长子串 - 滑动窗口与哈希表详解
算法·leetcode·字符串·双指针·滑动窗口·哈希表·中等
王璐WL3 小时前
【c++】c++第一课:命名空间
数据结构·c++·算法
空白到白4 小时前
机器学习-聚类
人工智能·算法·机器学习·聚类
索迪迈科技4 小时前
java后端工程师进修ing(研一版 || day40)
java·开发语言·学习·算法
zzzsde4 小时前
【数据结构】队列
数据结构·算法