fork(2) download
  1. import math
  2. t=input()
  3. for i in range(t):
  4. n=input()
  5. c=[]
  6. sum=0
  7. a=raw_input()
  8. b=map(int,raw_input().split())
  9. for j in range(0,len(a)):
  10. if(a[i]=='1'):
  11. c.append(j)
  12. sum=0
  13.  
  14. for j in range(0,n):
  15. if(a[j]=='1'):
  16. l=j+1
  17. minimum=1000000000000
  18. for f in range(l,n):
  19. if(a[f]!='1'):
  20. l=l+1
  21. #print l
  22. if(j+1==l or l>=n):
  23. continue
  24. for p in range(j+1,l+1,1):
  25. ans=b[l+j-p]-b[j]+b[l]-b[l-p+j+1]
  26. if(minimum>ans):
  27. minimum=ans
  28. sum=sum+minimum
  29. #print sum
  30. if(c[0]>a.index('0')):
  31. sum=sum+b[c[0]]-b[a.index('0')]
  32. if(c[len(c)-1]!=n-1):
  33. sum=sum+b[n-1]-b[c[len(c)-1]]
  34. print sum
Runtime error #stdin #stdout #stderr 0.01s 9016KB
stdin
1
2
01
1 2
stdout
Standard output is empty
stderr
Traceback (most recent call last):
  File "prog.py", line 30, in <module>
IndexError: list index out of range