Square的店面

求问楼主是哪个岗位呀

1. from collections import defaultdict
   
2. 
   
3. 
   
4. class Solution:
   
5.     def get_balance(self, transations):
   
6.         balance_dct = defaultdict(int)
   
7.         for amount, payer, payees in transations:
   
8.             balance_dct[payer] -= amount
   
9.             amount_payee = float(amount) / len(payees)
   
10.             for payee in payees:
    
11.                 balance_dct[payee] += amount_payee
    
12.         return balance_dct
    
13. 
    
14.     def get_due(self, balance_dct):
    
15.         if sum(balance_dct.values()) != 0:
    
16.             raise Exception('Not balance')
    
17.         debets = sorted([it for it in balance_dct.items() if it[1] > 0], key=lambda x: x[1])
    
18.         credit = sorted([it for it in balance_dct.items() if it[1] < 0], key=lambda x: -x[1])
    
19.         give_dct = defaultdict(list)
    
20.         while debets and credit:
    
21.             debet_user, debet_val = debets.pop()
    
22.             credit_user, credit_val = credit.pop()
    
23.             bal = debet_val + credit_val
    
24.             if bal > 0:
    
25.                 debets.append((debet_user, bal))
    
26.             elif bal < 0:
    
27.                 credit.append((credit_user, bal))
    
28. 
    
29.             give_dct[debet_user].append((credit_user, min(debet_val, -credit_val)))
    
30.         return give_dct
    

复制代码

1. import unittest
   
2. from solution import Solution
   
3. 
   
4. class Tests(unittest.TestCase):
   
5.     def test_1(self):
   
6.         sol = Solution()
   
7.         transations = [(12, 'A', ('A', 'B', 'C')), (10, 'B', ('A', 'B'))]
   
8.         res = sol.get_balance(transations)
   
9.         print res
   
10. 
    
11.     def test_2(self):
    
12.         sol = Solution()
    
13.         transations = [(12, 'A', ('A', 'B', 'C')), (10, 'B', ('A', 'B'))]
    
14.         balance_dct = sol.get_balance(transations)
    
15.         res = sol.get_due(balance_dct)
    
16.         print res
    
17. 
    
18. if __name__ == '__main__':
    
19.     unittest.main(verbosity=2)
    

复制代码