We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Count Triplets
  2. Discussions

Count Triplets

  • atomlove
     + 0 comments

    This Javascript solution works except all 1's test cases. I don't know the reason why time limit exceptions happened in many cases though.

    function countTriplets(arr, r) {
    let count = 0;
    
    for (let i=0; i<arr.length - 2; i++) {     
        let base = arr[i];
        let baseTwo = arr[i] * r;            
        let baseThree = arr[i] * r * r       
        let twoCount = 0;
        let threeCount = 0;   
    
        twoCount = arr.slice(i+1).filter(elem => elem === baseTwo).length;
        threeCount = arr.slice(i+1).filter(elem => elem === baseThree).length;
        count = count + (twoCount * threeCount);        
    }
    
    return count;
}