dn1problemi.ml

let quick_brown_fox = "THEQUICKBRWNFXJMPSOVLAZYDG"
let rot13 = "NOPQRSTUVWXYZABCDEFGHIJKLM"

let indeks c = Char.code c - Char.code 'A'
let crka i = Char.chr (i + Char.code 'A') 


let sifriraj kljuc =
  let sifriraj_znak znak =
    if 'A' <= znak && znak <= 'Z' then
      String.get kljuc (indeks znak)
    else znak
  in
  String.map sifriraj_znak   



  let inverz kljuc =
    let abeceda = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" in
    let crke = String.sub abeceda 0 (String.length kljuc) in
    let invertiraj_znak znak = 
      let i_v_kljucu = String.index kljuc znak in
      String.get abeceda i_v_kljucu
    in
    String.map invertiraj_znak crke


let dodaj_zamenjavo niz (mesto, menjava) =
  let i_mesta = indeks mesto in
  let menjati i znak =
    match i, znak with
    | i, '_' when i = i_mesta -> menjava
    | _, znak -> znak 
  in
  match String.get niz i_mesta with
  | '_' -> 
    (
    match String.index_opt niz menjava with
    | Some i -> None
    | None -> Some (String.mapi menjati niz)
    )
  | a when a = menjava -> Some niz
  | _ -> None


let dodaj_zamenjave niz (mesta, menjave) =
  let indeksi = String.length mesta in 
  let niz_opt = Some niz in
  let rec aux i str_opt =
    match i, str_opt with
    | i, Some s when i < indeksi -> aux (i + 1) (dodaj_zamenjavo s (String.get mesta i, String.get menjave i))
    | i, _ when i = indeksi -> str_opt (* pršli smo do konca *)
    | _, _ -> None (* i ni vecji od 1 in uno je pol karkol, tud ce je uno none*)
  in
  aux 0 niz_opt


(* Zgoraj sem imela med vajami probleme z ipynb, ker ni zaznaval ocaml-a.
   Spodaj za namen izboljšanja dešifriravanja: *)

let besede = "the of to and a in is it you that he was for on are with as i his they be at one have this from or had by word but what some we can out other were all there when up use your how said an each she which do their time if will way about many then them write would like so these her long make thing see him two has look more day could go come did number sound no most people my over know water than call first who may down side been now find any new work part take get place made live where after back little only round man year came show every good me give our under name very through just form sentence great think say help low line differ turn cause much mean before move right boy old too same tell does set three want air well also play small end put home read hand port large spell add even land here must big high such follow act why ask men change went light kind off need house picture try us again animal point mother world near build self earth father head stand own page should country found answer school grow study still learn plant cover food sun four between state keep eye never last let thought city tree cross farm hard start might story saw far sea draw left late run don't while press close night real life few north open seem together next white children begin got walk example ease paper group always music those both mark often letter until mile river car feet care second book carry took science eat room friend began idea fish mountain stop once base hear horse cut sure watch color face wood main enough plain girl usual young ready above ever red list though feel talk bird soon body dog family direct pose leave song measure door product black short numeral class wind question happen complete ship area half rock order fire south problem piece told knew pass since top whole king space heard best hour better true . during hundred five remember step early hold west ground interest reach fast verb sing listen six table travel less morning ten simple several vowel toward war lay against pattern slow center love person money serve appear road map rain rule govern pull cold notice voice unit power town fine certain fly fall lead cry dark machine note wait plan figure star box noun field rest correct able pound done beauty drive stood contain front teach week final gave green oh quick develop ocean warm free minute strong special mind behind clear tail produce fact street inch multiply nothing course stay wheel full force blue object decide surface deep moon island foot system busy test record boat common gold possible plane stead dry wonder laugh thousand ago ran check game shape equate hot miss brought heat snow tire bring yes distant fill east paint language among grand ball yet wave drop heart am present heavy dance engine position arm wide sail material size vary settle speak weight general ice matter circle pair include divide syllable felt perhaps pick sudden count square reason length represent art subject region energy hunt probable bed brother egg ride cell believe fraction forest sit race window store summer train sleep prove lone leg exercise wall catch mount wish sky board joy winter sat written wild instrument kept glass grass cow job edge sign visit past soft fun bright gas weather month million bear finish happy hope flower clothe strange gone jump baby eight village meet root buy raise solve metal whether push seven paragraph third shall held hair describe cook floor either result burn hill safe cat century consider type law bit coast copy phrase silent tall sand soil roll temperature finger industry value fight lie beat excite natural view sense ear else quite broke case middle kill son lake moment scale loud spring observe child straight consonant nation dictionary milk speed method organ pay age section dress cloud surprise quiet stone tiny climb cool design poor lot experiment bottom key iron single stick flat twenty skin smile crease hole trade melody trip office receive row mouth exact symbol die least trouble shout except wrote seed tone join suggest clean break lady yard rise bad blow oil blood touch grew cent mix team wire cost lost brown wear garden equal sent choose fell fit flow fair bank collect save control decimal gentle woman captain practice separate difficult doctor please protect noon whose locate ring character insect caught period indicate radio spoke atom human history effect electric expect crop modern element hit student corner party supply bone rail imagine provide agree thus capital won't chair danger fruit rich thick soldier process operate guess necessary sharp wing create neighbor wash bat rather crowd corn compare poem string bell depend meat rub tube famous dollar stream fear sight thin triangle planet hurry chief colony clock mine tie enter major fresh search send yellow gun allow print dead spot desert suit current lift rose continue block chart hat sell success company subtract event particular deal swim term opposite wife shoe shoulder spread arrange camp invent cotton born determine quart nine truck noise level chance gather shop stretch throw shine property column molecule select wrong gray repeat require broad prepare salt nose plural anger claim continent oxygen sugar death pretty skill women season solution magnet silver thank branch match suffix especially fig afraid huge sister steel discuss forward similar guide experience score apple bought led pitch coat mass card band rope slip win dream evening condition feed tool total basic smell valley nor double seat arrive master track parent shore division sheet substance favor connect post spend chord fat glad original share station dad bread charge proper bar offer segment slave duck instant market degree populate chick dear enemy reply drink occur support speech nature range steam motion path liquid log meant quotient teeth shell neck"
    
let slovar =
  besede 
  |> String.split_on_char ' '
  |> List.map (String.uppercase_ascii)    

let mozne_razsiritve kljuc sifrirana slovar =
  let rec aux acc =
    function
    | [] -> acc
    | x :: xs -> 
      (
        match dodaj_zamenjave kljuc (sifrirana, x) with
        | Some k -> aux (k :: acc) xs
        | None -> aux acc xs
      )
  in
  aux [] slovar 

(* Funkcija vrača string list option, ker se je izkazalo, da je možnih rešitev lahko več, 
   funkcija pa se ne zna odločati, katero sporočilo je vsebinsko najustreznejše.*)
let odsifriraj niz = 
  let besede = String.split_on_char ' ' niz in
  if besede = [] then None 
  else
  let prvi_kandidati = mozne_razsiritve (String.make 26 '_') (List.nth besede 0) slovar in
  let razsirjanje_kandidata_z_novimi beseda seznam_sifer star_kandidat =
    let rec pomozna1 se_primerni =
      function
      | [] -> se_primerni
      | x :: xs -> (
        match dodaj_zamenjave star_kandidat (beseda, x) with
        | Some s -> pomozna1 (s :: se_primerni) xs
        | None -> pomozna1 se_primerni xs
      )
    in
    pomozna1 [] seznam_sifer
  in
  let rec glavna (kandidati : string list) seznam_besed =
    match seznam_besed with
    | [] -> Some kandidati
    | beseda :: b's ->
      (
        let seznamek =
        (mozne_razsiritve (String.make 26 '_') beseda slovar) 
        |> List.map (fun kljuc -> sifriraj kljuc beseda)
        in
        let ostali =
          List.map (razsirjanje_kandidata_z_novimi beseda seznamek) kandidati
        in
        match ostali with 
        | [] -> None (* Že ni več možnih kandidatov za ključe, funkcija lahko konča. *)
        | _::_ -> glavna (List.flatten ostali) b's
      )
  in
  let iskani_kljuci = glavna prvi_kandidati (List.tl besede) in
  let sifriraj' niz kljuc = (* Nova funkcija zaradi delne uporabe na ključih in dodanega znaka '_', ki ga prva ni predpostavljala.*)
    let sifriraj_znak znak =
      if 'A' <= znak && znak <= 'Z' then
        String.get kljuc (indeks znak)
      else znak
    in
    String.map sifriraj_znak niz   
  in
  match iskani_kljuci with
  | None -> None
  | Some x -> Some (List.map (sifriraj' niz) x)