我正在尝试解决Hackerearth上的问题。问题陈述如下:
Sussutu是举世闻名的魔术师。最近,他很幸运能够从阵列中删除“仅一个”元素。给定一个具有N个元素的数组A(索引从0开始)。现在,Sussutu只能删除使其余所有元素之和可被7整除的元素。Sussutu在他的一生中一直忙于魔术,以至于他永远无法与数学相处。您的任务是帮助Sussutu找到他可以删除的最小元素的第一个数组索引。
输入:第一行包含一个整数N。下一行包含N个空格分隔的整数Ak,0 <k <N。
输出:打印包含一行整数的一行,这是他可以删除的最小元素的第一个数组索引,如果没有他可以删除的元素,则打印-1。
约束:1 <N <105
0 <Ak <109
这是我尝试的算法,但是在某些测试案例中,它超出了时间限制:
n = int(input())
A = list(map(int, input().split(' ')))
temp = sorted(A)
for i in range(n):
temp[i] = 0
s = sum(temp)
temp = sorted(A)
if s % 7 == 0:
flag = True
break
flag = False
if flag == True:
print(A.index(temp[i]))
else:
print(-1)
另一个工作正常的代码如下:
n = int(input())
A = list(map(int, input().split(' ')))
S = sum(A)
t = []
for a in A:
if (S - a) % 7 == 0:
t.append(a)
if len(t) == 0:
print(-1)
else:
print(A.index(min(t)))
谁能帮助我理解为什么第一代码超出了时间限制,为什么第二代码没有超过时间?
问题来源:stackoverflow
在第一种算法中,排序本身为O(n log n),因此第一个循环的复杂度为O(n)*(n log n)= O(n²log n)`。在第二个中,您只需遍历输入数组3次-因此它的复杂度是O(n),要低得多。对于非常大的输入,则第一个输入将超时,而第二个输入可能不会超时。
回答来源:stackoverflow
版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。