test/Hash/zobrist_hash/abc238-g.cpp

View this file on GitHub
Last update: 2025-03-23 20:36:27+09:00
Problem: https://atcoder.jp/contests/abc238/tasks/abc238_g

Depends on

Code

// competitive-verifier: PROBLEM https://atcoder.jp/contests/abc238/tasks/abc238_g
// competitive-verifier: IGNORE
// AtCoder's test cases are now private.

#include "Hash/zobrist_cubic_hash.hpp"
#include <bits/stdc++.h>

using namespace std;

int main()
{
    int N, Q;
    cin >> N >> Q;
    vector<long long> A(N);
    for (int i = 0; i < N; i++)
    {
        cin >> A[i];
    }

    ZobristCubicHash hashA(A);

    for (int i = 0; i < Q; i++)
    {
        int L, R;
        cin >> L >> R;
        L--;

        if (hashA.get_range_hash(L, R) == 0)
        {
            cout << "Yes" << endl;
        }
        else
        {
            cout << "No" << endl;
        }
    }

    return 0;
}

#line 1 "test/Hash/zobrist_hash/abc238-g.cpp"
// competitive-verifier: PROBLEM https://atcoder.jp/contests/abc238/tasks/abc238_g
// competitive-verifier: IGNORE
// AtCoder's test cases are now private.

#line 1 "Hash/zobrist_hash_base.hpp"
#include <bits/stdc++.h>
using namespace std;

class ZobristHashBase
{
private:
    static unsigned long long splitmix64(uint64_t x)
    {
        x += 0x9e3779b97f4a7c15;
        x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
        x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
        return x ^ (x >> 31);
    }

    static unsigned long long generate_random_value(unsigned long long x)
    {
        static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();
        return splitmix64(x + FIXED_RANDOM);
    }

protected:
    static unsigned long long hash(long long x)
    {
        static unordered_map<long long, unsigned long long> cache;
        if (cache.contains(x))
        {
            return cache[x];
        }

        cache[x] = generate_random_value(x);
        return cache[x];
    };

    virtual void build(const vector<long long> &v) = 0;
    ZobristHashBase() {};

public:
    vector<unsigned long long> hash_vector;
    virtual unsigned long long get_range_hash(int l, int r) const = 0;
    virtual ~ZobristHashBase() = default;
};
#line 2 "Math/prime_factorize.hpp"
using namespace std;

vector<pair<long long, long long>> prime_factorize(long long N)
{
    assert(0 < N);

    long long tempN = N;
    vector<pair<long long, long long>> dst;
    for (long long i = 2; i * i <= tempN; i++)
    {
        if (N % i == 0)
        {
            dst.push_back(make_pair(i, 0));
            while (N % i == 0)
            {
                dst.back().second++;
                N /= i;
            }
        }
    }
    if (N != 1)
        dst.push_back(make_pair(N, 1));
    return dst;
}
#line 3 "Hash/zobrist_cubic_hash.hpp"

#line 5 "Hash/zobrist_cubic_hash.hpp"
using namespace std;

class ZobristCubicHash : private ZobristHashBase
{
private:
    void build(const vector<long long> &v) override
    {
        hash_vector = vector<unsigned long long>(v.size() + 1, 0);

        unordered_map<long long, long long> prime_counts;
        for (int i = 0; i < v.size(); i++)
        {
            hash_vector[i + 1] = hash_vector[i];
            for (auto [p, cnt] : prime_factorize(v[i]))
            {
                cnt %= 3;
                for (int j = 0; j < cnt; j++)
                {
                    prime_counts[p]++;
                    if (prime_counts[p] % 3 == 0)
                    {
                        hash_vector[i + 1] ^= ZobristHashBase::hash(p);
                    }
                    if (prime_counts[p] % 3 == 1)
                    {
                        hash_vector[i + 1] ^= ZobristHashBase::hash(p * p);
                    }
                    if (prime_counts[p] % 3 == 2)
                    {
                        hash_vector[i + 1] ^= ZobristHashBase::hash(p) ^ ZobristHashBase::hash(p * p);
                    }
                }
            }
        }
    }

public:
    unsigned long long get_range_hash(int l, int r) const override
    {
        assert(0 <= l && l < r && r < hash_vector.size());
        return hash_vector[r] ^ hash_vector[l];
    }

    ZobristCubicHash(const vector<long long> &v)
    {
        build(v);
    };
};
#line 7 "test/Hash/zobrist_hash/abc238-g.cpp"

using namespace std;

int main()
{
    int N, Q;
    cin >> N >> Q;
    vector<long long> A(N);
    for (int i = 0; i < N; i++)
    {
        cin >> A[i];
    }

    ZobristCubicHash hashA(A);

    for (int i = 0; i < Q; i++)
    {
        int L, R;
        cin >> L >> R;
        L--;

        if (hashA.get_range_hash(L, R) == 0)
        {
            cout << "Yes" << endl;
        }
        else
        {
            cout << "No" << endl;
        }
    }

    return 0;
}