Improve performance for closest string 2

closest_string_assignment_2/solution0.out

@@ -1 +1 @@
-csafydvimgvnrhraodeieklynoxgajbjhhsvntdfkvuptfuonsdkyxqculmterhvbwrdytoeggorionubonlkcafauvrfyfinexf
+csafgdvimgenrhramdeieklynoxgajbjhhsvntdfkvupnjuonsdkylqculmterhvbwrsytosggoxionubonlqcafaujrfyfinexf

closest_string_assignment_2/solution1.out

@@ -1 +1 @@
-csaaydvimgenfhraodeixklynorgajxjhhsvgtdfkvupnjuonsdfylqculmterhqbwsdytoeggorienubombkcafauvrsyfiunxfydutnfhomxvqneunkzpdmurffyqtjyncxkzzklifrpopskobxshsrirnmarylsxffdunbparpzhswqdcbxsjrqwmnnmtwoedfbunwvwikywvpgygashfpvxxldndrbibdoobjceojyhnfmaxxbheyqdddahbdluisejzdturigzhsjwpntxcriqsbbvuqtphjmgmuqqksrclcrmbrlihzpcayatgyhiddwjccnzyibwpigwajqgotdyzjxrjmpqlagfpikakqtiplyasxnglbwieemfvqhjitddujbimxcwykyokxbyfhakpdywutxfrjnyztzeefpazptwbsysfndbzpedbrxfkhrahgxmztnhqcmojshiajwbbegfmpehuawrczvhqrwohigzalcbvyvjkshtgmpmberurbjxzgzysnvqxkknrdryhfnvxlbzntrzmypofoefjowrrrlltgwxvdanbfbyijjzeoxcqbvjlcbdabzwkodtimuapgkgbahmwyglchntycbjttbviiihzfuvjuoohcjpuriotvylnufyziqgyaihlxujdtcyjgrrnmrqhooljsyycrkqexwimkruninplbapzdlpuwtgdvmqcjadhibklmnvcaaeyncbttubnbziblidaspkcvbrzzzpmpvlqixrhzwjfumoeaxrcdpypvcjivzegljbcurkyqdijhubnxvidzhfdzxcbppxuxbjvzcicpordjzcljnijvzjiedngrlrcbgfsjqlztqexmafjhloscgsdkxsejvwnvfzxnfldsurnkstjopgshygxrolcwoxkqlbhrfjmzkzttwhsvbmtrbcdriplrxfncwljctgvmlwhghqpnxyawqgkpzwrjeoxqfiofrmv
+cusfydvimgenrhraodeixklynoxgajxjhhsvntdfkvupkjuonsdfylqculmterhvlwrdytoeggorionubombkcafauvrsyfiuexfyduwnfhomxvqneunkzpdmurffyrtjyncxkzzklifrpooskobfshsrirnmarylsxfxdunbparpzhmwqdcbxsjrowgnnmtwleddbuhwqwikrwvpgygashfpvexzdndsbibdoobjcexjyhvfnavxahjyqdseqhbdluknejzdwurirzvsjwpntxerjqsbbvustphjmgmutkksrclcldbrlihzpcakktgyzidwwjcqnmyigwpigyqjqjotfyzjxrjgpqhagfpikbkqtiplyfsxnglbwieemfvqhjitddvmbimxywykyokxbyrhakpdywuzxbrfnyztzwefpazptybsysfndbzpederxvxhrahgxmztbhocmolsvvajlbbegempehuiwrozvhqrqogigzalcbvyjqkshtgmpmkerubbjxpazysnvcxkknrdcyhfnvxlbznmrqmypoqoefjowrrrlcxmfxvdanbfbyifaztoxcybvjlprkapzwkomtimuapqxgbahvwygichntycbjtdbvqiigzfuvjuoohckyuriztvylnufyziqgyaihlxfadtcojgqrnmrqhoolrsyycrkqlxwinkrunpnblbanzdlpuwtgdvmfcxadhvbklmnaoausyzcettudnbzibwidaspkivbrzzzpmpqtqixrbzwjfumoexxrcdpypvcdivzegljbcurkysdbjhubncvidzhfdzxcbppeuxbjvzcickordjzctiniuvzaiedngrlzcbgfskqlztyexjafjclbmcgstkrsejvwnvfzxnfldsurnkskjopgiiyfxrxlcwdhkjlbhsxjmtkzttwhsvbmkrdcvriprrxfncllscztvmzwhehqpnhyakqgkpzwrjloxqfiofrmy

closest_string_assignment_2/solution2.out

@@ -1 +1 @@
-mjyhvvxwfnmfqkiehyqoujfpezjtfnusibfwvqqdtywxfgbyrsaevvkteupsjebatbqdjemjoneygrhqswzzijeqdourkpbdhjhkzjvezughojeyktyssyudchhlitpjjvzkqpqlmvwfjspopxgudwwcumsrbfxmoevpplligxwlehgoqbszxrgpjohrurlfxjzuvxmasyzlzdorckgfqgoeanrpavqexopwslgcflgmepmkhzouiajxbcmagptxaytgieqruivqfsbpruugatoexyegbvisemmgjipgaygrcgmligjoksbhamlbvxbjclfwdqkcoeschpeqexnwohcxybiuukoyhewemsjkqcgfcsdjlvzkknbbvlkfxentestcrdresjxuijwlwilsaqynzwseieixrbknoyloivyvccxwplomopgxlbankjsxuqadsxtltkcndielbomecqrffdegahjoassbfakiwmkvzwdwwyaudzrbpsozeawyalmunxnbuyddwztgbknhoseojxhhkmcuwmmgteqbyifksaujqrclzxpmhcjcrskjievbtjjvzpyqoxsaavznbtrnonrgmyqlbpuaehodxatwpyccclemnshpmkcyfysttfzfqgyiaujawjxoioujxthzyaqwewwafbsbjecadkmjqlrrftvpiedngujblwxlfnmdvtqsjnsjhcatruetbywndpccorxulqbnzbhqfknzyxnmmfqueqckflqaputwcthcliwpgygynssylwvqnebvrovsjewsixsyhlpbbxdeomdzljmrbkntreemdwolbpfysselzqbywdhaquimrdoszxxkieybpkoxgyagsgaesmnuimriphupxhetzkthdgcavipapofgyezwbmzucokaeeqqisuzwtskosxmixgscwqnhemwzlgvecabqzcvfvrjmudrlaapxrjandtbniysjxghgheytscdtsqz
+mjyhvvxwfnmfqkuehyqoujfkezptfnusibfwvqqdtywxfgbyrsaevvkthupsjebahbqdjbmjoneygrhqswzzijeqdourkpbdhjhkzjvezughojeyktyssyudchhlitpjjvzkqqqlmvwfjspopxgudwwcumsrbfxmoevpplligxwlehgoqbszxrgpjohrurlfxjzuvxmasyzlzdfrckgfqgoeanrpabqexopwklgcflgmepmkhzouiajwbcmaggtxaytgilqruivifsbpruugatzexyegbvwsemmgjipgaygrcgmlihjokibyamlbvibkclfwdqgcoeschpeqexnwohclybiuukoyhewemsjkqcgfcsdjlvcksnbbvlkfxfntestcrdresjxuijwlwiluaqynzwseieixrbknoyloivyvccowplomopgxlbankjsxuqaddxtltkcnkiekbomecqrffdegahjoassbfapiwmkvzwdwwyagdzxbprozeewealmunxnbuvddlztgbknhoseojxhhkocowbogteqbyifcssujqrclzxpmhcjcbskjievbtojvppyqkxsaavzdbtrnonrgmyqlbpuaehodxatwpyccclemnshiukcyfysttvzfqgyiaujawnxsioujxthzyaqwewwafbsbjecadkmjqlrrrtvpiedigujbloxlfnmdvtqsjnsjhcajruetbywnxpbcorxulqbnkbhqfknzyxnmmfquecckflqavutwcthcligpgygynssylwvqnebvfovsdewsixsyhltbbxdeomdzljmrgkntreemdwolbpfyoselzqbywdhaqjimrdoszxxkieybpkoxgyagsgaespnuimriphupxhetzkthdecavipapofgyezwbmducokaeeqqisuzwtskosxmixgscwqnhemwzlgvecabqzcvfvrjmudrlaapxrjandtbniysjxghgheytscdtsqz

closest_string_assignment_2/src/main.rs

@@ -98,16 +98,38 @@ fn solve(input: &[Vec<u8>]) -> String {
 
     let mut result = String::new();
 
+    let mut hash = HashMap::new();
+
+    let mut previous = HashMap::new();
+
     for (i, col) in columns.iter().enumerate() {
+        if !hash.contains_key(col) {
             let t = model
                 .get_obj_attr_batch(attr::X, variables[col].clone())
-            .unwrap();
+                .unwrap()
+                .into_iter()
+                .map(|v| v as usize)
+                .collect_vec();
 
-        let mut p = columns[0..i].iter().filter(|&c| c == col).count();
+            hash.insert(col, t);
+        }
+
+        let t = hash.get(col).unwrap();
+
+        let mut p = match previous.entry(col) {
+            std::collections::hash_map::Entry::Occupied(mut occupied_entry) => {
+                *occupied_entry.get_mut() += 1;
+                *occupied_entry.get()
+            }
+            std::collections::hash_map::Entry::Vacant(vacant_entry) => {
+                vacant_entry.insert(0);
+                0
+            }
+        };
 
         let mut j = 0;
-        while t[j] as usize <= p {
-            p -= t[j] as usize;
+        while t[j] <= p {
+            p -= t[j];
             j += 1;
         }