不开辟新空间统计正整数数组元素个数

最新推荐文章于 2020-02-26 17:11:14 发布

xiekch

最新推荐文章于 2020-02-26 17:11:14 发布

阅读量404

点赞数 1

文章标签：在原数组中统计元素个数 python 计数

本文链接：https://blog.csdn.net/xiekch/article/details/81356495

版权

算法要求

给定一个数组，长度为N，元素取值范围为[1,N]。
统计各个元素出现的次数，时间复杂度为O(N)，空间复杂度为O(1)。

python实现

#xiekch
#2018.8.2
import numpy as np
import pandas as pd
n=100
arr=np.random.randint(1,n+1,n)
print(arr)
count_ans=pd.value_counts(arr).sort_index()

for i in range(n):
    if arr[i]<=0:
        continue
    j=arr[i]-1
    arr[i]=0
    while j>=0:
        k=arr[j]-1
        if arr[j]>=0:
            arr[j]=-1
        else:
            arr[j]=arr[j]-1
        j=k
print(arr)
#test
ok=True
for i in range(n):
    if arr[i]<0 and count_ans[i+1]!=-arr[i]:
        print('Wrong answer!')
        ok=False
        break
if ok:
    print('Correct answer!')