You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.3 KiB
35 lines
1.3 KiB
/*
|
|
* SD2x Homework #5
|
|
* Implement the method below according to the specification in the assignment description.
|
|
* Please be sure not to change the method signature!
|
|
*/
|
|
|
|
import java.util.List;
|
|
import java.util.PriorityQueue;
|
|
import java.util.TreeMap;
|
|
|
|
public class MovieRatingsParser {
|
|
|
|
public static TreeMap<String, PriorityQueue<Integer>> parseMovieRatings(List<UserMovieRating> allUsersRatings) {
|
|
TreeMap<String, PriorityQueue<Integer>> movieRatings = new TreeMap<String, PriorityQueue<Integer>>();
|
|
if(allUsersRatings == null) return movieRatings;
|
|
|
|
for(UserMovieRating m : allUsersRatings){
|
|
|
|
if(m == null) continue;
|
|
if(m.getMovie() == null) continue;
|
|
if(m.getMovie().isEmpty()) continue;
|
|
if(m.getUserRating() < 0) continue;
|
|
if(!movieRatings.containsKey(m.getMovie().toLowerCase())){
|
|
PriorityQueue<Integer> ratingHeap = new PriorityQueue<Integer>();
|
|
ratingHeap.add(m.getUserRating());
|
|
movieRatings.put(m.getMovie().toLowerCase(), ratingHeap);
|
|
continue;
|
|
}
|
|
movieRatings.get(m.getMovie().toLowerCase()).add(m.getUserRating());
|
|
|
|
}
|
|
return movieRatings;
|
|
}
|
|
|
|
}
|