[已解决]谁能帮我优化一下代码，重重有赏！

sfqxx · 发表于 2023-4-15 20:58:02

链接->https://atcoder.jp/contests/abc298/tasks/abc298_d
我的代码：

mod = 998244353
s = "1"
q = int(input())
for i in range(q):
query = input().split()
if query[0] == "1":
s += query[1]
elif query[0] == "2":
s = s[1:]
else:
num = int(s) % mod
print(num)

复制代码

TLE了，有哪位大神帮我优化一下代码，谢谢！

最佳答案

月排行榜 / 总排行榜

liuhongrun2022

2023-4-15 20:58:03

本帖最后由 liuhongrun2022 于 2023-4-15 21:46 编辑

注：并非我写
实测 AC

from collections import deque
mod = 998244353
arr = deque([1])
num = 1
q = int(input())
for i in range(q):
op = input().split()
if op[0] == '1':
x = int(op[1])
arr.append(x)
num = (num * 10 + x) % mod
elif op[0] == '2':
x = arr.popleft()
num = ((num - x * pow(10, len(arr), mod)) % mod + mod) % mod
else:
print(num)

复制代码

跳转到最佳答案楼层

liuhongrun2022 · 发表于 2023-4-15 20:58:03

这个最佳答案由 liuhongrun2022 给出，感谢 liuhongrun2022 的回答。

单击隐藏图章

本帖最后由 liuhongrun2022 于 2023-4-15 21:46 编辑

注：并非我写
实测 AC

from collections import deque
mod = 998244353
arr = deque([1])
num = 1
q = int(input())
for i in range(q):
op = input().split()
if op[0] == '1':
x = int(op[1])
arr.append(x)
num = (num * 10 + x) % mod
elif op[0] == '2':
x = arr.popleft()
num = ((num - x * pow(10, len(arr), mod)) % mod + mod) % mod
else:
print(num)

复制代码

sfqxx · 发表于 2023-4-15 21:00:28

liuhongrun2022 · 发表于 2023-4-15 21:13:47

进去链接404了

sfqxx · 发表于 2023-4-15 21:17:58

liuhongrun2022 发表于 2023-4-15 21:13
进去链接404了

稍等

liuhongrun2022 · 发表于 2023-4-15 21:18:53

sfqxx 发表于 2023-4-15 21:17
稍等

能用了

sfqxx · 发表于 2023-4-15 21:30:47

liuhongrun2022 发表于 2023-4-15 21:18
能用了

会吗？

liuhongrun2022 · 发表于 2023-4-15 21:40:35

本帖最后由 liuhongrun2022 于 2023-4-15 21:41 编辑

sfqxx 发表于 2023-4-15 21:30
会吗？

AC，但是问了gpt

isdkz · 发表于 2023-4-15 21:45:39

你的代码已经很好了，但是可以使用一个变量来记录当前的字符串长度，从而避免在每次删除首位数字时都需要切片。以下是优化后的代码：

mod = 998244353

s = "1"

length = 1

q = int(input())

for i in range(q):

query = input().split()

if query[0] == "1":

      s += query[1]

      length += 1

elif query[0] == "2":

      s = s[length - 1:]

      length -= 1

else:

      num = int(s) % mod

      print(num)
复制代码

这个优化可以提高代码的性能，因为在每次删除首位数字时，不再需要对整个字符串进行切片。

sfqxx · 发表于 2023-4-15 22:26:36

isdkz 发表于 2023-4-15 21:45
你的代码已经很好了，但是可以使用一个变量来记录当前的字符串长度，从而避免在每次删除首位数字时都需要切 ...

你的代码RE了

sfqxx · 发表于 2023-4-15 22:27:16

liuhongrun2022 发表于 2023-4-15 20:58
注：并非我写
实测 AC

哪个GPT，这么厉害？

isdkz · 发表于 2023-4-15 22:38:12

sfqxx 发表于 2023-4-15 22:26
你的代码RE了

试试这个：

from collections import deque

import array

mod = 998244353

s = deque(array.array("I", [1]))  # 使用array模块创建整数数组

current_mod = 1

q = int(input())

# 预计算10的幂次模值

pow_10_mod = [1]

for _ in range(q):

pow_10_mod.append((pow_10_mod[-1] * 10) % mod)

for i in range(q):

query = input().split()

if query[0] == "1":

      s.append(int(query[1]))

      current_mod = (current_mod * 10 + int(query[1])) % mod

elif query[0] == "2":

      current_mod = (current_mod - s.popleft() * pow_10_mod[len(s) - 1]) % mod

else:

      print(current_mod)

复制代码

sfqxx · 发表于 2023-4-15 22:46:46

isdkz 发表于 2023-4-15 22:38
试试这个：

isdkz · 发表于 2023-4-15 23:14:46

sfqxx 发表于 2023-4-15 22:46

好吧，难道我的gpt不够厉害，还是我不会问？

sfqxx · 发表于 2023-4-15 23:17:10

isdkz 发表于 2023-4-15 23:14
好吧，难道我的gpt不够厉害，还是我不会问？

我觉得是……
题目的问题

sfqxx · 发表于 2023-4-15 23:17:53

isdkz 发表于 2023-4-15 23:14
好吧，难道我的gpt不够厉害，还是我不会问？

你重新问问试试看？
所有题目，信息都复制进去

isdkz · 发表于 2023-4-15 23:22:55

sfqxx 发表于 2023-4-15 23:17
你重新问问试试看？
所有题目，信息都复制进去

我全部复制进去了，连着问好几次得到的代码才能AC，评测不如liuhongrun2022的那个代码优

sfqxx · 发表于 2023-4-15 23:25:18

isdkz 发表于 2023-4-15 23:22
我全部复制进去了，连着问好几次得到的代码才能AC，评测不如liuhongrun2022的那个代码优

到时候问问他吧
对了，qq通过一下
您的chATgpt是那个网页的？

isdkz · 发表于 2023-4-15 23:26:13

sfqxx 发表于 2023-4-15 23:25
到时候问问他吧
对了，qq通过一下
您的chATgpt是那个网页的？

ai.com

isdkz · 发表于 2023-4-15 23:27:54

sfqxx 发表于 2023-4-15 23:25
到时候问问他吧
对了，qq通过一下
您的chATgpt是那个网页的？

你确定你有加我？我都没有收到请求

账号		自动登录	找回密码
密码			立即注册

[已解决]谁能帮我优化一下代码，重重有赏！

最佳答案

浏览过的版块