[HackerRank] Equal stacks

[HackerRank] Equal stacks

---

 

 

• 문제 링크

https://www.hackerrank.com/challenges/equal-stacks/problem?isFullScreen=true 

 

---

시간복잡도: O(n)

 

• 코드

def equalStacks(h1, h2, h3):

    sumH1 = sum(h1)
    sumH2 = sum(h2)
    sumH3 = sum(h3)
      
    while 1:

        minNum=min(sumH1, sumH2, sumH3)
        
        if(sumH1==sumH2==sumH3): #equal?
            break

        
        if(sumH1 > minNum): #sum > min?
            sumH1 -= h1.pop(0) 
            
        if(sumH2 > minNum):
            sumH2 -= h2.pop(0)
            
        if(sumH3 > minNum):
            sumH3 -= h3.pop(0)
 
    return minNum