Codeforces · 2195B

Heapify 1

This C++ solution uses disjoint set union for Codeforces 2195B Heapify 1. Read the reasoning, inspect the code, or try your own test case below.

2195BGraphs & treesDisjoint set unionC++50 lines
Solution110of 248
Open official problem ↗ Download C++ file ↓ Search the library → Open full Code Lab ↗ Report an issue ↗

Approach

Disjoint set union

Heapify 1 operations and component structure generated by i to 2i match the DSU computation.

Graphs & trees

Problem and code

Useful links.

Written by benbenyaojifen. Try the problem first, then compare your approach with the code.

Open official problem ↗View exact source file ↗
Implementation

B_Heapify_1.cpp

C++

    #include <bits/stdc++.h>
    using namespace std;
    using ll = long long;
    const int inf = 1e9;
    const long long INF = 1e17; //❄️
    vector<int> parent;
    int find(int x) {
        return x == parent[x] ? x : parent[x] = find(parent[x]);
    }
    void unite(int x, int y) {
        x = find(x); y = find(y);
        parent[x] = y;
    }
    void solve() {
        int n; cin >> n;
        parent.resize(n + 1);
        for (int i = 1; i <= n; i++) parent[i] = i;
        vector<int> v(n + 1);
        for (int i = 1; i <= n; i++) {
            cin >> v[i];
        }
        for (int i = 1; i <= n / 2; i++) {
            unite(i, 2 * i);
        }
        unordered_map<int, vector<int>> indx, val;
        for (int i = 1; i <= n; i++) {
            int p = find(i);
            indx[p].push_back(i);
            val[p].push_back(v[i]);
        }
        for (auto &a : indx) {
            auto &index = a.second;
            auto &value = val[a.first];
            sort(index.begin(), index.end());
            sort(value.begin(), value.end());
            if (index != value) {
                cout << "No" << '\n';
                return;
            }
        }
        cout << "Yes" << '\n';
    }
    int main() {
        ios::sync_with_stdio(0); cin.tie(0); 
        int t; cin >> t;
        while (t--) {
            solve();
        }
        return 0;
    }
        

Did this explanation save you time? I'm a Grade 11 student building this free library to make difficult algorithms easier to understand.

Buy me a coffee ↗Keep studying →

Test this problem

Run your code here.

Paste your code, run a test case, compare the output, or trace selected values.

Full trace, comparison & stress testing ↗
StatusReady
Output
No run yet.
Diagnostics
No diagnostics yet.

Each run is isolated and has strict limits. Passing one test does not guarantee the judge will accept the solution.