Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
127 changes: 127 additions & 0 deletions bit_manipulation/sliding_window_xor.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,127 @@
/**
* @file
* @brief Implementation to [calculate XOR of sliding window of size k in an
* array of n integers] (https://cses.fi/problemset/task/3426)
*
* @details
* We are given an array of n integers. Our task is to calculate the bitwise XOR
* of each window of k elements, from left to right, and cumulatively XOR the
* results into a single value.
*
* Worst Case Time Complexity: O(n)
* Space Complexity: O(n)
*
* @author [Abhiraj Mandal](https://github.com/DataWorshipper)
*/

#include <cassert> /// for assert
#include <cstdint> /// for std::uint32_t
#include <iostream> /// for IO operations
#include <vector> /// for std::vector

/**
* @namespace bit_manipulation
* @brief Bit manipulation algorithms
*/
namespace bit_manipulation {
/**
* @namespace sliding_window_xor
* @brief Functions for cumulative XOR of sliding windows in arrays
*/
namespace sliding_window_xor {

/**
* @brief Computes cumulative XOR of all windows of size k
*
* @param n Size of the array
* @param k Window size
* @param x Initial value to generate the array
* @param multiplier Multiplier in array generation
* @param increment Increment in array generation
* @param modulo Modulo in array generation
* @returns std::uint64_t The cumulative XOR of all windows of size k
*
* @details
* This function generates the array using the recurrence:
* arr[0] = x
* arr[i] = (multiplier * arr[i-1] + increment) % modulo
*
* It maintains a sliding window of size k using two pointers l and r:
* - x1 stores the XOR of the current window
* - x2 stores the cumulative XOR of all valid windows
*
* This approach ensures that the algorithm runs in O(n) time.
*/
std::uint64_t compute(std::uint64_t n, std::uint64_t k, std::uint64_t x,
std::uint64_t multiplier, std::uint64_t increment,
std::uint64_t modulo) {
// Generate the array of n elements
std::vector<std::uint64_t> arr(n);
arr[0] = x; // First element of the array

for (std::uint64_t i = 1; i < n; ++i) {
arr[i] = (multiplier * arr[i - 1] + increment) %
modulo; // recurrence relation
}

std::uint64_t x1 = 0; // XOR of the current window
std::uint64_t x2 = 0; // Cumulative XOR of all windows of size k
std::uint64_t left = 0; // Left pointer of sliding window
std::uint64_t right = 0; // Right pointer of sliding window

// Slide the window over the array
while (right < n) {
x1 ^= arr[right]; // include current element in window XOR

// Shrink window from left if size exceeds k
while (right - left + 1 > k) {
x1 ^= arr[left]; // remove leftmost element from window XOR
++left;
}

// If window size equals k, add it to cumulative XOR
if (right - left + 1 == k) {
x2 ^= x1;
}

++right; // Move right pointer
}

return x2; // Return cumulative XOR of all windows
}

} // namespace sliding_window_xor
} // namespace bit_manipulation

/**
* @brief Self-test implementation
*/
static void test() {
using bit_manipulation::sliding_window_xor::compute;

// Testcase 1: n = 100, k = 20, expected = 1019
assert(compute(100, 20, 3, 7, 1, 997) == 1019);

// Testcase 2: n = 2, k = 1, expected = 2
assert(compute(2, 1, 2, 3, 4, 5) == 2);

// Testcase 3: n = 5, k = 2
assert(compute(5, 2, 1, 1, 1, 100) == 0 ^ 3 ^ 1 ^ 7);

// Testcase 4: n = 3, k = 5, expected = 0
assert(compute(3, 5, 5, 2, 1, 100) == 0);

// Testcase 5: n = 4, k = 4, expected = 0
assert(compute(4, 4, 3, 1, 0, 10) == 0);

std::cout << "All test cases successfully passed!" << std::endl;
}

/**
* @brief Main function
* @returns 0 on exit
*/
int main() {
test(); // run self-test implementations
return 0;
}
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
}
}