【每日一题】补档 CF371 D. Vessels | 并查集 | 简单

题目内容

原题链接

从上到下有 n n n 个碗,每个碗中水满了之后会往下一个碗流,最后一个碗水满了后会直接流出去。

现在有两种操作:

  • 1 i x 表示往第 i i i 个碗里加 x x x 的水
  • 2 i 表示查询第 i i i 个碗里现在的水量

给定你 m m m 次操作,每次操作是上面两种之一,当面临第二种操作时,请输出第 i i i 个碗里现在的水量。

数据范围

  • 1 ≤ n , m ≤ 2 ⋅ 1 0 5 1\leq n,m\leq 2\cdot 10^5 1≤n,m≤2⋅105
  • 1 ≤ i ≤ n 1\leq i\leq n 1≤i≤n
  • 1 ≤ x ≤ 1 0 9 1\leq x\leq 10^9 1≤x≤109

题解

复杂在于如果当前加入水的第 i i i 个碗及其之后的若干个碗都是满的,如何快速找到第一个没满的碗 j j j,水必然是流向 j j j 的。

并查集维护当前碗 i i i 指向的第一个没满的碗 j j j 即可。

时间复杂度: O ( n log ⁡ n ) O(n\log n) O(nlogn)

代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;

const int MAXBIT = 30;
const int MOD = 1e9 + 7;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n;
    cin >> n;

    vector<int> a(n);
    vector<int> cur(n);
    for (int i = 0; i < n; ++i) cin >> a[i];

    vector<int> p(n + 1);
    iota(p.begin(), p.end(), 0);

    auto find = [&](int x) {
        int root = x;
        while (root != p[root]) root = p[root];
        while (x != p[x]) {
            int nx = p[x];
            p[x] = root;
            x = nx;
        }
        return root;
    };

    int m;
    cin >> m;

    while (m--) {
        int op, idx, x;
        cin >> op >> idx;
        idx--;

        int fa = find(idx);

        if (op == 1) {
            cin >> x;
            if (fa == n) continue;
            int l = fa;
            while (x > 0 && l < n) {
                int need = min(x, a[l] - cur[l]);
                cur[l] += need;
                x -= need;
                if (cur[l] == a[l]) l += 1;
            }
            for (int i = fa; i < l; ++i) p[find(i)] = find(l);
        } else {
            cout << cur[idx] << "\n";
        }
    }

    return 0;
}
相关推荐
焜昱错眩..36 分钟前
代码随想录训练营二十六天| 654.最大二叉树 617.合并二叉树 700.二叉搜索树的搜索 98.验证二叉搜索树
数据结构·算法
jndingxin43 分钟前
OpenCV CUDA模块中用于稠密光流计算的 TV-L1(Dual TV-L1)算法类cv::cuda::OpticalFlowDual_TVL1
人工智能·opencv·算法
geneculture1 小时前
路径=算法=操作:复杂系统行为的统一数学框架
人工智能·算法·数学建模·课程设计·智慧系统·融智学的重要应用·复杂系统
AcrelGHP1 小时前
建筑末端配电回路安全用电解决方案:筑牢电气防火最后一道防线
人工智能·算法·安全
Watink Cpper1 小时前
[灵感源于算法] 链表类问题技巧总结
数据结构·算法·链表
PassLink_3 小时前
AlgorithmVisualizer项目改进与部署-网页算法可视化
算法·编程·开源项目·本地部署·算法可视化·源码改进
GalaxyPokemon4 小时前
LeetCode - 2. 两数相加
java·前端·javascript·算法·leetcode·职场和发展
编程绿豆侠4 小时前
力扣HOT100之堆:347. 前 K 个高频元素
算法·leetcode·哈希算法
GalaxyPokemon6 小时前
归并排序:分治思想的高效排序
数据结构·算法·排序算法
ThreeYear_s6 小时前
基于FPGA的PID算法学习———实现PI比例控制算法
学习·算法·fpga开发