我正在使用Python解决Facebook Hackercup上的查找最小值
问题,我的代码可以很好地处理样本输入,但对于大输入(10^9),需要花费数小时才能完成。
那么,这个问题的解决方案是否可能无法在6分钟内使用Python计算?或者我的方法太糟糕了?
问题陈述:
在发送表情符号后,John决定玩玩数组。您知道黑客喜欢玩数组吗?John有一个从零开始索引的数组m
,其中包含n
个非负整数。然而,他只知道数组的前k
个值,并且他想弄清剩下的值。
John知道以下内容:对于每个索引i,其中k <= i < n
,m [i]
是不包含在前*k*
个m
值中的最小非负整数。
k = 3
,n = 4
,并且已知m
的值为[2, 3, 0]
,他可以推断出m[3] = 1
。John非常忙碌,致力于使世界更加开放和连接,因此他没有时间去计算数组的其余部分。你的任务是帮助他。
给定
m
的前k
个值,计算该数组的第n
个值(即m[n - 1]
)。由于
n
和k
的值可能非常大,我们使用伪随机数生成器来计算m
的前k
个值。给定正整数a
、b
、c
和r
,可以按以下方式计算m
的已知值:m[0] = a
m[i] = (b * m[i - 1] + c) % r, 0 < i < k
输入
第一行包含一个整数T(T <= 20),表示测试用例的数量。
接下来是T个测试用例,每个测试用例由两行组成。
每个测试用例的第一行包含两个以空格分隔的整数n和k(
1 <= k <= 10^5
,k < n <= 10^9
)。每个测试用例的第二行包含四个以空格分隔的整数a、b、c、r(0 <= a、b、c <= 10^9,1 <= r <= 10^9)。
我尝试了两种方法,但都无法在6分钟内返回结果,以下是我的两种方法:
第一种:
import sys
cases=sys.stdin.readlines()
def func(line1,line2):
n,k=map(int,line1.split())
a,b,c,r =map(int,line2.split())
m=[None]*n #initialize the list
m[0]=a
for i in xrange(1,k): #set the first k values using the formula
m[i]= (b * m[i - 1] + c) % r
#print m
for j in range(0,n-k): #now set the value of m[k], m[k+1],.. upto m[n-1]
temp=set(m[j:k+j]) # create a set from the K values relative to current index
i=-1 #start at 0, lowest +ve integer
while True:
i+=1
if i not in temp: #if that +ve integer is not present in temp
m[k+j]=i
break
return m[-1]
for ind,case in enumerate(xrange(1,len(cases),2)):
ans=func(cases[case],cases[case+1])
print "Case #{0}: {1}".format(ind+1,ans)
第二个:
import sys
cases=sys.stdin.readlines()
def func(line1,line2):
n,k=map(int,line1.split())
a,b,c,r =map(int,line2.split())
m=[None]*n #initialize
m[0]=a
for i in xrange(1,k): #same as above
m[i]= (b * m[i - 1] + c) % r
#instead of generating a set in each iteration , I used a
# dictionary this time.
#Now, if the count of an item is 0 then it
#means the item is not present in the previous K items
#and can be added as the min value
temp={}
for x in m[0:k]:
temp[x]=temp.get(x,0)+1
i=-1
while True:
i+=1
if i not in temp:
m[k]=i #set the value of m[k]
break
for j in range(1,n-k): #now set the values of m[k+1] to m[n-1]
i=-1
temp[m[j-1]] -= 1 #decrement it's value, as it is now out of K items
temp[m[k+j-1]]=temp.get(m[k+j-1],0)+1 # new item added to the current K-1 items
while True:
i+=1
if i not in temp or temp[i]==0: #if i not found in dict or it's val is 0
m[k+j]=i
break
return m[-1]
for ind,case in enumerate(xrange(1,len(cases),2)):
ans=func(cases[case],cases[case+1])
print "Case #{0}: {1}".format(ind+1,ans)
第二种方法中的最后一个for循环也可以写成:
for j in range(1,n-k):
i=-1
temp[m[j-1]] -= 1
if temp[m[j-1]]==0:
temp.pop(m[j-1]) #same as above but pop the key this time
temp[m[k+j-1]]=temp.get(m[k+j-1],0)+1
while True:
i+=1
if i not in temp:
m[k+j]=i
break
示例输入:
5
97 39
34 37 656 97
186 75
68 16 539 186
137 49
48 17 461 137
98 59
6 30 524 98
46 18
7 11 9 46
输出:
Case #1: 8
Case #2: 38
Case #3: 41
Case #4: 40
Case #5: 12
我已经尝试过 codereview,但还没有人回复。