Arpakuutiot

warning: unused variable: `f`
   --> input/code.rs:102:33
    |
102 | fn arrow_left(o: Orientation3D, f: Orientation3D) -> Orientation3D {
    |                                 ^ help: consider prefixing with an underscore: `_f`
    |
    = note: `#[warn(unused_variables)]` on by default

warning: unused variable: `f`
   --> input/code.rs:106:34
    |
106 | fn arrow_right(o: Orientation3D, f: Orientation3D) -> Orientation3D {
    |                                  ^ help: consider prefixing with an underscore: `_f`

use std::io;
use std::io::prelude::*;

// Arpakuutioiden luoktus:
// 1. Mikä sivu 1:stä vastapäätä
// 2. Mikä seuraavista permutaatioista
// 1234
// 1324
// 2134
// 2314
// 3124
// 3214

#[derive(Debug, Clone, Copy)]
enum Orientation3D {
	PX,
	NX,
	PY,
	NY,
	PZ,
	NZ,
}

// On flat surface always points left

#[derive(Debug, Clone, Copy)]
enum Orientation2D {
	Left,
	Right,
	Top,
	Bottom,
}

// Cube faces:
// +X = 0
// -X = 1
// +Y = 2
// -Y = 3
// +Z = 4
// -Z = 5


fn rotate(face: Orientation3D, rot: Orientation3D) -> Orientation3D {
	use Orientation3D::*;
	match (rot, face) {
		(PX, PX) => PX,
		(PX, NX) => NX,
		(PX, PY) => PZ,
		(PX, NY) => NZ,
		(PX, PZ) => NY,
		(PX, NZ) => PY,
		
		(NX, PX) => PX,
		(NX, NX) => NX,
		(NX, PY) => NZ,
		(NX, NY) => PZ,
		(NX, PZ) => PY,
		(NX, NZ) => NY,
		
		(PY, PX) => NZ,
		(PY, NX) => PZ,
		(PY, PY) => PY,
		(PY, NY) => NY,
		(PY, PZ) => PX,
		(PY, NZ) => NX,
		
		(NY, PX) => PZ,
		(NY, NX) => NZ,
		(NY, PY) => PY,
		(NY, NY) => NY,
		(NY, PZ) => NX,
		(NY, NZ) => PX,
		
		(PZ, PX) => PY,
		(PZ, NX) => NY,
		(PZ, PY) => NX,
		(PZ, NY) => PX,
		(PZ, PZ) => PZ,
		(PZ, NZ) => NZ,
		
		(NZ, PX) => NY,
		(NZ, NX) => PY,
		(NZ, PY) => PX,
		(NZ, NY) => NX,
		(NZ, PZ) => PZ,
		(NZ, NZ) => NZ,
	}
}

fn negate(o: Orientation3D) -> Orientation3D {
	use Orientation3D::*;
	match o {
		PX => NX,
		NX => PX,
		PY => NY,
		NY => PY,
		PZ => NZ,
		NZ => PZ,
	}
}

fn arrow_left(o: Orientation3D, f: Orientation3D) -> Orientation3D {
	o
}

fn arrow_right(o: Orientation3D, f: Orientation3D) -> Orientation3D {
	negate(o)
}

fn arrow_top(o: Orientation3D, f: Orientation3D) -> Orientation3D {
	negate(arrow_bottom(o, f))
}

fn arrow_bottom(o: Orientation3D, f: Orientation3D) -> Orientation3D {
	rotate(o, f)
}

// Return first orientation then position
fn apply_orientation(o: Orientation3D, f: Orientation3D, d: Orientation2D) -> (Orientation3D, Orientation3D) {
	use Orientation2D::*;
	match d {
		Left => (rotate(o, arrow_bottom(o, f)), rotate(f, arrow_bottom(o, f))),
		Right => (rotate(o, arrow_top(o, f)), rotate(f, arrow_top(o, f))),
		Top => (o, rotate(f, arrow_left(o, f))),
		Bottom => (o, rotate(f, arrow_right(o, f))),
	}
}

fn cp(f: Orientation3D) -> usize {
	use Orientation3D::*;
	match f {
		PX => 0,
		NX => 1,
		PY => 2,
		NY => 3,
		PZ => 4,
		NZ => 5,
	}
}

fn pc(u: usize) -> Orientation3D {
	use Orientation3D::*;
	match u {
		0 => PX,
		1 => NX,
		2 => PY,
		3 => NY,
		4 => PZ,
		5 => NZ,
		_ => panic!(),
	}
}

fn read_and_classify(data: &mut [Option<u32>]) -> u32 {
	let mut cube = [u32::MAX; 6];
	for y in 0..5 {
		for x in 0..5 {
			if data[y*5 + x].is_some() {
				read_cube(&mut cube, data, x, y, Orientation3D::PY, Orientation3D::PX);
				break;
			}
		}
	}
	classify_cube(&mut cube)
}

fn read_cube(cube: &mut [u32], data: &mut [Option<u32>], x: usize, y: usize, o: Orientation3D, f: Orientation3D) {
	//println!("Called read_cube with x: {}, y: {}, o: {:?}, f: {:?}", x, y, o, f);
	use Orientation2D::*;
	if cube[cp(f)] != u32::MAX {
		//println!("Returning read_cube (early)");
		return
	}
	if let Some(n) = data[y*5 + x] {
		//println!("read_cube: is_some");
		cube[cp(f)] = n;
		if y > 0 {
			let (no, nf) = apply_orientation(o, f, Top);
			read_cube(cube, data, x, y - 1, no, nf);
		}
		if y < 5 {
			let (no, nf) = apply_orientation(o, f, Bottom);
			read_cube(cube, data, x, y + 1, no, nf);
		}
		if x > 0 {
			let (no, nf) = apply_orientation(o, f, Left);
			read_cube(cube, data, x - 1, y, no, nf);
		}
		if x < 5 {
			let (no, nf) = apply_orientation(o, f, Right);
			read_cube(cube, data, x + 1, y, no, nf);
		}
	}
	//println!("Returning read_cube (end)");
}

fn classify_cube(cube: &[u32]) -> u32 {
	//println!("Called classify_cube with cube: {:?}", cube);
	use Orientation3D::*;
	let mut j = 6;
	for i in 0..6 {
		if cube[i] == 1 {
			j = i;
			break
		}
	}
	let one = pc(j);
	let oppi = cp(negate(one));
	let opp = cube[oppi];
	let mut round = 
	match one {
		PX | NX => {
			[cube[cp(PY)], cube[cp(PZ)], cube[cp(NY)], cube[cp(NZ)]]
		}
		PY | NY => {
			[cube[cp(PZ)], cube[cp(PX)], cube[cp(NZ)], cube[cp(NX)]]
		}
		PZ | NZ => {
			[cube[cp(PX)], cube[cp(PY)], cube[cp(NX)], cube[cp(NY)]]
		}
	};
	match one {
		NX | NY | NZ => {
			round.reverse()
		}
		_ => {}
	};
	(opp - 1)*6 + classify_cyclic_permutation(&round)
}

// 1234
// 1324
// 2134
// 2314
// 3124
// 3214

fn classify_cyclic_permutation(round: &[u32]) -> u32 {
	let mi = round.iter().enumerate().max_by_key(|(_, vx)| *vx).unwrap().0;
	let v : Vec<_> = round.iter().cycle().skip(mi + 1).take(3).collect();
	match (v[0] < v[1], v[1] < v[2]) {
		(true, true) => 0,
		(true, false) => if v[0] < v[2] {1} else {3},
		(false, true) => if v[0] < v[2] {2} else {4}
		(false, false) => 5
	}
}

fn main() {
	let stdin = io::stdin();
    let mut lines = stdin.lock().lines();
	let num = lines.next().unwrap().unwrap().parse::<usize>().unwrap();
	let mut cubes = Vec::new();
	for k in 0..num {
		if k != 0 {
			lines.next();
		}
		let mut dat = Vec::new();
		for _ in 0..5 {
			for c in lines.next().unwrap().unwrap().chars() {
				dat.push(c.to_string().parse::<u32>().ok());
			}
		}
		//println!("Dat: {:?}", dat);
		cubes.push(read_and_classify(&mut dat));
	}
	//println!("Cubes: {:?}", cubes);
	//println!();
	//println!();
	for i in 0..cubes.len() {
		let mut some = false;
		for j in 0..cubes.len() {
			if i == j {
				continue
			}
			if cubes[i] == cubes[j] {
				print!("{} ", j + 1);
				some = true;
			}
		}
		if !some {
			print!("-");
		}
		println!();
	}
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: ACCEPTED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 1, 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 1, 2

Verdict: ACCEPTED

Test 10

Group: 1, 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED

Test 14

Group: 2

Verdict: ACCEPTED

Test 15

Group: 2

Verdict: ACCEPTED

Test 16

Group: 2

Verdict: ACCEPTED

Test 17

Group: 2

Verdict: ACCEPTED

Test 18

Group: 2

Verdict: ACCEPTED

Test 19

Group: 2

Verdict: ACCEPTED

Test 20

Group: 2

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: ACCEPTED

Test 22

Group: 2

Verdict: ACCEPTED

Test 23

Group: 2

Verdict: ACCEPTED

Test 24

Group: 2

Verdict: ACCEPTED

Test 25

Group: 2

Verdict: ACCEPTED

Test 26

Group: 2

Verdict: ACCEPTED

Test 27

Group: 2

Verdict: ACCEPTED

Test 28

Group: 2

Verdict: ACCEPTED

Test 29

Group: 2

Verdict: ACCEPTED

Test 30

Group: 2

Verdict: ACCEPTED

Test 31

Group: 1, 2

Verdict: ACCEPTED