fork download
  1. seq_diff = lambda arr: [abs(b - a) for a, b in zip(arr, arr[1:])]
  2.  
  3. is_jolly = lambda arr: set(seq_diff(sorted(seq_diff(arr)))) == {1}
  4.  
  5. def jollies(inputs):
  6. for line in inputs.splitlines():
  7. arr = map(int, line.split())[1:]
  8. print '{} {}JOLLY'.format(line, 'NOT ' * (not is_jolly(arr)))
  9.  
  10. jollies('''\
  11. 4 1 4 2 3
  12. 5 1 4 2 -1 6
  13. 4 19 22 24 21
  14. 4 19 22 24 25
  15. 4 2 -1 0 2''')
Success #stdin #stdout 0s 23352KB
stdin
Standard input is empty
stdout
4 1 4 2 3 JOLLY
5 1 4 2 -1 6 NOT JOLLY
4 19 22 24 21 NOT JOLLY
4 19 22 24 25 JOLLY
4 2 -1 0 2 JOLLY