【华为OD机试2023】相同数字的积木游戏1(Python)

题目描述：

小华和小薇一起通过玩积木游戏学习数学。

他们有很多积木，每个积木块上都有一个数字，积木块上的数字可能相同。

小华随机拿一些积木挨着排成一排，请小薇找到这排积木中数字相同且所处位置最远的2块积木块，计算他们的距离。

小薇请你帮忙替解决这个问题。

输入描述：

第一行输入为N，表示小华排成一排的积木总数。

接下来N行每行一个数字，表示小华排成一排的积木上数字。

输出描述：

相同数字的积木的位置最远距离；

如果所有积木数字都不相同，请返回-1.

补充说明：

0<=积木上的数字<10^9

1<=积木长度<=10^5

示例1

输入：

5

1

2

3

1

4

输出：

3

说明：

一共有5个积木，第1个积木和第4个积木数字相同，其距离为3；

示例2

输入：

2

1

2

输出：

-1

说明：

一共有2个积木，没有积木数字相同，返回-1；

解题思路：

①先拿到所有的重复数字，用set()函数去重（我这里写麻烦了，不需要.remove()，直接.count()计数大于等于2即可，这样后面就不需要重新设一个zhengxiang列表了，可以自行调整）

②用.index()获得索引，我这里设置了正向和反向列表，可以获得原列表中某数字第一个和最后一个索引

③拿到索引后，计算距离即可

#优化方法
n=int(input())
lst=[int(input()) for i in range(n)]
dict={}
for i,value in enumerate(lst):
    if dict.get(value)==None:
        dict[value]=[i]
    else:
        dict[value].append(i)
max_dis=0
for shuzi in dict:
    temp=dict[shuzi]
    if len(temp)>=2:
        dis=temp[-1]-temp[0]
        max_dis=max(max_dis,dis)
print(max_dis if max_dis!=0 else -1)


#旧方法
def youchongfu(lst:list):
    num=list(set(lst))
    chongfu=[]
    for word in num:
        lst.remove(word)
        if lst.count(word)>0:
            chongfu.append(word)
    return chongfu

n=int(input())
jimu=[]

for i in range(n):
    jimu.append(int(input()))
fanxiang=jimu[::-1]
zhengxiang=fanxiang[::-1]
chongfushuzi=youchongfu(jimu)
max_dis=0
if chongfushuzi==[]:
    print(-1)
else:
    for shuzi in chongfushuzi:
        dis=n-1-fanxiang.index(shuzi)-zhengxiang.index(shuzi)
        max_dis=max(dis,max_dis)
    print(max_dis)