123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922 |
- use crate::group::*;
- use std::string::String;
- const SIZE: u8 = 3;
- const WIDTH: u8 = SIZE*SIZE;
- const MAX_SIZE: u8 = WIDTH * WIDTH;
- use bit_field::BitField;
- extern crate rand_chacha;
- use rand::seq::SliceRandom;
- use rand_chacha::rand_core::SeedableRng;
- use rand_chacha::ChaCha20Rng;
- use rand::distributions::{Distribution, Uniform};
- #[derive(Debug, Copy, Clone, PartialEq)]
- pub struct Bits(u16);
- pub const fn set_bits(bits: u8) -> u16 {
- (1 << (bits)) - 1
- }
- impl Bits {
-
- pub fn clear(&mut self) {
- self.0 = 0;
- }
-
- pub fn set(&mut self, bit: u8, value: bool) {
- self.0.set_bit(bit as usize, value);
- }
-
- pub fn get(&self, bit: u8) -> bool {
- self.0.get_bit(bit as usize)
- }
-
- pub fn set_bits(&mut self, bits: u8) {
- self.0 = set_bits(bits);
- }
-
- pub fn count_set(&self) -> u8 {
- let mut count = 0;
- for i in 0..u16::BIT_LENGTH {
- if self.get(i as u8) {
- count += 1;
- }
- }
- count
- }
- }
- struct BitsIterator<'a> {
- possible: &'a Bits,
- index: u8,
- }
- impl Bits {
- fn iter(&self) -> BitsIterator {
- BitsIterator {
- possible: self,
- index: 1,
- }
- }
- }
- impl<'a> Iterator for BitsIterator<'a> {
- type Item = u8;
- fn next(&mut self) -> Option<u8> {
- while (self.index < u16::BIT_LENGTH as u8) && (!self.possible.get(self.index)) {
- self.index += 1;
-
- }
- if self.index == u16::BIT_LENGTH as u8 {
- None
- } else {
- self.index += 1;
- Some(self.index - 1)
- }
- }
- }
- #[cfg(test)]
- mod tests {
- use crate::sudoku::*;
- #[test]
- fn check_possible_bitset() {
- let mut p = Bits(0);
- p.clear();
- for i in 0..9 {
- let mut result = p.get(i);
- assert_eq!(result, false);
- p.set(i, true);
- result = p.get(i);
- assert_eq!(result, true);
- }
- }
- #[test]
- fn check_possible_iter() {
- let mut p = Bits(0);
- p.set(3, true);
- p.set(5, true);
- p.set(6, true);
- assert_eq!(3, p.count_set());
- let values: Vec<u8> = p.iter().collect();
- assert_eq!(values, vec!(3, 5, 6));
- assert_eq!(3, p.count_set());
- p.set(0, true);
- assert_eq!(4, p.count_set());
- }
- #[test]
- fn check_bits() {
-
- let p = Bits(set_bits(6));
- for i in 0..6 {
- let result = p.get(i);
- assert_eq!(result, true);
- }
- assert_eq!(p.get(6), false);
- }
- }
- pub type SudokuBoard = [u8; MAX_SIZE as usize];
- pub type Possible = [Bits; MAX_SIZE as usize];
- pub type SudokuPossible = [Bits; MAX_SIZE as usize];
- #[derive(Debug, Clone, Copy)]
- pub struct Board([u8; MAX_SIZE as usize]);
- #[derive(Debug, Clone, Copy)]
- pub struct BoardPossible([Bits; MAX_SIZE as usize]);
-
- impl BoardPossible {
-
-
- pub fn display(&self) {
- for y in 0..WIDTH {
- for x in 0..WIDTH {
-
- let mut possible = String::new();
- for p in self.0[pos(x, y) as usize].iter() {
-
- possible += format!("{},", p).as_str();
- }
-
-
- print!("({},{}):", x, y);
-
- print!("{:9}", possible);
-
-
-
- }
- println!("");
- }
- }
- }
- #[derive(Debug, Clone, Copy)]
- pub struct Sudoku {
- pub board: [u8; MAX_SIZE as usize],
- pub possible: [Bits; MAX_SIZE as usize],
- }
- pub const fn pos(x: u8, y: u8) -> u8 {
- x + (y * WIDTH as u8)
- }
- pub const fn pos1(x: u8, y: u8) -> u8 {
- (x - 1) + ((y - 1) * WIDTH as u8)
- }
- pub const fn xy(pos: u8) -> (u8, u8) {
- ((pos % WIDTH), (pos / WIDTH))
- }
- const DEBUG_OUTPUT: bool = false;
- impl Sudoku {
- pub fn new() -> Self {
-
-
-
- let mut initial: Bits = Bits(set_bits(10));
- initial.set(0, false);
- let s = Sudoku {
- board: [0; MAX_SIZE as usize],
- possible: [initial; MAX_SIZE as usize],
- };
- s
- }
- pub fn clear_possible(&mut self) {
- let mut initial = Bits(set_bits(10));
- initial.set(0, false);
- self.possible = [initial; MAX_SIZE as usize];
- }
- pub fn clear(&mut self) {
- let mut initial = Bits(set_bits(10));
- initial.set(0, false);
- self.board = [0; MAX_SIZE as usize];
- self.possible = [initial; MAX_SIZE as usize];
- }
-
-
- pub fn load_from_tld(&mut self, start_ch: char, blank: char, s: &str) {
- self.clear();
- let mut x: u8 = 0;
- let mut y: u8 = 0;
- for ch in s.chars() {
- if ch != blank {
- self.set(x, y, (ch as u8 - start_ch as u8) + 1);
- }
- y += 1;
- if y >= WIDTH {
- y = 0;
- x += 1;
- }
- }
- }
-
-
- pub fn load_from_tlr(&mut self, start_ch: char, blank: char, s: &str) {
- self.clear();
- let mut i: u8 = 0;
- for ch in s.chars() {
- if ch != blank {
- self.set(xy(i).0, xy(i).1, (ch as u8 - start_ch as u8) + 1);
- }
- i += 1;
- }
- }
- pub fn save_to_tld(&self, mut start_ch: char, blank: char) -> String {
- let mut result = String::new();
- result.reserve(MAX_SIZE as usize);
- start_ch = (start_ch as u8 - 1) as char;
- let mut x: u8 = 0;
- let mut y: u8 = 0;
- for _i in 0..MAX_SIZE {
- if self.board[pos(x, y) as usize] == 0 {
- result.push(blank);
- } else {
- result.push((start_ch as u8 + self.board[pos(x,y) as usize]) as char);
- }
- y += 1;
- if y >= WIDTH {
- y = 0;
- x += 1;
- }
- }
- result
- }
- pub fn save_to_tlr(&self, mut start_ch: char, blank: char) -> String {
- let mut result = String::new();
- result.reserve(MAX_SIZE as usize);
- start_ch = (start_ch as u8 - 1) as char;
- for i in 0..MAX_SIZE {
- if self.board[i as usize] == 0 {
- result.push(blank);
- } else {
- result.push((start_ch as u8 + self.board[i as usize]) as char);
- }
- }
- result
- }
- pub fn get(&self, x:u8, y:u8) -> u8 {
- self.board[pos(x,y) as usize]
- }
-
- pub fn set(&mut self, x: u8, y: u8, value: u8) {
- self.board[pos(x, y) as usize] = value;
-
- let mut g = for_row(y);
-
- for g in g.0 {
-
- self.possible[g as usize].set(value, false);
-
- }
- g = for_column(x);
-
- for g in g.0 {
-
- self.possible[g as usize].set(value, false);
-
- }
- g = for_cell(which_cell(x, y));
-
- for g in g.0 {
-
- self.possible[g as usize].set(value, false);
-
- }
- self.possible[pos(x, y) as usize].clear();
- }
-
-
-
- pub fn reset_possible(&mut self) {
-
- self.clear_possible();
- for y in 0..WIDTH {
- for x in 0..WIDTH {
- let item: u8 = self.board[pos(x as u8, y as u8) as usize];
- if item != 0 {
- let mut g: &Group = for_row(y);
- for g in g.0 {
- self.possible[g as usize].set(item, false);
- }
- g = for_column(x);
- for g in g.0 {
- self.possible[g as usize].set(item, false);
- }
- g = for_cell(which_cell(x,y));
- for g in g.0 {
- self.possible[g as usize].set(item, false);
- }
- self.possible[pos(x,y) as usize].clear();
- }
- }
- }
- }
-
- pub fn display(&self) {
- println!("╔═══╦═══╦═══╗");
- for y in 0..WIDTH {
- print!("║");
- for x in 0..WIDTH {
- let item = self.board[pos(x as u8, y as u8) as usize];
- if item == 0 {
- print!(" ");
- } else if (item >= 1) && (item <= 9) {
- print!("{}", item);
- }
- if x % 3 == 2 {
- print!("║");
- }
- }
- println!("");
- if y % 3 == 2 {
- if y + 1 == WIDTH {
- println!("╚═══╩═══╩═══╝");
- } else {
- println!("╠═══╬═══╬═══╣");
- }
- }
- }
- }
-
-
- pub fn display_possible(&self) {
- for y in 0..WIDTH {
- for x in 0..WIDTH {
-
- let mut possible = String::new();
- for p in self.possible[pos(x, y) as usize].iter() {
-
- possible += format!("{},", p).as_str();
- }
-
-
- print!("({},{}):", x, y);
-
- print!("{:9}", possible);
-
-
-
- }
- println!("");
- }
- }
-
-
-
-
- pub fn puzzle_complete(&self) -> bool {
- for i in 0..MAX_SIZE {
- if self.board[i as usize] == 0 {
- return false;
- }
- }
- true
- }
-
-
-
- fn calculate_possible(&mut self, total_solutions: &mut u16, solutions: &mut Vec<SudokuBoard>) -> bool {
- for idx in 0..MAX_SIZE {
- if self.board[idx as usize] == 0 {
-
- let (x, y) = xy(idx);
- if DEBUG_OUTPUT {
- println!("idx={} ({},{})", idx, x, y);
- self.display();
- }
- 'outer: for possible in 1..=9 {
- let mut g = for_row(y);
- for p in g.0 {
- if self.board[p as usize] == possible {
- continue 'outer;
- }
- }
- g = for_column(x);
- for p in g.0 {
- if self.board[p as usize] == possible {
- continue 'outer;
- }
- }
-
- let cell = which_cell(x, y);
- g = for_cell(cell);
- for p in g.0 {
- if self.board[p as usize] == possible {
- continue 'outer;
- }
- }
-
- if DEBUG_OUTPUT {
- println!("({},{})={}", x, y, possible);
- }
- self.board[idx as usize] = possible;
- if self.puzzle_complete() {
- if *total_solutions < solutions.capacity() as u16 {
- solutions.push(self.board);
- }
- *total_solutions += 1;
-
- break;
- } else {
- if self.calculate_possible(total_solutions, solutions) {
- return true;
- }
- }
- }
- self.board[idx as usize] = 0;
- return false;
- }
- }
- false
- }
-
-
-
- pub fn bruteforce_solver(&self) -> u16 {
- let mut workset = Sudoku {
- board: self.board,
- possible: [Bits(0); MAX_SIZE as usize],
- };
- let mut total_solutions: u16 = 0;
- let mut solutions = Vec::new();
- solutions.reserve(1);
- workset.calculate_possible(&mut total_solutions, &mut solutions);
-
- if solutions.len() > 0 {
- println!("*** A solution:");
- workset.board = solutions[0];
- workset.display();
- println!("***");
- }
- total_solutions
- }
-
-
- pub fn make(&mut self) {
- let mut rng = ChaCha20Rng::from_entropy();
- self.fill_board(&mut rng);
-
- }
-
-
-
- fn fill_board(&mut self, rng: &mut ChaCha20Rng) -> bool {
- let backup = Sudoku {
- board: self.board,
- possible: self.possible,
- };
- for idx in 0..MAX_SIZE {
- if self.board[idx as usize] == 0 {
- let (x, y) = xy(idx);
- let mut available: [u8; WIDTH as usize] = [0; WIDTH as usize];
- let mut total_available: u8 = 0;
- for t in self.possible[idx as usize].iter() {
- available[total_available as usize] = t;
- total_available += 1;
- }
- if total_available == 0 {
-
-
- return false;
- }
-
- available[0..total_available as usize].shuffle(rng);
- for v_idx in 0..total_available {
- let value = available[v_idx as usize];
- self.set(x, y, value);
- if self.fill_board(rng) {
- return true;
- }
-
- self.board = backup.board;
- self.possible = backup.possible;
- }
-
- return false;
- }
- }
-
- return true;
- }
-
-
-
-
- pub fn remove(&mut self) -> bool {
-
- let mut rng = ChaCha20Rng::from_entropy();
- let puzrange = Uniform::new(0, WIDTH);
- let mut x = 0;
- let mut y = 0;
- let mut value: u8 = 0;
- while value == 0 {
- x = puzrange.sample(&mut rng);
- y = puzrange.sample(&mut rng);
- value = self.get(x,y);
- }
- self.set(x,y,0);
-
- let mut puzcopy = self.clone();
- puzcopy.reset_possible();
-
-
- while puzcopy.solve(false) {};
-
- if puzcopy.puzzle_complete() {
- return true;
- }
-
-
- self.set(x,y,value);
- return false;
- }
-
-
-
-
- pub fn solve(&mut self, debug: bool) -> bool {
-
- let mut found_something = false;
- for i in 0..MAX_SIZE {
- if self.possible[i as usize].count_set() == 1 {
-
- let value = self.possible[i as usize].iter().next().unwrap();
-
- if debug {
- println!("Set1 {:?} to {}", xy(i), value);
- }
- self.set(xy(i).0, xy(i).1, value);
- found_something = true;
- }
- }
- let mut g = Group::new();
- let mut values = Bits(0);
- let mut group_process = |this: &mut Self, grp: &Group| {
-
- values.clear();
- for gidx in 0..WIDTH {
-
- for v in this.possible[grp.0[gidx as usize] as usize].iter() {
- values.set(v, true);
- }
-
-
- }
-
-
- for v in values.iter() {
- let mut count = 0;
- let mut pos = 0;
- for gidx in 0..WIDTH {
- if this.possible[grp.0[gidx as usize] as usize].get(v) {
-
- count += 1;
- pos = grp.0[gidx as usize];
- if count > 1 {
- break;
- }
- }
- }
- if count == 1 {
-
-
- if debug {
- println!("Set2 {:?} to {}", xy(pos), v);
- }
- this.set(xy(pos).0, xy(pos).1, v);
- found_something = true;
- }
- }
- };
-
- for i in 0..WIDTH {
- let mut g = for_column(i);
-
- group_process(self, &g);
- g = for_row(i);
-
- group_process(self, &g);
- g = for_cell(i);
-
- group_process(self, &g);
- }
- if found_something {
- return found_something;
- }
- if debug {
- println!("Looking for pairs...");
- }
-
- for i in 0..WIDTH {
- g.for_iter(i);
- for gidx in 0..WIDTH - 1 {
- let gpos = g.0[gidx as usize];
- if self.possible[gpos as usize].count_set() == 2 {
-
- for fidx in gidx + 1..WIDTH {
- let fpos = g.0[fidx as usize];
- if self.possible[fpos as usize].count_set() == 2 {
-
-
- if self.possible[gpos as usize] == self.possible[fpos as usize] {
-
-
-
- let mut values: [u8; 2] = [0, 0];
- let mut vpos = 0;
- for z in self.possible[gpos as usize].iter() {
- values[vpos] = z;
- vpos += 1;
- }
- let mut pair_removed = false;
-
- for remove in 0..WIDTH {
- if (gidx == remove) || (fidx == remove) {
- continue;
- }
-
- let rpos = g.0[remove as usize];
- if self.possible[rpos as usize].get(values[0]) {
- self.possible[rpos as usize].set(values[0], false);
- found_something = true;
- pair_removed = true;
- }
- if self.possible[rpos as usize].get(values[1]) {
- self.possible[rpos as usize].set(values[1], false);
- found_something = true;
- pair_removed = true;
- }
- }
-
- if xy(gpos).0 == xy(fpos).0 {
-
- let column = xy(gpos).0;
- vpos = 0;
- for z in self.possible[gpos as usize].iter() {
- values[vpos] = z;
- vpos += 1;
- }
- for remove in 0..WIDTH {
- if (remove == xy(gpos).1) || (remove == xy(fpos).1) {
- continue;
- }
- if self.possible[pos(column, remove) as usize]
- .get(values[0])
- {
- self.possible[pos(column, remove) as usize]
- .set(values[0], false);
- found_something = true;
- pair_removed = true;
- }
- if self.possible[pos(column, remove) as usize]
- .get(values[1])
- {
- self.possible[pos(column, remove) as usize]
- .set(values[1], false);
- found_something = true;
- pair_removed = true;
- }
- }
- }
- if xy(gpos).1 == xy(fpos).1 {
-
- let row = xy(gpos).1;
- vpos = 0;
- for z in self.possible[gpos as usize].iter() {
- values[vpos] = z;
- vpos += 1;
- }
- for remove in 0..WIDTH {
- if (remove == xy(gpos).0) || (remove == xy(fpos).0) {
- continue;
- }
- if self.possible[pos(remove, row) as usize].get(values[0]) {
- self.possible[pos(remove, row) as usize]
- .set(values[0], false);
- found_something = true;
- pair_removed = true;
- }
- if self.possible[pos(remove, row) as usize].get(values[1]) {
- self.possible[pos(remove, row) as usize]
- .set(values[1], false);
- found_something = true;
- pair_removed = true;
- }
- }
- }
- if pair_removed {
- if debug {
- println!(
- "Pair found! {} {}: {} {:?} and {} {:?} !",
- gidx,
- fidx,
- gpos,
- xy(gpos),
- fpos,
- xy(fpos)
- );
- }
- }
- }
- }
- }
- }
- }
- }
- found_something
- }
- }
|