树状数组练习

先看一下最后一题,这是一个树状数组的题目,那就水一下吧,但是由于没有注意问题,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;
        }
    }
};
相关推荐
NAGNIP11 小时前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
美团技术团队12 小时前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法
Fanxt_Ja17 小时前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下17 小时前
最终的信号类
开发语言·c++·算法
茉莉玫瑰花茶17 小时前
算法 --- 字符串
算法
博笙困了17 小时前
AcWing学习——差分
c++·算法
NAGNIP17 小时前
认识 Unsloth 框架:大模型高效微调的利器
算法
NAGNIP17 小时前
大模型微调框架之LLaMA Factory
算法
echoarts17 小时前
Rayon Rust中的数据并行库入门教程
开发语言·其他·算法·rust
Python技术极客17 小时前
一款超好用的 Python 交互式可视化工具,强烈推荐~
算法