fork download
copy 
  1. # -*- coding: cp1252 -*-
  2. """ Problem SPOJ 1025: Fashion shows: Mix hottest couples (FASHION) """
  3. import sys
  4. # import psyco
  5. # psyco.full()
  6.  
  7. def init_testcase():
  8.     cases = [
  9.             '2',
  10.             '2',
  11.             '1 1',
  12.             '3 2',
  13.             '3',
  14.             '2 3 2',
  15.             '1 3 2',
  16.             ]
  17.     return cases
  18.  
  19. def hotsum (params):
  20.     # print params
  21.     men = map(int, params[0])
  22.     women = map(int, params[1])
  23.     men.sort()
  24.     women.sort()
  25.     result = 0
  26.     for m, w in zip (men, women):
  27.         result += m * w
  28.     return result
  29.  
  30.  
  31. def split_it(s):
  32.     return s.split()
  33.  
  34. def main():
  35.     testflag = 0
  36.     if testflag:
  37.         cases = []
  38.         cases.extend(init_testcase())
  39.         cases.append('') # dummy to cope with suppression of last line when reading from stdin
  40.         print cases
  41.     else:
  42.         cases = sys.stdin.read().split('\n')
  43.  
  44.     results = []
  45.     for line in zip(*[iter(cases[1:-1])]*3):
  46.         # print 'line', line
  47.         if line:
  48.             result = hotsum(map(split_it, line[1:]))
  49.             results.append(str(result))
  50.     sys.stdout.writelines ('\n'.join(results))
  51.     sys.stdout.write('\n')
  52.  
  53.  
  54. if __name__ == '__main__':
  55.     main()
Success #stdin #stdout 0.02s 4780KB
comments (?)
stdin
copy 
2
2
1 1
3 2
3
2 3 2
1 3 2
stdout
copy 
5
15
https://ideone.com/nPbR5
language:
Python (cpython 2.7.16)
created:
14 years ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!