Submission #1166457

#	Time	Username	Problem	Language	Result	Execution time	Memory
1166457		aminabouakaz	Pyramids (IOI24_pyramids)	C++20	100 / 100	38 ms	5036 KiB

pyramids

#include <iostream>
#include <vector>

class PyramidTransform {
public:
    PyramidTransform(const std::vector<int>& A, const std::vector<int>& B)
        : prefixSumA(A.size() + 1, 0LL), prefixSumB(B.size() + 1, 0LL) {
        for (size_t i = 0; i < A.size(); ++i) {
            prefixSumA[i + 1] = prefixSumA[i] + A[i];
            prefixSumB[i + 1] = prefixSumB[i] + B[i];
        }
    }

    bool canTransform(int L, int R, int X, int Y) {
        long long sumA = prefixSumA[R + 1] - prefixSumA[L];
        long long sumB = prefixSumB[Y + 1] - prefixSumB[X];
        return sumA == sumB;
    }

private:
    std::vector<long long> prefixSumA;
    std::vector<long long> prefixSumB;
};

PyramidTransform* pyramidTransform;

void init(std::vector<int> A, std::vector<int> B) {
    pyramidTransform = new PyramidTransform(A, B);
}

bool can_transform(int L, int R, int X, int Y) {
    return pyramidTransform->canTransform(L, R, X, Y);
}

• Subtask 0: 0 / 0
• Subtask 1: 10 / 10
• Subtask 2: 40 / 40
• Subtask 3: 20 / 20
• Subtask 4: 30 / 30