-
Notifications
You must be signed in to change notification settings - Fork 35
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
"Player 0xc30edf0147c46d0a5f79bfe9b15ce9de9b8879be submitted 'test_dy…
…namic' for challenge knapsack"
- Loading branch information
0xc30edf0147c46d0a5f79bfe9b15ce9de9b8879be
committed
Apr 25, 2024
1 parent
5270068
commit 79270e9
Showing
2 changed files
with
59 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,58 @@ | ||
use tig_challenges::knapsack::*; | ||
|
||
pub fn solve_challenge(challenge: &Challenge) -> anyhow::Result<Option<Solution>> { | ||
let max_weight = challenge.max_weight; | ||
let min_value = challenge.min_value; | ||
let num_items = challenge.difficulty.num_items; | ||
|
||
// Sort items by value-to-weight ratio in descending order | ||
let mut sorted_items: Vec<usize> = (0..num_items).collect(); | ||
sorted_items.sort_by(|&a, &b| { | ||
let ratio_a = challenge.values[a] as f64 / challenge.weights[a] as f64; | ||
let ratio_b = challenge.values[b] as f64 / challenge.weights[b] as f64; | ||
ratio_b.partial_cmp(&ratio_a).unwrap() | ||
}); | ||
|
||
// Initialize combinations with a single empty combo | ||
let mut combinations: Vec<(Vec<bool>, u32, u32)> = vec![(vec![false; num_items], 0, 0)]; | ||
|
||
let mut items = Vec::new(); | ||
for &item in &sorted_items { | ||
// Create new combos with the current item | ||
let mut new_combinations: Vec<(Vec<bool>, u32, u32)> = combinations | ||
.iter() | ||
.map(|(combo, value, weight)| { | ||
let mut new_combo = combo.clone(); | ||
new_combo[item] = true; | ||
( | ||
new_combo, | ||
value + challenge.values[item], | ||
weight + challenge.weights[item], | ||
) | ||
}) | ||
.filter(|&(_, _, weight)| weight <= max_weight) // Keep only combos within weight limit | ||
.collect(); | ||
|
||
// Check if any new combination meets the minimum value requirement | ||
if let Some((combo, _, _)) = new_combinations | ||
.iter() | ||
.find(|&&(_, value, _)| value >= min_value) | ||
{ | ||
items = combo | ||
.iter() | ||
.enumerate() | ||
.filter_map(|(i, &included)| if included { Some(i) } else { None }) | ||
.collect(); | ||
break; | ||
} | ||
|
||
// Merge new_combinations with existing combinations | ||
combinations.append(&mut new_combinations); | ||
|
||
// Deduplicate combinations by keeping the highest value for each weight | ||
combinations.sort_by(|a, b| a.2.cmp(&b.2).then_with(|| b.1.cmp(&a.1))); // Sort by weight, then by value | ||
combinations.dedup_by(|a, b| a.2 == b.2 && a.1 <= b.1); // Deduplicate by weight, keeping highest value | ||
} | ||
|
||
Ok(Some(Solution { items })) | ||
} |