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

算法要求

• 给定一个数组，长度为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!')