1366. Rank Teams by Votes

In a special ranking system, each voter gives a rank from highest to lowest to all teams participated in the competition.

The ordering of teams is decided by who received the most position-one votes. If two or more teams tie in the first position, we consider the second position to resolve the conflict, if they tie again, we continue this process until the ties are resolved. If two or more teams are still tied after considering all positions, we rank them alphabetically based on their team letter.

Given an array of strings votes which is the votes of all voters in the ranking systems. Sort all teams according to the ranking system described above.

Return a string of all teams sorted by the ranking system.

 

Example 1:

Input: votes = ["ABC","ACB","ABC","ACB","ACB"]
Output: "ACB"
Explanation: Team A was ranked first place by 5 voters. No other team was voted as first place so team A is the first team.
Team B was ranked second by 2 voters and was ranked third by 3 voters.
Team C was ranked second by 3 voters and was ranked third by 2 voters.
As most of the voters ranked C second, team C is the second team and team B is the third.

Example 2:

Input: votes = ["WXYZ","XYZW"]
Output: "XWYZ"
Explanation: X is the winner due to tie-breaking rule. X has same votes as W for the first position but X has one vote as second position while W doesn't have any votes as second position. 

Example 3:

Input: votes = ["ZMNAGUEDSJYLBOPHRQICWFXTVK"]
Output: "ZMNAGUEDSJYLBOPHRQICWFXTVK"
Explanation: Only one voter so his votes are used for the ranking.

Example 4:

Input: votes = ["BCA","CAB","CBA","ABC","ACB","BAC"]
Output: "ABC"
Explanation: 
Team A was ranked first by 2 voters, second by 2 voters and third by 2 voters.
Team B was ranked first by 2 voters, second by 2 voters and third by 2 voters.
Team C was ranked first by 2 voters, second by 2 voters and third by 2 voters.
There is a tie and we rank teams ascending by their IDs.

Example 5:

Input: votes = ["M","M","M","M"]
Output: "M"
Explanation: Only team M in the competition so it has the first rank.

 

Constraints:

  • 1 <= votes.length <= 1000
  • 1 <= votes[i].length <= 26
  • votes[i].length == votes[j].length for 0 <= i, j < votes.length.
  • votes[i][j] is an English upper-case letter.
  • All characters of votes[i] are unique.
  • All the characters that occur in votes[0] also occur in votes[j] where 1 <= j < votes.length.

Rust Solution

struct Solution;
use std::cmp::Ordering::*;

impl Solution {
    fn rank_teams(votes: Vec<String>) -> String {
        let n = votes[0].len();
        let mut count: Vec<Vec<usize>> = vec![vec![0; n]; 26];
        for s in &votes {
            for (i, c) in s.bytes().enumerate() {
                count[(c - b'A') as usize][i] += 1;
            }
        }
        let mut v: Vec<u8> = votes[0].bytes().map(|b| b - b'A').collect();
        v.sort_by(|&a, &b| {
            for i in 0..n {
                match &count[a as usize][i].cmp(&count[b as usize][i]) {
                    Equal => {}
                    Less => {
                        return Greater;
                    }
                    Greater => {
                        return Less;
                    }
                }
            }
            a.cmp(&b)
        });
        v.into_iter().map(|b| (b'A' + b) as char).collect()
    }
}

#[test]
fn test() {
    let votes = vec_string!["ABC", "ACB", "ABC", "ACB", "ACB"];
    let res = "ACB".to_string();
    assert_eq!(Solution::rank_teams(votes), res);
    let votes = vec_string!["WXYZ", "XYZW"];
    let res = "XWYZ".to_string();
    assert_eq!(Solution::rank_teams(votes), res);
    let votes = vec_string!["ZMNAGUEDSJYLBOPHRQICWFXTVK"];
    let res = "ZMNAGUEDSJYLBOPHRQICWFXTVK".to_string();
    assert_eq!(Solution::rank_teams(votes), res);
    let votes = vec_string!["BCA", "CAB", "CBA", "ABC", "ACB", "BAC"];
    let res = "ABC".to_string();
    assert_eq!(Solution::rank_teams(votes), res);
    let votes = vec_string!["M", "M", "M", "M"];
    let res = "M".to_string();
    assert_eq!(Solution::rank_teams(votes), res);
}

Having problems with this solution? Click here to submit an issue on github.