[ABC309A~D Python]デンソークリエイトプログラミングコンテスト2023（AtCoder Beginner Contest 309）

A問題

A - Nine

# 入力A,B
A, B = map(int, input().split())
# A=3,B=4 A=6,B=7の組み合わせの場合は条件を満たさない
if A == 3 and B == 4 or A == 6 and B == 7:
    print("No")
# 他の組み合わせでB-A=1の場合は条件を満たす
elif B - A == 1:
    print("Yes")
# それ以外の組み合わせは条件を満たさない
else:
    print("No")

基本的にはB-A=1の時マスが左右に隣接してます。
ただ、A,Bが行をまたぐ時条件を満たさないときもあります。

B問題

B - Rotate

# 入力N
N = int(input())
# 入力A
A = [list(input()) for _ in range(N)]
# ずらした後のマス目
movedA = [["" for __ in range(N)] for _ in range(N)]
# 全てのマス目について
for i in range(N):
    for j in range(N):
        # 右上の場合は下向き
        if i == 0 and j == N-1:
            movedA[i+1][j] = A[i][j]
        # 上辺は右向き
        elif i == 0:
            movedA[i][j+1] = A[i][j]
        # 右下の場合は左向き
        elif i == N-1 and j == N-1:
            movedA[i][j-1] = A[i][j]
        # 右辺は下向き
        elif j == N-1:
            movedA[i+1][j] = A[i][j]
        # 左下の場合は上向き
        elif i == N-1 and j == 0:
            movedA[i-1][j] = A[i][j]
        # 下辺は左向き
        elif i == N-1:
            movedA[i][j-1] = A[i][j]
        # 左辺は上向き
        elif j == 0:
            movedA[i-1][j] = A[i][j]
        # それ以外はそのまま
        else:
            movedA[i][j] = A[i][j]
# 1行ずつ区切りなしで出力
for row in movedA:
    print("".join(row))

ずらした後のマス目を新たに作ります。
時計回りにずらす作業ですが、
角については注意が必要です。

C問題

C - Medicine

# 入力N,K
N, K = map(int, input().split())
# 1日目に飲む薬の合計
k = 0
# aとbを保存しておく
# 処方された日を追加する
d = [[0, 0]]
for i in range(N):
    # 入力a,b
    a, b = map(int, input().split())
    # 合計にプラス
    k += b
    # dに保存
    d.append([a, b])
# dをソートする
d.sort()

for i in d:
    # 次の日に飲み終わる薬の錠剤分引くと、
    # K錠になる場合は、
    # 次の日の日付を出力して終了
    if k-i[1] <=K:
        print(i[0]+1)
        break
    # 実際に錠剤分引く
    else:
        k -= i[1]

薬の期間が切れれば、その薬の錠剤分1日に飲む錠剤が少なくなります。
ですので、1日目から見ていき、
錠剤がK以下になったら日付を出力します

D問題

D - Add One Edge

# インポート
from collections import deque, defaultdict
# 入力
N1, N2, M = map(int, input().split())
# 既に訪れた頂点を保存する
visited = [False for _ in range(N1+N2)]
# 辺の情報を保存する
side = defaultdict(list)
# 頂点1もしくは頂点N1+N2との距離
dis = [0 for _ in range(N1+N2)]

# 辺の保存
for i in range(M):
    a, b = map(int, input().split())
    a -= 1
    b -= 1
    side[a].append(b)
    side[b].append(a)

# 頂点1から頂点N1までの全ての頂点について、
# 頂点1または頂点N1+N2からの距離を調べる

# 次に訪れる予定の頂点のそこまでの距離を保存するdeque
d = deque()
# 頂点1を追加
d.append((0, 0))
# 頂点N1+N2を追加
d.append((N1+N2-1, 0))
# 訪れる頂点がなくなるまで
while d:
    D = d.popleft()
    # 今いる頂点
    now = D[0]
    # これまでの距離
    l = D[1]
    # nowがまだ未到達だったら
    if not visited[now]:
        # nowを到達にする
        visited[now] = True
        # 距離を1追加する
        l += 1
        # 辺を伝って頂点を移動する
        for next in side[now]:
            # nextが未到達だったら
            if not visited[next]:
                # dequeに追加する
                d.append((next, l))
                # まだ頂点1からの距離が0だったら、
                # 距離を更新する
                if dis[next] == 0:
                    dis[next] = l
                # 距離はその頂点に行くまでの最短距離になる
                else:
                    dis[next] = min(dis[next], l)
# 頂点1からの距離が最大の点までの距離と
# 頂点N1+N2からの距離が最大の点までの距離と
# 操作によって結ぶ辺の和         
print(max(dis[:N1])+max(dis[N1:N1+N2])+1)

DFSやBFSを使用して、解いていきます。

頂点1または頂点N1+N2からの距離をたどっていきます。
それぞれの連結成分について、
最大距離と新たにつなげる辺の合計を最後に出力します。