added all files
This commit is contained in:
parent
868900e6d5
commit
919496f25f
35
.vscode/settings.json
vendored
Normal file
35
.vscode/settings.json
vendored
Normal file
@ -0,0 +1,35 @@
|
||||
{
|
||||
"files.associations": {
|
||||
"*.ts": "typescriptreact",
|
||||
"vector": "cpp",
|
||||
"iostream": "cpp",
|
||||
"array": "cpp",
|
||||
"bitset": "cpp",
|
||||
"__bit_reference": "cpp",
|
||||
"__split_buffer": "cpp",
|
||||
"cctype": "cpp",
|
||||
"clocale": "cpp",
|
||||
"cmath": "cpp",
|
||||
"complex": "cpp",
|
||||
"cstddef": "cpp",
|
||||
"cstdint": "cpp",
|
||||
"cstdlib": "cpp",
|
||||
"cstring": "cpp",
|
||||
"execution": "cpp",
|
||||
"memory": "cpp",
|
||||
"initializer_list": "cpp",
|
||||
"ios": "cpp",
|
||||
"iosfwd": "cpp",
|
||||
"istream": "cpp",
|
||||
"limits": "cpp",
|
||||
"locale": "cpp",
|
||||
"ostream": "cpp",
|
||||
"stdexcept": "cpp",
|
||||
"streambuf": "cpp",
|
||||
"string": "cpp",
|
||||
"string_view": "cpp",
|
||||
"tuple": "cpp",
|
||||
"typeinfo": "cpp",
|
||||
"algorithm": "cpp"
|
||||
}
|
||||
}
|
684
bignum.h
Normal file
684
bignum.h
Normal file
@ -0,0 +1,684 @@
|
||||
#pragma once
|
||||
|
||||
#include <math.h>
|
||||
#include <stdint.h>
|
||||
#include <limits.h>
|
||||
|
||||
#include <vector>
|
||||
#include <algorithm>
|
||||
#include <iostream>
|
||||
|
||||
class Num {
|
||||
public:
|
||||
typedef uint64_t word;
|
||||
|
||||
std::vector<word> words;
|
||||
bool neg;
|
||||
|
||||
static word word_mask(){
|
||||
return (word)-1;
|
||||
}
|
||||
|
||||
static size_t word_bits(){
|
||||
return sizeof(word)*CHAR_BIT;
|
||||
}
|
||||
|
||||
static word word_half_mask(){
|
||||
return word_mask() >> word_bits()/2;
|
||||
}
|
||||
|
||||
static word char_to_word(char c){
|
||||
switch (c){
|
||||
case '0': return 0;
|
||||
case '1': return 1;
|
||||
case '2': return 2;
|
||||
case '3': return 3;
|
||||
case '4': return 4;
|
||||
case '5': return 5;
|
||||
case '6': return 6;
|
||||
case '7': return 7;
|
||||
case '8': return 8;
|
||||
case '9': return 9;
|
||||
case 'a': case 'A': return 10;
|
||||
case 'b': case 'B': return 11;
|
||||
case 'c': case 'C': return 12;
|
||||
case 'd': case 'D': return 13;
|
||||
case 'e': case 'E': return 14;
|
||||
case 'f': case 'F': return 15;
|
||||
case 'g': case 'G': return 16;
|
||||
case 'h': case 'H': return 17;
|
||||
case 'i': case 'I': return 18;
|
||||
case 'j': case 'J': return 19;
|
||||
case 'k': case 'K': return 20;
|
||||
case 'l': case 'L': return 21;
|
||||
case 'm': case 'M': return 22;
|
||||
case 'n': case 'N': return 23;
|
||||
case 'o': case 'O': return 24;
|
||||
case 'p': case 'P': return 25;
|
||||
case 'q': case 'Q': return 26;
|
||||
case 'r': case 'R': return 27;
|
||||
case 's': case 'S': return 28;
|
||||
case 't': case 'T': return 29;
|
||||
case 'u': case 'U': return 30;
|
||||
case 'v': case 'V': return 31;
|
||||
case 'w': case 'W': return 32;
|
||||
case 'x': case 'X': return 33;
|
||||
case 'y': case 'Y': return 34;
|
||||
case 'z': case 'Z': return 35;
|
||||
default: return word_mask();
|
||||
}
|
||||
}
|
||||
|
||||
static word word_gcd(word a, word b){
|
||||
while (1){
|
||||
if (a == 0) return b;
|
||||
b %= a;
|
||||
if (b == 0) return a;
|
||||
a %= b;
|
||||
}
|
||||
}
|
||||
|
||||
Num(): neg(false){}
|
||||
|
||||
Num(size_t n, word w, bool neg = false): words(n, w), neg(neg){}
|
||||
|
||||
Num(const word *a, const word *b, bool neg = false) : words(a, b), neg(neg){
|
||||
truncate();
|
||||
}
|
||||
|
||||
Num(const Num &a){
|
||||
words = a.words;
|
||||
neg = a.neg;
|
||||
}
|
||||
|
||||
Num& operator = (const Num &a){
|
||||
words = a.words;
|
||||
neg = a.neg;
|
||||
return *this;
|
||||
}
|
||||
|
||||
Num(int i): neg(i < 0){
|
||||
unsigned u = (i < 0) ? -(unsigned)i : (unsigned)i;
|
||||
if (sizeof(u) <= word_bits()){
|
||||
if (u > 0) push_back(u);
|
||||
}else{
|
||||
for (; u; u >>= word_bits()) push_back(u);
|
||||
}
|
||||
}
|
||||
|
||||
Num(const char *c, word base = 10, char **endptr = NULL): neg(false){
|
||||
// read sign
|
||||
if (*c == '-'){
|
||||
c++;
|
||||
neg = true;
|
||||
}
|
||||
// read digits
|
||||
for (;*c; c++){
|
||||
mul_word(base);
|
||||
word b = char_to_word(*c);
|
||||
if (b >= base) break;
|
||||
add_word(b);
|
||||
}
|
||||
if (endptr) *endptr = (char*)c;
|
||||
}
|
||||
|
||||
void resize(size_t n){
|
||||
words.resize(n);
|
||||
}
|
||||
|
||||
void pop_back(){
|
||||
words.pop_back();
|
||||
}
|
||||
|
||||
void push_back(word b){
|
||||
words.push_back(b);
|
||||
}
|
||||
|
||||
word& back(){
|
||||
return words.back();
|
||||
}
|
||||
|
||||
const word& back() const {
|
||||
return words.back();
|
||||
}
|
||||
|
||||
size_t size() const {
|
||||
return words.size();
|
||||
}
|
||||
|
||||
word& operator [] (size_t i){
|
||||
return words[i];
|
||||
}
|
||||
|
||||
const word& operator [] (size_t i) const {
|
||||
return words[i];
|
||||
}
|
||||
|
||||
Num& set_neg(bool neg){
|
||||
this->neg = neg;
|
||||
return *this;
|
||||
}
|
||||
|
||||
Num& truncate(){
|
||||
while (size() > 0 && words.back() == 0) pop_back();
|
||||
return *this;
|
||||
}
|
||||
|
||||
size_t bitlength() const {
|
||||
if (size() == 0) return 0;
|
||||
size_t last = size() - 1;
|
||||
size_t result = word_bitlength((*this)[last]) + last*word_bits();
|
||||
return result;
|
||||
}
|
||||
|
||||
size_t count_trailing_zeros() const {
|
||||
for (size_t i = 0; i < size(); i++){
|
||||
word w = (*this)[i];
|
||||
if (w) return word_count_trailing_zeros(w) + i*word_bits();
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cmp_abs(const Num &a, const Num &b){
|
||||
size_t na = a.size(), nb = b.size();
|
||||
if (na != nb){
|
||||
return na < nb ? -1 : +1;
|
||||
}
|
||||
for (size_t i = na; i --> 0;){
|
||||
word wa = a[i], wb = b[i];
|
||||
if (wa != wb){
|
||||
return wa < wb ? -1 : +1;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cmp(const Num &a, const Num &b){
|
||||
if (a.size() == 0 && b.size() == 0) return 0;
|
||||
if (!a.neg && !b.neg) return +cmp_abs(a, b);
|
||||
if ( a.neg && b.neg) return -cmp_abs(a, b);
|
||||
return a.neg && !b.neg ? -1 : +1;
|
||||
}
|
||||
|
||||
static size_t word_bitlength(word a){
|
||||
for (int i = word_bits() - 1; i >= 0; i--) if ((a >> i) & 1) return i+1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
static size_t word_count_trailing_zeros(word a){
|
||||
for (int i = 0; i < (int)word_bits(); i++) if ((a >> i) & 1) return i;
|
||||
return word_bits();
|
||||
}
|
||||
|
||||
static word add_carry(word *a, word b){
|
||||
return (*a += b) < b;
|
||||
}
|
||||
|
||||
static word sub_carry(word *a, word b){
|
||||
word tmp = *a;
|
||||
return (*a = tmp - b) > tmp;
|
||||
}
|
||||
|
||||
static word word_mul_hi(word a, word b){
|
||||
size_t n = word_bits()/2;
|
||||
word a_hi = a >> n;
|
||||
word a_lo = a & word_half_mask();
|
||||
word b_hi = b >> n;
|
||||
word b_lo = b & word_half_mask();
|
||||
word tmp = ((a_lo*b_lo) >> n) + a_hi*b_lo;
|
||||
tmp = (tmp >> n) + ((a_lo*b_hi + (tmp & word_half_mask())) >> n);
|
||||
return tmp + a_hi*b_hi;
|
||||
}
|
||||
|
||||
static Num& add_unsigned_overwrite(Num &a, const Num &b){
|
||||
size_t i, na = a.size(), nb = b.size(), n = std::max(na, nb);
|
||||
a.resize(n);
|
||||
word carry = 0;
|
||||
for (i = 0; i < nb; i++){
|
||||
carry = add_carry(&a[i], carry);
|
||||
carry += add_carry(&a[i], b[i]);
|
||||
}
|
||||
for (; i < n && carry; i++) carry = add_carry(&a[i], carry);
|
||||
if (carry) a.push_back(carry);
|
||||
return a.truncate();
|
||||
}
|
||||
|
||||
static Num& sub_unsigned_overwrite(Num &a, const Num &b){
|
||||
//assert(cmp_abs(a, b) >= 0);
|
||||
size_t i, na = a.size(), nb = b.size();
|
||||
word carry = 0;
|
||||
for (i = 0; i < nb; i++){
|
||||
carry = sub_carry(&a[i], carry);
|
||||
carry += sub_carry(&a[i], b[i]);
|
||||
}
|
||||
for (; i < na && carry; i++) carry = sub_carry(&a[i], carry);
|
||||
//assert(!carry);
|
||||
return a.truncate();
|
||||
}
|
||||
|
||||
static Num mul_long(const Num &a, const Num &b){
|
||||
size_t na = a.size(), nb = b.size(), nc = na + nb + 1;
|
||||
Num c(nc, 0, a.neg ^ b.neg), carries(nc, 0);
|
||||
for (size_t ia = 0; ia < na; ia++){
|
||||
for (size_t ib = 0; ib < nb; ib++){
|
||||
size_t i = ia + ib, j = i + 1;
|
||||
// WARNING: Might overflow if word size is chosen too small
|
||||
carries[i + 1] += add_carry(&c[i], a[ia] * b[ib]);
|
||||
carries[j + 1] += add_carry(&c[j], word_mul_hi(a[ia], b[ib]));
|
||||
}
|
||||
}
|
||||
return add_unsigned_overwrite(c, carries).truncate();
|
||||
}
|
||||
|
||||
static Num mul_karatsuba(const Num &a, const Num &b){
|
||||
size_t na = a.size(), nb = b.size(), n = std::max(na, nb), m2 = n/2 + (n & 1);
|
||||
Num a_parts[2], b_parts[2];
|
||||
split(a, a_parts, 2, m2);
|
||||
split(b, b_parts, 2, m2);
|
||||
m2 *= word_bits();
|
||||
Num z0 = a_parts[0] * b_parts[0];
|
||||
Num z1 = (a_parts[0] + a_parts[1]) * (b_parts[0] + b_parts[1]);
|
||||
Num z2 = a_parts[1] * b_parts[1];
|
||||
Num result = z2;
|
||||
result <<= m2;
|
||||
result += z1 - z2 - z0;
|
||||
result <<= m2;
|
||||
result += z0;
|
||||
return result;
|
||||
}
|
||||
|
||||
static Num mul(const Num &a, const Num &b){
|
||||
size_t karatsuba_threshold = 20;
|
||||
if (a.size() > karatsuba_threshold && b.size() > karatsuba_threshold){
|
||||
return mul_karatsuba(a, b);
|
||||
}
|
||||
return mul_long(a, b);
|
||||
}
|
||||
|
||||
static Num add_signed(const Num &a, bool a_neg, const Num &b, bool b_neg){
|
||||
if (a_neg == b_neg) return add_unsigned(a, b).set_neg(a_neg);
|
||||
if (cmp_abs(a, b) >= 0) return sub_unsigned(a, b).set_neg(a_neg);
|
||||
return sub_unsigned(b, a).set_neg(b_neg);
|
||||
}
|
||||
|
||||
Num& operator >>= (size_t n_bits){
|
||||
if (n_bits == 0) return *this;
|
||||
size_t n_words = n_bits / word_bits();
|
||||
if (n_words >= size()){
|
||||
resize(0);
|
||||
return *this;
|
||||
}
|
||||
n_bits %= word_bits();
|
||||
if (n_bits == 0){
|
||||
for (size_t i = 0; i < size() - n_words; i++){
|
||||
(*this)[i] = (*this)[i + n_words];
|
||||
}
|
||||
}else{
|
||||
word hi, lo = (*this)[n_words];
|
||||
for (size_t i = 0; i < size() - n_words - 1; i++){
|
||||
hi = (*this)[i + n_words + 1];
|
||||
(*this)[i] = (hi << (word_bits() - n_bits)) | (lo >> n_bits);
|
||||
lo = hi;
|
||||
}
|
||||
(*this)[size() - n_words - 1] = lo >> n_bits;
|
||||
}
|
||||
resize(size() - n_words);
|
||||
return truncate();
|
||||
}
|
||||
|
||||
Num& operator <<= (size_t n_bits){
|
||||
if (n_bits == 0) return *this;
|
||||
size_t n_words = n_bits / word_bits();
|
||||
n_bits %= word_bits();
|
||||
size_t old_size = size();
|
||||
size_t n = old_size + n_words + (n_bits != 0);
|
||||
resize(n);
|
||||
if (n_bits == 0){
|
||||
for (size_t i = n; i --> n_words;){
|
||||
(*this)[i] = (*this)[i - n_words];
|
||||
}
|
||||
}else{
|
||||
word lo, hi = 0;
|
||||
for (size_t i = n - 1; i > n_words; i--){
|
||||
lo = (*this)[i - n_words - 1];
|
||||
(*this)[i] = (hi << n_bits) | (lo >> (word_bits() - n_bits));
|
||||
hi = lo;
|
||||
}
|
||||
(*this)[n_words] = hi << n_bits;
|
||||
}
|
||||
for (size_t i = 0; i < n_words; i++) (*this)[i] = 0;
|
||||
return truncate();
|
||||
}
|
||||
|
||||
static void div_mod(const Num &numerator, Num denominator, Num "ient, Num &remainder){
|
||||
quotient = 0;
|
||||
remainder = numerator;
|
||||
if (cmp_abs(remainder, denominator) >= 0){
|
||||
int n = numerator.bitlength() - denominator.bitlength();
|
||||
denominator <<= n;
|
||||
for (; n >= 0; n--){
|
||||
if (cmp_abs(remainder, denominator) >= 0){
|
||||
sub_unsigned_overwrite(remainder, denominator);
|
||||
quotient.set_bit(n);
|
||||
}
|
||||
denominator >>= 1;
|
||||
}
|
||||
}
|
||||
quotient.set_neg(numerator.neg ^ denominator.neg);
|
||||
remainder.set_neg(numerator.neg);
|
||||
}
|
||||
|
||||
static void div_mod_half_word(const Num &numerator, word denominator, Num "ient, word &remainder){
|
||||
remainder = 0;
|
||||
Num dst(numerator.size(), 0);
|
||||
|
||||
for (size_t i = numerator.size(); i --> 0;){
|
||||
word dst_word = 0;
|
||||
word src_word = numerator[i];
|
||||
word parts[2];
|
||||
parts[0] = src_word >> word_bits()/2;
|
||||
parts[1] = src_word & word_half_mask();
|
||||
|
||||
for (size_t j = 0; j < 2; j++){
|
||||
remainder <<= word_bits()/2;
|
||||
remainder |= parts[j];
|
||||
|
||||
word div_word = remainder / denominator;
|
||||
word mod_word = remainder % denominator;
|
||||
remainder = mod_word;
|
||||
|
||||
dst_word <<= word_bits()/2;
|
||||
dst_word |= div_word;
|
||||
}
|
||||
|
||||
dst[i] = dst_word;
|
||||
}
|
||||
|
||||
quotient = dst.truncate().set_neg(numerator.neg);
|
||||
}
|
||||
|
||||
static void split(const Num &a, Num *parts, size_t n_parts, size_t n){
|
||||
size_t i = 0;
|
||||
for (size_t k = 0; k < n_parts; k++){
|
||||
Num &part = parts[k];
|
||||
part.resize(n);
|
||||
for (size_t j = 0; j < n && i < a.size(); j++) part[j] = a[i++];
|
||||
part = part.truncate();
|
||||
}
|
||||
}
|
||||
|
||||
static Num div(const Num &numerator, const Num& denominator){
|
||||
Num quotient, remainder;
|
||||
div_mod(numerator, denominator, quotient, remainder);
|
||||
return quotient;
|
||||
}
|
||||
|
||||
static Num mod(const Num &numerator, const Num& denominator){
|
||||
Num quotient, remainder;
|
||||
div_mod(numerator, denominator, quotient, remainder);
|
||||
return remainder;
|
||||
}
|
||||
|
||||
static Num add_unsigned(const Num &a, const Num &b){
|
||||
Num result(a);
|
||||
return add_unsigned_overwrite(result, b);
|
||||
}
|
||||
|
||||
static Num sub_unsigned(const Num &a, const Num &b){
|
||||
Num result(a);
|
||||
return sub_unsigned_overwrite(result, b);
|
||||
}
|
||||
|
||||
static Num add(const Num &a, const Num &b){
|
||||
Num result = add_signed(a, a.neg, b, b.neg);
|
||||
return result;
|
||||
}
|
||||
|
||||
static Num sub(const Num &a, const Num &b){
|
||||
Num result = add_signed(a, a.neg, b, !b.neg);
|
||||
return result;
|
||||
}
|
||||
|
||||
static Num gcd(const Num &a0, const Num &b0){
|
||||
|
||||
if (a0.size() == 1 && b0.size() == 1){
|
||||
return Num(1, word_gcd(a0[0], b0[0]));
|
||||
}
|
||||
|
||||
Num a(a0), b(b0);
|
||||
a.neg = b.neg = false;
|
||||
|
||||
if (a.size() == 0) return b0;
|
||||
if (b.size() == 0) return a0;
|
||||
|
||||
size_t n = a.count_trailing_zeros();
|
||||
size_t m = b.count_trailing_zeros();
|
||||
if (n > m) {
|
||||
std::swap(n, m);
|
||||
a.words.swap(b.words);
|
||||
}
|
||||
|
||||
a >>= n;
|
||||
b >>= n;
|
||||
|
||||
do {
|
||||
b >>= b.count_trailing_zeros();
|
||||
if (cmp_abs(a, b) > 0) a.words.swap(b.words);
|
||||
sub_unsigned_overwrite(b, a);
|
||||
} while (b.size() > 0);
|
||||
|
||||
a <<= n;
|
||||
|
||||
return a;
|
||||
}
|
||||
|
||||
typedef void (*random_func)(uint8_t *bytes, size_t n_bytes);
|
||||
|
||||
static Num random_bits(size_t n_bits, random_func func){
|
||||
if (n_bits == 0) return 0;
|
||||
size_t partial_bits = n_bits % word_bits();
|
||||
size_t n_words = n_bits / word_bits() + (partial_bits > 0);
|
||||
size_t n_bytes = n_words*sizeof(word);
|
||||
Num result(n_words, 0);
|
||||
uint8_t *bytes = (uint8_t*)&result[0];
|
||||
func(bytes, n_bytes);
|
||||
if (partial_bits){
|
||||
size_t too_many_bits = word_bits() - partial_bits;
|
||||
result.back() >>= too_many_bits;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
static Num random_inclusive(const Num &inclusive, random_func func){
|
||||
size_t n_bits = inclusive.bitlength();
|
||||
while (true){
|
||||
Num result = random_bits(n_bits, func);
|
||||
if (result <= inclusive) return result;
|
||||
}
|
||||
}
|
||||
|
||||
static Num random_exclusive(const Num &exclusive, random_func func){
|
||||
size_t n_bits = exclusive.bitlength();
|
||||
while (true){
|
||||
Num result = random_bits(n_bits, func);
|
||||
if (result < exclusive) return result;
|
||||
}
|
||||
}
|
||||
|
||||
static Num random_second_exclusive(const Num &inclusive_min_val, const Num &exclusive_max_val, random_func func){
|
||||
return inclusive_min_val + random_exclusive(exclusive_max_val - inclusive_min_val, func);
|
||||
}
|
||||
|
||||
static Num random_both_inclusive(const Num &inclusive_min_val, const Num &inclusive_max_val, random_func func){
|
||||
return inclusive_min_val + random_inclusive(inclusive_max_val - inclusive_min_val, func);
|
||||
}
|
||||
|
||||
Num& set_bit(size_t i){
|
||||
size_t i_word = i / word_bits();
|
||||
size_t i_bit = i % word_bits();
|
||||
if (size() <= i_word) resize(i_word + 1);
|
||||
(*this)[i_word] |= ((word)1) << i_bit;
|
||||
return *this;
|
||||
}
|
||||
|
||||
word get_bit(size_t i) const {
|
||||
size_t i_word = i / word_bits();
|
||||
size_t i_bit = i % word_bits();
|
||||
if (i_word >= size()) return 0;
|
||||
return ((*this)[i_word] >> i_bit) & 1;
|
||||
}
|
||||
|
||||
void clr_bit(size_t i){
|
||||
size_t i_word = i / word_bits();
|
||||
size_t i_bit = i % word_bits();
|
||||
if (i_word >= size()) return;
|
||||
word mask = 1;
|
||||
mask <<= i_bit;
|
||||
(*this)[i_word] &= ~mask;
|
||||
}
|
||||
|
||||
Num& mul_word(word b){
|
||||
word carry = 0;
|
||||
for (size_t i = 0; i < size(); i++){
|
||||
word a = (*this)[i];
|
||||
word tmp = a * b;
|
||||
carry = add_carry(&tmp, carry);
|
||||
carry += word_mul_hi(a, b);
|
||||
(*this)[i] = tmp;
|
||||
}
|
||||
if (carry) push_back(carry);
|
||||
return truncate();
|
||||
}
|
||||
|
||||
Num& add_word(word carry, size_t i0 = 0){
|
||||
if (i0 >= size()) resize(i0 + 1);
|
||||
for (size_t i = i0; i < size() && carry; i++){
|
||||
carry = add_carry(&(*this)[i], carry);
|
||||
}
|
||||
if (carry) push_back(carry);
|
||||
return truncate();
|
||||
}
|
||||
|
||||
void print(
|
||||
std::vector<char> &text,
|
||||
word base = 10,
|
||||
const char *alphabet = "0123456789abcdefghijklmnopqrstuvwxyz"
|
||||
) const {
|
||||
if (size() == 0){
|
||||
text.push_back('0');
|
||||
}else{
|
||||
Num tmp(*this);
|
||||
while (tmp.size() > 0){
|
||||
word remainder;
|
||||
div_mod_half_word(tmp, base, tmp, remainder);
|
||||
text.push_back(alphabet[remainder]);
|
||||
}
|
||||
if (neg) text.push_back('-');
|
||||
std::reverse(text.begin(), text.end());
|
||||
}
|
||||
text.push_back('\0');
|
||||
}
|
||||
|
||||
friend std::ostream& operator << (std::ostream& out, const Num &num){
|
||||
std::vector<char> tmp;
|
||||
num.print(tmp);
|
||||
out << &tmp[0];
|
||||
return out;
|
||||
}
|
||||
|
||||
double to_double() const {
|
||||
if (size() == 0) return 0.0;
|
||||
double d = 0.0, base = ::pow(2.0, word_bits());
|
||||
for (size_t i = size(); i --> 0;) d = d * base + (*this)[i];
|
||||
return neg ? -d : d;
|
||||
}
|
||||
|
||||
bool can_convert_to_int(int *result){
|
||||
if (*this < Num(INT_MIN) || *this > Num(INT_MAX)) return false;
|
||||
|
||||
unsigned temp = 0;
|
||||
|
||||
if (word_bits() >= sizeof(temp)*CHAR_BIT){
|
||||
if (words.size() > 0){
|
||||
temp = (*this)[0];
|
||||
}
|
||||
}else{
|
||||
for (size_t i = words.size(); i --> 0;){
|
||||
temp <<= word_bits();
|
||||
temp += (*this)[i];
|
||||
}
|
||||
}
|
||||
|
||||
*result = neg ? -temp : temp;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
Num pow(size_t exponent) const {
|
||||
Num result(1), p(*this);
|
||||
for (; exponent; exponent >>= 1){
|
||||
if (exponent & 1){
|
||||
result = result * p;
|
||||
exponent--;
|
||||
}
|
||||
p = p*p;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
Num mod_pow(Num exponent, const Num &modulus) const {
|
||||
Num result(1), base = (*this) % modulus;
|
||||
for (; exponent.size() > 0; exponent >>= 1){
|
||||
if (exponent.get_bit(0)){
|
||||
result = (result * base) % modulus;
|
||||
}
|
||||
base = (base * base) % modulus;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
Num sqrt() const {
|
||||
Num n = *this;
|
||||
int bit = bitlength();
|
||||
if (bit & 1) bit ^= 1;
|
||||
Num result = 0;
|
||||
for (; bit >= 0; bit -= 2){
|
||||
Num tmp = result;
|
||||
tmp.set_bit(bit);
|
||||
if (n >= tmp){
|
||||
n -= tmp;
|
||||
result.set_bit(bit + 1);
|
||||
}
|
||||
result >>= 1;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
Num& operator ++(){
|
||||
add_word(1);
|
||||
return *this;
|
||||
}
|
||||
|
||||
Num& operator += (const Num &b){ return *this = add(*this, b); }
|
||||
Num& operator -= (const Num &b){ return *this = sub(*this, b); }
|
||||
Num& operator *= (const Num &b){ return *this = mul(*this, b); }
|
||||
Num& operator /= (const Num &b){ return *this = div(*this, b); }
|
||||
Num& operator %= (const Num &b){ return *this = mod(*this, b); }
|
||||
|
||||
bool operator == (const Num &b) const { return cmp(*this, b) == 0; }
|
||||
bool operator != (const Num &b) const { return cmp(*this, b) != 0; }
|
||||
bool operator <= (const Num &b) const { return cmp(*this, b) <= 0; }
|
||||
bool operator >= (const Num &b) const { return cmp(*this, b) >= 0; }
|
||||
bool operator < (const Num &b) const { return cmp(*this, b) < 0; }
|
||||
bool operator > (const Num &b) const { return cmp(*this, b) > 0; }
|
||||
|
||||
Num operator + (const Num &b) const { return add(*this, b); }
|
||||
Num operator - (const Num &b) const { return sub(*this, b); }
|
||||
Num operator * (const Num &b) const { return mul(*this, b); }
|
||||
Num operator / (const Num &b) const { return div(*this, b); }
|
||||
Num operator % (const Num &b) const { return mod(*this, b); }
|
||||
Num operator - ( ) const { return Num(*this).set_neg(!neg); }
|
||||
|
||||
Num operator >> (size_t n_bits) const { return Num(*this) >>= n_bits; }
|
||||
Num operator << (size_t n_bits) const { return Num(*this) <<= n_bits; }
|
||||
};
|
38
bindings.cpp
Normal file
38
bindings.cpp
Normal file
@ -0,0 +1,38 @@
|
||||
#include <emscripten/bind.h>
|
||||
#include "cards.h"
|
||||
|
||||
using namespace emscripten;
|
||||
|
||||
emscripten::val textToPackOfCardsJS(std::string text) {
|
||||
std::array<int, 52> cards = textToPackOfCards(text);
|
||||
return emscripten::val::array(cards.begin(), cards.end());
|
||||
}
|
||||
|
||||
std::string packOfCardsToTextJS(emscripten::val jsArray) {
|
||||
std::array<int, 52> cards;
|
||||
for (int i = 0; i < 52; i++) {
|
||||
cards[i] = jsArray[i].as<int>();
|
||||
}
|
||||
return packOfCardsToText(cards);
|
||||
}
|
||||
|
||||
std::string packOfCardsToTextEncryptedJS(emscripten::val jsArray, std::string key) {
|
||||
std::array<int, 52> cards;
|
||||
for (int i = 0; i < 52; i++) {
|
||||
cards[i] = jsArray[i].as<int>();
|
||||
}
|
||||
return packOfCardsToTextEncrypted(cards, key);
|
||||
}
|
||||
|
||||
emscripten::val textToPackOfCardsEncryptedJS(std::string text, std::string key) {
|
||||
std::array<int, 52> cards = textToPackOfCardsEncrypted(text, key);
|
||||
return emscripten::val::array(cards.begin(), cards.end());
|
||||
}
|
||||
|
||||
EMSCRIPTEN_BINDINGS(cardcode){
|
||||
function("textToPackOfCards", &textToPackOfCardsJS);
|
||||
function("packOfCardsToText", &packOfCardsToTextJS);
|
||||
function("packOfCardsToTextEncrypted", &packOfCardsToTextEncryptedJS);
|
||||
function("textToPackOfCardsEncrypted", &textToPackOfCardsEncryptedJS);
|
||||
}
|
||||
|
1
build_web.sh
Executable file
1
build_web.sh
Executable file
@ -0,0 +1 @@
|
||||
emcc -std=c++20 bindings.cpp cards.cpp -o web/public/libs/cardcode.js -s WASM=1 -s EXPORTED_RUNTIME_METHODS=['ccall','cwrap'] -s MODULARIZE=1 -s NO_DISABLE_EXCEPTION_CATCHING -s EXPORT_NAME='createModule' -s ENVIRONMENT=web -lembind -s ASSERTIONS=1 --bind
|
172
cards.cpp
Normal file
172
cards.cpp
Normal file
@ -0,0 +1,172 @@
|
||||
#include <bitset>
|
||||
#include <cstdint>
|
||||
#include <vector>
|
||||
#include "bignum.h"
|
||||
#include <set>
|
||||
#include <iostream>
|
||||
#include <algorithm>
|
||||
#include <array>
|
||||
#include "cards.h"
|
||||
|
||||
std::vector<Num> factorials;
|
||||
std::string alphabet_str = " .,-\"/abcdefghijklmnopqrstuvwxyz";
|
||||
std::vector<char> alphabet = std::vector<char>(alphabet_str.begin(), alphabet_str.end());
|
||||
|
||||
Num getFactorial(Num n){
|
||||
if(factorials.size() == 0){
|
||||
for(Num i=0; i<1000; i+=1){
|
||||
if(i==0) factorials.push_back(Num(1));
|
||||
else factorials.push_back(factorials.back() * i);
|
||||
}
|
||||
}
|
||||
return factorials[n.to_double()];
|
||||
}
|
||||
|
||||
std::vector<Num> decimalToFactoradic(Num decimal){
|
||||
std::vector<Num> result;
|
||||
int divisor = 1;
|
||||
while(decimal > 0){
|
||||
result.insert(result.begin(), (decimal % divisor));
|
||||
decimal /= divisor;
|
||||
divisor++;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
Num factoradicToDecimal(std::vector<Num> factoradic){
|
||||
Num result(0);
|
||||
for(int i=0; i<factoradic.size(); i++){
|
||||
// result += factoradic[i]*factorials[factoradic.size()-i-1];
|
||||
result += factoradic[i] * getFactorial(factoradic.size()-i-1);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
std::vector<Num> decimalToPermutation(Num decimal, Num permutationSize){
|
||||
std::vector<Num> factoradic = decimalToFactoradic(decimal);
|
||||
while(factoradic.size()<permutationSize.to_double()){
|
||||
factoradic.insert(factoradic.begin(), 0);
|
||||
}
|
||||
|
||||
std::vector<Num> available;
|
||||
for(Num i(0); i<permutationSize.to_double(); i+=1){
|
||||
available.push_back(i);
|
||||
}
|
||||
|
||||
std::vector<Num> permutation;
|
||||
|
||||
for(int i=0; i<factoradic.size(); i++){
|
||||
permutation.push_back(available[factoradic[i].to_double()]);
|
||||
available.erase(available.begin()+factoradic[i].to_double());
|
||||
}
|
||||
|
||||
return permutation;
|
||||
}
|
||||
|
||||
Num permutationToDecimal(std::vector<Num> permutation){
|
||||
std::vector<Num> factoradic;
|
||||
|
||||
std::vector<Num> numbers;
|
||||
for(Num i = 0; i<permutation.size(); i+=1){
|
||||
numbers.push_back(i);
|
||||
}
|
||||
std::cout << "numbers size: "<<numbers.size()<<"\n";
|
||||
for(Num num : permutation){
|
||||
Num index = std::find(numbers.begin(), numbers.end(), num) - numbers.begin();
|
||||
factoradic.push_back(index);
|
||||
numbers.erase(numbers.begin()+index.to_double());
|
||||
}
|
||||
std::cout << "factoradic size: "<<factoradic.size()<<"\n";
|
||||
return factoradicToDecimal(factoradic);
|
||||
|
||||
}
|
||||
|
||||
std::bitset<225> textToBitset(std::string text){
|
||||
std::bitset<225> data = 0;
|
||||
|
||||
for(int i=0; i<text.size(); i++){
|
||||
data <<= 5;
|
||||
uint8_t character = std::find(alphabet.begin(), alphabet.end(), text[i]) - alphabet.begin();
|
||||
data |= character;
|
||||
}
|
||||
data <<= (5*(45-text.size()));
|
||||
return data;
|
||||
}
|
||||
|
||||
std::string bitsetToText(std::bitset<225> data){
|
||||
std::vector<uint64_t> words(4,0);
|
||||
for(int i=0; i<225; i++){
|
||||
int index = i / 64;
|
||||
int bit = i % 64;
|
||||
if(data[i]){
|
||||
words[index] |= uint64_t(1)<<bit;
|
||||
}
|
||||
}
|
||||
|
||||
Num num(words.data(), words.data()+words.size());
|
||||
std::string result = "";
|
||||
for(int i=0; i<45; i++){
|
||||
uint8_t index = (num % 32).to_double();
|
||||
num /= 32;
|
||||
result = alphabet[index]+result;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
std::array<int, 52> binaryToPackOfCards(std::bitset<225> data){
|
||||
std::vector<uint64_t> words(4,0);
|
||||
for(int i=0; i<225; i++){
|
||||
int index = i / 64;
|
||||
int bit = i % 64;
|
||||
if(data[i]){
|
||||
words[index] |= uint64_t(1)<<bit;
|
||||
}
|
||||
}
|
||||
|
||||
Num num(words.data(), words.data()+words.size());
|
||||
|
||||
std::cout << "bin: "<<data.to_string()<<"\n";
|
||||
|
||||
std::cout << "num: "<<num<<"\n";
|
||||
std::vector<Num> permutation = decimalToPermutation(num, 52);
|
||||
std::array<int, 52> cards;
|
||||
for(int i=0; i<52; i++){
|
||||
cards[i]=permutation[i].to_double();
|
||||
}
|
||||
std::cout << "permsize: "<<cards.size() <<"\n";
|
||||
return cards;
|
||||
}
|
||||
std::array<int, 52> textToPackOfCards(std::string text){
|
||||
std::bitset<225> data = textToBitset(text);
|
||||
return binaryToPackOfCards(data);
|
||||
}
|
||||
std::array<int, 52> textToPackOfCardsEncrypted(std::string text, std::string key){
|
||||
std::bitset<225> data = textToBitset(text);
|
||||
std::bitset<225> keyData = textToBitset(key);
|
||||
data ^= keyData;
|
||||
return binaryToPackOfCards(data);
|
||||
|
||||
}
|
||||
std::string packOfCardsToText(std::array<int, 52> cards){
|
||||
std::string result = "";
|
||||
std::vector<Num> permutation;
|
||||
for(int i=0; i<52; i++){
|
||||
permutation.push_back(Num(cards[i]));
|
||||
}
|
||||
std::cout << "permsize: "<<permutation.size() <<"\n";
|
||||
Num decimal(permutationToDecimal(permutation));
|
||||
std::cout << "decimal: "<<decimal<<"\n";
|
||||
for(int i=0; i<45; i++){
|
||||
uint8_t index = (decimal % 32).to_double();
|
||||
decimal /= 32;
|
||||
result = alphabet[index]+result;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
std::string packOfCardsToTextEncrypted(std::array<int, 52> cards, std::string key){
|
||||
std::string text = packOfCardsToText(cards);
|
||||
std::bitset<225> data = textToBitset(text);
|
||||
std::bitset<225> keyData = textToBitset(key);
|
||||
data ^= keyData;
|
||||
return bitsetToText(data);
|
||||
}
|
8
cards.h
Normal file
8
cards.h
Normal file
@ -0,0 +1,8 @@
|
||||
#pragma once
|
||||
#include "bignum.h"
|
||||
#include <array>
|
||||
|
||||
std::array<int, 52> textToPackOfCards(std::string text);
|
||||
std::string packOfCardsToText(std::array<int, 52> cards);
|
||||
std::array<int, 52> textToPackOfCardsEncrypted(std::string text, std::string key);
|
||||
std::string packOfCardsToTextEncrypted(std::array<int, 52> cards, std::string key);
|
117
main.py
Normal file
117
main.py
Normal file
@ -0,0 +1,117 @@
|
||||
factorials = []
|
||||
alphabet = " .,-\"/abcdefghijklmnopqrstuvwxyz"
|
||||
cards = [
|
||||
'A♠', '2♠', '3♠', '4♠', '5♠', '6♠', '7♠', '8♠', '9♠', '10♠', 'J♠', 'Q♠', 'K♠',
|
||||
'A♥', '2♥', '3♥', '4♥', '5♥', '6♥', '7♥', '8♥', '9♥', '10♥', 'J♥', 'Q♥', 'K♥',
|
||||
'A♦', '2♦', '3♦', '4♦', '5♦', '6♦', '7♦', '8♦', '9♦', '10♦', 'J♦', 'Q♦', 'K♦',
|
||||
'A♣', '2♣', '3♣', '4♣', '5♣', '6♣', '7♣', '8♣', '9♣', '10♣', 'J♣', 'Q♣', 'K♣'
|
||||
]
|
||||
for i in range(1000):
|
||||
if(i==0):
|
||||
factorials.append(1)
|
||||
else:
|
||||
factorials.append(i*factorials[i-1])
|
||||
|
||||
def convertToFactoradic(number: int) -> [int]:
|
||||
result = []
|
||||
divisor = 1
|
||||
while(number > 0):
|
||||
result.insert(0, number % divisor)
|
||||
number //= divisor
|
||||
divisor+=1
|
||||
return result
|
||||
|
||||
def convertToDenary(number: [int]) -> int:
|
||||
global factorials
|
||||
result = 0
|
||||
numlen = len(number)
|
||||
for i in range(numlen):
|
||||
result += number[i]*factorials[numlen-i-1]
|
||||
return result
|
||||
|
||||
# Will return a list of permutationElements size 0 indexed elements in the order from the number
|
||||
# The number msut be less than permutationElements!
|
||||
def convertToPermutation(number: int, permutationElements: int) -> [int]:
|
||||
global factorials
|
||||
if(number >= factorials[permutationElements]):
|
||||
raise ValueError("Given number is outside of permutation range")
|
||||
return
|
||||
|
||||
factoradic = convertToFactoradic(number)
|
||||
|
||||
while len(factoradic) < permutationElements:
|
||||
factoradic.insert(0,0)
|
||||
|
||||
available = list(range(permutationElements))
|
||||
|
||||
permutation = []
|
||||
|
||||
for i in factoradic:
|
||||
permutation.append(available[i])
|
||||
available.pop(i)
|
||||
return permutation
|
||||
|
||||
def convertToDecimal(permutation: [int]) -> int:
|
||||
|
||||
permlength = len(permutation)
|
||||
factoradic = []
|
||||
|
||||
numbers = list(range(permlength))
|
||||
|
||||
for i in permutation:
|
||||
index = numbers.index(i)
|
||||
factoradic.append(index)
|
||||
numbers.pop(index)
|
||||
return convertToDenary(factoradic)
|
||||
|
||||
def textToPackOfCards(text: str) -> [int]:
|
||||
global alphabet
|
||||
if(len(text) > 45):
|
||||
raise ValueError("Text must be <=45 characters to be represented in one deck of cards")
|
||||
for i in text.lower():
|
||||
if not i in alphabet:
|
||||
raise ValueError("Text must only be characters A-Z or '., -\"/' in order to use 5bit")
|
||||
return
|
||||
data = 0
|
||||
for i in text.lower():
|
||||
data<<=(4 if not i else 5)
|
||||
data|=alphabet.index(i)
|
||||
print(bin(data))
|
||||
data <<= (5*(45-len(text)))
|
||||
print(f"bin: {bin(data)}")
|
||||
print(f"num: {data}")
|
||||
return convertToPermutation(data, 52)
|
||||
def packOfCardsToText(cards: [int]) -> str:
|
||||
global alphabet
|
||||
if(len(cards)!=52):
|
||||
return ValueError("Pack must have 52 distinct 0 indexed cards")
|
||||
testcards = list(range(52))
|
||||
for i in cards:
|
||||
testcards.pop(testcards.index(i))
|
||||
if(len(testcards)>0):
|
||||
return ValueError("Pack must have 52 distinct 0 indexed cards")
|
||||
number = convertToDecimal(cards)
|
||||
result = ""
|
||||
binstring = bin(number)[2:]
|
||||
while(len(binstring)<45*5):
|
||||
binstring = "0" + binstring;
|
||||
for i in range(45):
|
||||
x = int(binstring[i*5:(i+1)*5], 2)
|
||||
result += alphabet[x]
|
||||
return result
|
||||
|
||||
|
||||
# This is 45 characters of text which can be stored in a pack of 52 playing cards
|
||||
# abcdefghijklmnopqrstuvwxyzabcdefghijklnnopqrs
|
||||
#tnum = 22460398060890687638644604344344764536689096168995043803493220483072
|
||||
#print(convertToPermutation(tnum, 52))
|
||||
#print(convertToDecimal(convertToPermutation(tnum, 52)))
|
||||
tstr = "hello world"
|
||||
secretcards = textToPackOfCards(tstr)
|
||||
|
||||
#for i in secretcards:
|
||||
# print(cards[i])
|
||||
print(textToPackOfCards(tstr))
|
||||
print(packOfCardsToText(textToPackOfCards(tstr)))
|
||||
print(packOfCardsToText([0, 24, 4, 46, 19, 51, 20, 23, 21, 7, 14, 48, 43, 34, 17, 15, 13, 3, 42, 36, 50, 25, 32, 12, 33, 27, 1, 47, 37, 39, 31, 2, 18, 6, 22, 38, 30, 41,
|
||||
28, 10, 26, 29, 40, 35, 49, 16, 45, 8, 5, 9, 11, 44]))
|
Loading…
x
Reference in New Issue
Block a user