author | nemo |
Mon, 13 Jan 2014 14:40:06 -0500 | |
changeset 9983 | 1b62d993cd2f |
parent 9464 | 901e363d5837 |
permissions | -rw-r--r-- |
9462 | 1 |
module Test where |
2 |
||
3 |
import Control.Monad |
|
4 |
import Data.Word |
|
5 |
import qualified Data.IntSet as IS |
|
6 |
||
7 |
data OP = Sum |
|
8 |
| Mul |
|
9 |
| Sub |
|
10 |
deriving Show |
|
11 |
||
12 |
||
13 |
genOps :: Int -> [[OP]] |
|
14 |
genOps 1 = [[Sum], [Mul], [Sub]] |
|
15 |
genOps n = [a : as | a <- [Sum, Mul, Sub], as <- genOps (n - 1)] |
|
16 |
||
17 |
||
18 |
genPos :: Int -> Int -> [[Int]] |
|
19 |
genPos m 1 = map (:[]) [-m..m - 1] |
|
20 |
genPos m n = [a : as | a <- [-m..m - 1], as <- genPos m (n - 1)] |
|
21 |
||
22 |
||
23 |
hash :: [Int] -> [OP] -> [Int] -> Int |
|
24 |
hash poss op s = foldl applyOp s' (zip ss op) |
|
25 |
where |
|
26 |
applyOp v (n, Sum) = (v + n) `mod` 256 |
|
27 |
applyOp v (n, Mul) = (v * n) `mod` 256 |
|
28 |
applyOp v (n, Sub) = (v - n) `mod` 256 |
|
29 |
(s' : ss) = map (\p -> if p >= 0 then s !! p else s !! (l + p)) poss |
|
30 |
l = length s |
|
31 |
||
32 |
||
33 |
test = do |
|
34 |
a <- liftM lines getContents |
|
35 |
let w = minimum $ map length a |
|
36 |
let opsNum = 4 |
|
37 |
let opsList = genOps (opsNum - 1) |
|
38 |
let posList = genPos w opsNum |
|
39 |
let target = length a |
|
40 |
let wordsList = map (map fromEnum) a |
|
41 |
let hashedSize = IS.size . IS.fromList |
|
42 |
print $ length a |
|
43 |
putStrLn . unlines . map show $ filter (\l -> fst l == length a) $ [(hs, (p, o)) | p <- posList, o <- opsList, let hs = hashedSize . map (hash p o) $ wordsList] |
|
44 |
||
45 |
didIunderstand' = do |
|
46 |
a <- liftM lines getContents |
|
9464
901e363d5837
Finish rework of default binds system. Default binds now work even before first turn.
unc0rr
parents:
9462
diff
changeset
|
47 |
print $ length a |
9462 | 48 |
print . IS.size . IS.fromList . map (testHash . map fromEnum) $ a |
49 |
where |
|
9464
901e363d5837
Finish rework of default binds system. Default binds now work even before first turn.
unc0rr
parents:
9462
diff
changeset
|
50 |
testHash s = let l = length s in ( |
9462 | 51 |
(s !! (l - 2) * s !! 1) + s !! (l - 1) - s !! 0 |
9464
901e363d5837
Finish rework of default binds system. Default binds now work even before first turn.
unc0rr
parents:
9462
diff
changeset
|
52 |
) `mod` 256 |