以下是我编写的用于计算第 n 个素数的代码。我初始化 arr = [2,3,5]; 并且 prime(arr,n) 应该修改 arr 以包含前 n 个素数。但 arr 不反映 prime(arr,n) 执行后的变化。我读到列表是作为引用类型传递的,所以下面的代码有什么问题。t 表示必须进行验证的测试用例的数量。
该程序导致错误,指出数组索引超出了 arr 的范围。
import sys;
def prime (arr,n):
while(len(arr)< n):
num=arr[len(arr)-1]+1;prime=0;
while(prime==0):
prime = 1
for val in arr:
if(num%val==0):
prime=0;
break;
if(prime == 1):
print "hello";
arr = arr + [num];print arr; print "--";
else:
num = num+1;
t=raw_input();
t=int(t);
arr=[2,3,5];
ans =[];
for v in range (0,t):
n = raw_input();
n = int(n);
if(n<=len(arr)):
ans = ans + [arr[n-1]];
else:
prime(arr,n);print arr;print"arr was printed"
ans= ans + [arr[n-1]];
print ans;print 'ans ';
以下是上述代码的示例运行
>>>
1
4
hello
[2, 3, 5, 7]
--
[2, 3, 5]
arr was printed
Traceback (most recent call last):
File "C:\Users\Pulkit\Desktop\Random\nth_prime.py", line 30, in <module>
ans= ans + [arr[n-1]];
IndexError: list index out of range
提前致谢 :)