aboutsummaryrefslogtreecommitdiff
path: root/challenge-239/barroff/nim/ch_2.nim
blob: 34d06af62024e36e588447f89bf2d6cc36c2a2e4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import std/[sequtils, setutils, sugar, unittest]

# run tests with following command:
# nim c -r ch_2.nim

proc consistent_strings(allowed: string, arr1: openArray[string]): int =
  let
    allowed_chars = toSet(allowed)
  result = countIt(mapIt(arr1, toSet(it)), it <= allowed_chars)

suite "consistent strings":
  test "\"ab\", [\"ad\", \"bd\", \"aaab\", \"baa\", \"badab\"]":
    let
      allowed = "ab"
      arr1 = ["ad", "bd", "aaab", "baa", "badab"]

    check(consistent_strings(allowed, arr1) == 2)

  test "\"abc\", [\"a\", \"b\", \"c\", \"ab\", \"ac\", \"bc\", \"abc\"]":
    let
      allowed = "abc"
      arr1 = ["a", "b", "c", "ab", "ac", "bc", "abc"]

    check(consistent_strings(allowed, arr1) == 7)

  test "\"cad\", [\"cc\", \"acd\", \"b\", \"ba\", \"bac\", \"bad\", \"ac\", \"d\"]":
    let
      allowed = "cad"
      arr1 = ["cc", "acd", "b", "ba", "bac", "bad", "ac", "d"]

    check(consistent_strings(allowed, arr1) == 4)