QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#765655#5460. Sum of NumbersO_start#TL 865ms17012kbPython31.8kb2024-11-20 14:55:032024-11-20 14:55:07

Judging History

你现在查看的是最新测评结果

  • [2024-11-20 14:55:07]
  • 评测
  • 测评结果:TL
  • 用时:865ms
  • 内存:17012kb
  • [2024-11-20 14:55:03]
  • 提交

answer

import sys
sys.set_int_max_str_digits(0)
t = int(input())
n, k = 0, 0
s = ''

d = 7


def solve():
    len0 = n // (k + 1)
    pos0 = [len0 * i + len0 - 1 for i in range(k)]
    dp = [[-1 for i in range(d * 2)] for _ in range(k)]
    for i in range(k):
        if i == 0:
            for j in range(d * 2):
                p1 = pos0[i] + (j - d)
                if p1 < 0 or p1 >= n:
                    continue
                dp[i][j] = int(s[0: p1 + 1])
            continue
        for j in range(d * 2):
            for pre_j in range(d * 2):
                p1 = pos0[i] + (j - d)
                pre_p1 = pos0[i - 1] + (pre_j - d)
                if p1 < 0 or p1 >= n:
                    continue
                if pre_p1 < 0 or pre_p1 >= n:
                    continue
                if not pre_p1 < p1:
                    continue
                if dp[i - 1][pre_j] == -1:
                    continue
                temp = dp[i - 1][pre_j] + int(s[pre_p1 + 1: p1 + 1])
                # dp[i][j] = temp if dp[i][j] == -1 or temp < dp[i][j] else -1
                if dp[i][j] == -1:
                    dp[i][j] = temp
                else:
                    dp[i][j] = min(dp[i][j], temp)
    ans = -1
    for pre_j in range(d * 2):
        pre_p1 = pos0[len(pos0) - 1] + (pre_j - d)
        if pre_p1 < 0 or pre_p1 >= n - 1:
            continue
        if dp[k - 1][pre_j] == -1:
            continue
        temp = dp[k - 1][pre_j] + int(s[pre_p1 + 1:])
        if ans == -1:
            ans = temp
        else:
            ans = min(ans, temp)

    # for i in range(k):
    #     print(dp[i])
    print(ans)


for _ in range(t):
    ins = input().split(' ')
    n, k = int(ins[0]), int(ins[1])
    s = input()
    # for _ in range(n):
    #     s += '5'
    solve()


'''
1
8 1
45455151
'''

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 17ms
memory: 10696kb

input:

2
8 1
45455151
2 1
42

output:

9696
6

result:

ok 2 lines

Test #2:

score: 0
Accepted
time: 653ms
memory: 13520kb

input:

10
1301 6
56328399613959594774559774218276494124991536454496431869449134772679831477279356599352619469813771742358572734317965823527349354276551857226632977613336815474383422853946661428822284645652423563864641261338984158269966469425994769486371736593879954275146732544891889693921182364554588732946...

output:

2861837555106640794797067737879913860686764066159587941287350938727749577629356630565034353414526438507603808735990935008225192080065174423508575377930722196909797866802717925250679901255
1330897896655974774035586406544907434842835048336411271110427836483063457950873824562288934364096546537492367401...

result:

ok 10 lines

Test #3:

score: 0
Accepted
time: 221ms
memory: 14616kb

input:

3
68312 1
97721793232462295468345161395388245623318759168492992575579675893432997968276399269361852215552871434587866112799189725614696256526239439933158479577223186954475137216722241699568491426254779997896288355965839555276546655174417599498493234323661877664997543475525992228661739933395881544363...

output:

165361439370989969611281150911906045608758081895436745302721903170895741831200164992289491213814537683869854746077326282091529052307364873668669947061960472772133388934416763345394409983809105665214082485429685401199520114459892594221344388768178495952936142426649338494017013932243993161711816767032...

result:

ok 3 lines

Test #4:

score: 0
Accepted
time: 865ms
memory: 17012kb

input:

3
104062 2
6586987777965872861989232258194733262765989634859654379877935579528781688159322513128986886481716832258115626748535521126479244461411522889744479186513298679746684345233214226141199846787117661159318966767135455745184384324833645134232699525177668629239933843658643667417337451334591594819...

output:

536246497591460214899497073157707308719964754106858053886979792537707870310253164299747153718261114131636990130350872742851787818427135113617679220183754863021102973854172956224432906133335696002890140207617299467927865340000078212572980076032917286214327823207585568582552662612028254042515906063023...

result:

ok 3 lines

Test #5:

score: -100
Time Limit Exceeded

input:

3
139639 3
6796297676823528648589397343663999328612414278384367347213637689464348185237425534836886677731351726963453579377613988437426483671299519186839344132333793569544718489728294346989818592279444871423953477919473799463194216678119582972131632322347549538925164854516141164554772823372476647126...

output:

126364763403906564661708542497996384758493064461499142133308596067494936258655259133526434841726196274703766844552726915088706942994746309192813887599486034682622494344506129054929148805249803660505979746418821689885230718372401950523582639142168746487771785553740446005221635828358569536474437471762...

result: