feat: add count distinct primes from binary string algorithm

bit_manipulation/count_distinct_primes_from_binary_string.cpp

@@ -0,0 +1,85 @@
+/**
+ * @file count_distinct_primes_from_binary_string.cpp
+ * @brief Count distinct primes formed from binary strings using allowed operations.
+ * 
+ * @author Rudraksh Tank
+ * @date July 2025
+ *
+ * @details
+ * Given a binary string, the task is to count how many distinct prime decimal numbers
+ * can be formed by:
+ * - Swapping any two characters (makes position irrelevant)
+ * - Changing any '1' to '0' (not the reverse)
+ *
+ * Efficient solution using bit manipulation and Sieve of Eratosthenes.
+ *
+ * Tags: Bit Manipulation, Prime Numbers, Combinatorics, Greedy, Bitmask
+ */
+
+#include <bit>
+#include <iostream>
+#include <vector>
+#include <unordered_set>
+#include <algorithm>
+
+const int MAX = 1e6;
+static std::vector<bool> is_prime;
+
+/**
+ * @namespace bit_manipulation
+ * @brief Bit manipulation algorithms
+ */
+namespace bit_manipulation {
+
+    void precomputePrimes() {
+        is_prime.assign(MAX + 1, true);
+        is_prime[0] = is_prime[1] = false;
+        for (int i = 2; i * i <= MAX; i++) {
+            if (is_prime[i]) {
+                for (int j = i * i; j <= MAX; j += i) {
+                    is_prime[j] = false;
+                }
+            }
+        }
+    }
+
+    /**
+     * @brief Counts distinct prime numbers that can be formed from the given binary string.
+     * @param s Binary string input
+     * @return Number of distinct primes possible after allowed transformations
+     */
+    int countPrimeBinaryStrings(const std::string &s) {
+        int n = s.length();
+        int k = std::count(s.begin(), s.end(), '1');
+        int cnt = 0;
+        int limit = 1 << n;
+
+        std::unordered_set<int> seen;
+
+        for (int i = 2; i < limit; i++) {
+            if (std::popcount(i) <= k && is_prime[i]) {
+                if (!seen.count(i)) {
+                    cnt++;
+                    seen.insert(i);
+                }
+            }
+        }
+
+        return cnt;
+    }
+
+    void tests(){
+        precomputePrimes();
+        std::string s;
+        std::cin >> s;
+        std::cout << countPrimeBinaryStrings(s) << std::endl;
+    }
+
+} //bit manipulation
+/**
+ * @brief Main function to test the algorithm.
+ */
+int main() {
+    tests();
+    return 0;
+}