题解 | #编程题1#

自己想了很久，不会，看了下讨论里的思路，ac

核心：

• 根据a、b、c三支队伍的得分多少d1，d2，可以分为4种情况
• 我们不确定那种情况是可以达到要求的，所以需要遍历4种情况
• 每种情况都可以用公式求解出对应的abc的值，也就是三队各打了多少场比赛
• 在abc时，可以根据是否能整除？结果是否大于0？两个条件剪枝
• 每算出一种情况后计算n-k，看剩余的比赛场次能否使得最后三队得分相同

代码看着较长，但是逻辑还是比较清晰的，一看就懂

import collections,bisect,heapq
while True:
    try:
        m=int(input().strip())
        mat=[]
        for _ in range(m):
            a=list(map(int,input().strip().split(' ')))
            mat.append(a)
        for i in range(len(mat)):
            n,k,d1,d2=mat[i][0],mat[i][1],mat[i][2],mat[i][3]
            if n-k<d1+d2:
                print('no')
            else:
                #此时剩余rest场比赛，并且各队持平
                #需要考虑三个队伍胜负场与k的大小
                #这里应该可以形成一个公式
                for flag in '1234':
                    #a队比b队多d1分，b队比c队多d2分
                    if flag=='1':
                        if (k+2*d1+d2)%3!=0 or (k-d1+d2)%3!=0 or (k-d1-2*d2)%3!=0:
                            continue
                        a=(k+2*d1+d2)//3
                        b=(k-d1+d2)//3
                        c=(k-d1-2*d2)//3
                        if a<0 or b<0 or c<0:
                            continue
                    #a队比b队多d1分，c队比b队多d2分
                    if flag=='2':
                        if (k+2*d1-d2)%3!=0 or (k-d1-d2)%3!=0 or (k-d1+2*d2)%3!=0:
                            continue
                        a=(k+2*d1-d2)//3
                        b=(k-d1-d2)//3
                        c=(k-d1+2*d2)//3
                        if a<0 or b<0 or c<0:
                            continue
                    #a队比b队少d1分，b队比c队多d2分
                    if flag=='3':
                        if (k-2*d1+d2)%3!=0 or (k+d1+d2)%3!=0 or (k+d1-2*d2)%3!=0:
                            continue
                        a=(k-2*d1+d2)//3
                        b=(k+d1+d2)//3
                        c=(k+d1-2*d2)//3
                        if a<0 or b<0 or c<0:
                            continue
                    #a队比b队少d1分，c队比b队多d2分
                    if flag=='4':
                        if (k-2*d1-d2)%3!=0 or (k+d1-d2)%3!=0 or (k+d1+2*d2)%3!=0:
                            print('no')
                            break
                        a=(k-2*d1-d2)//3
                        b=(k+d1-d2)//3
                        c=(k+d1+2*d2)//3
                        if a<0 or b<0 or c<0:
                            print('no')
#                             print('bb')
                            break

                    #后面的计算
                    rest=n-k
                    max_=max(a,b,c)
                    lst=[max_-i for i in [a,b,c]]
                    sum_=sum(lst)
#                     print(i+1,flag,a,b,c,sum_,rest)
                    if rest>sum_:
                        r=rest-sum_
                        if r%3==0:
#                             print(i+1,r)
                            print('yes')
                            break
                        else:
                            if flag=='4':
                                print('no')
                                break
                            else:
                                continue
                    elif rest==sum_:
                        print('yes')
                        break
                    else:
                        if flag=='4':
                            print('no')
                            break
                        else:
                            continue
    except:
        break