small updates

Author: rishabhdeepsingh

Cpp/library/generators.hpp

@@ -3,6 +3,12 @@
 
 namespace GEN {
 
+template <typename T>
+T rand(T low, T high) {
+  std::uniform_int_distribution<T> uni(low, high);
+  return uni(rng);
+}
+
 template <typename T>
 vector<vector<T>> subarrays(vector<T>& arr) {
   int n = arr.size();

Cpp/library/number/Sieve.hpp

@@ -56,6 +56,28 @@ struct Sieve {
     return ans;
   }
 
+  long long countDivisors(long long n) {
+    if (n == 1) return 1;
+    long long ans = 1;
+    for (int i = 0;; i++) {
+      if (primes[i] * primes[i] * primes[i] > n) break;
+      int cnt = 1;
+      while (n % primes[i] == 0) {
+        n /= primes[i];
+        cnt++;
+      }
+      ans *= cnt;
+    }
+    
+    if (isPrime[n])
+      ans = ans * 2;
+    else if (isPrime[sqrt(n)] && (long long) sqrt(n) == sqrt(n))
+      ans = ans * 3;
+    else if (n != 1)
+      ans = ans * 4;
+    return ans;
+  }
+  
   vector<pll> divisorPair(long long x) {
     vector<pll> res;
     for (long long i = 1; i * i <= x; i++) {

Cpp/output/test.sh

@@ -6,5 +6,5 @@ for ((i = 1; ; ++i)); do
   # ./a < int > out1
   # ./brute < int > out2
   # diff -w out1 out2 || break
-  diff -w <(./a <int) <(./brute <int) || break
+  diff -w <(./main <int) <(./brute <int) || break
 done