fork download
copy 
  1. import math
  2. t=input()
  3. for i in range(t):
  4. 	n=input()
  5. 	c=[]
  6. 	a=raw_input()
  7. 	b=map(int,raw_input().split())
  8. 	sum=0
  9. 	for j in range(0,n):
  10. 		x=0
  11. 		if(a[j]=='1'):
  12. 			c=j
  13. 			l=j+1
  14. 			minimum=1000000000000
  15. 			for f in range(l,n):
  16. 				if(a[f]!='1'):
  17. 					l=l+1
  18. 				else:
  19. 					break
  20. 			#print l	
  21. 			if(j+1==l or l>=n):
  22. 				continue
  23. 			for p in range(j+1,l+1,1):
  24. 				ans=b[l+j-p]-b[j]+b[l]-b[l-p+j+1]
  25. 				if(minimum>ans):
  26. 					minimum=ans
  27. 			#print sum		
  28. 			sum=sum+minimum	
  29. 			#print sum
  30. 		elif(a[j]=='0' and j==0):
  31. 			x=0
  32. 			while(a[x]!=1):
  33. 				x=x+1
  34. 			sum=sum+b[x]-b[0]
  35. 			#print sum
  36. 	if(a[n-1]=='0'):
  37. 		sum=sum+b[n-1]-b[c]
  38.  
  39. 	print sum
Runtime error #stdin #stdout #stderr 0.02s 9016KB
comments (?)
stdin
copy 
2
2
01
1 2
3
100
1 5 6
stdout
copy 
Standard output is empty
stderr
copy 
Traceback (most recent call last):
  File "prog.py", line 32, in <module>
IndexError: string index out of range
https://ideone.com/tGF3OJ
language:
Python (cpython 2.7.16)
created:
7 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!