2017京东实习生招聘编程题——站队抓小偷（python）

站队（京东2017实习生真题）
题目描述

有一条很长的队伍，队伍里面一共有n个人。所有的人分为三类：警察，小偷和普通人。将队伍里面的人从前到后由1到n编号，编号为i的人与编号为j的人的距离为i与j之差的绝对值。每一个警察有一个能力值x，表示他能够监视与他距离不超过x的所有人，小偷被警察发现当且仅当他被一个或多个警察监视到。你知道在整条队伍中，一共有多少个小偷会被警察发现吗？

输入 输入有两行，第一行一个数n(1<=n<=100000)，接下来一行有一个长度为n的字符串，依次表示队伍中的每一个人。如果某一位是1-9的某个数字x，表示这一位是一个能力值为x的警察；如果某一位是字符X表示这一位是小偷；如果某一位是字符#表示这是一个普通人。输入保证不会出现其它字符。	样例输入 9 X1X#2X#XX
输出 输出一个数，整条队伍中被警察发现的小偷总数。	样例输出 3

# -*- coding: utf-8 -*-
n=raw_input()
n=int(n)
sque=raw_input()
person_type_dict={}
police_distance_dict={}
for i in range(1,n+1):
p=sque[i-1]
if p=='X':
person_type_dict[i]='steal'
elif p=='#':
person_type_dict[i]='ordinary'
else:
person_type_dict[i]='police'
police_distance_dict[i]=int(p)
goal=[]
for k in police_distance_dict.keys():
dis=police_distance_dict[k]
min_num=max(1,k-dis)
max_num=min(k+dis,n)
pra=[]
for j in range(min_num,k)+range(k+1,max_num+1):
if person_type_dict[j]=='steal':
pra.append(j)
goal=goal+pra
goal=set(goal)
print len(goal)