#655235 (PyPy3) No.1502 Many Simple Additions

• 問題: No.1502 Many Simple Additions
• ユーザー: 👑 SPD_9X2
• 提出日時: 2021-05-07 22:31:02
• 言語: PyPy3 (7.3.15)
• 結果: AC
• 実行時間: 340 ms / 2,000 ms
• コード長: 2,914 bytes
• コンパイル時間: 362 ms
• コンパイル使用メモリ: 82,252 KB
• 実行使用メモリ: 96,376 KB
• 最終ジャッジ日時: 2024-09-15 18:19:27
• 合計ジャッジ時間: 6,021 ms
• ジャッジサーバーID （参考情報）: judge4 / judge5

ファイルパターン	結果
sample	AC * 5
other	AC * 39

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
"""
まず、
連結成分ごとに考えられる
奇閉路がある場合、連結成分内が一意に決まる
Aが全てK以下の場合から、K-1以下の場合を引けばいい
全てK以下の場合の数を求めたい
"""
import sys
from sys import stdin
from collections import deque
mod = 10**9+7
def solve(LIM):
    state = [0] * N
    plus  = [0] * N
    ans = 1
    for i in range(N):
        maxx = LIM
        minx = 1
        X = None
        if state[i] == 0:
            nset = set([i])
            q = deque([i])
            state[i] = 1
            while q:
                v = q.popleft()
                #xの可能性範囲を狭める
                if state[v] == 1: # 1 <= X+plus[v] <= LIM
                    maxx = min(maxx,LIM-plus[v])
                    minx = max(minx,1-plus[v])
                else: # 1 <= -X+plus[v] <= LIM
                    maxx = min(maxx, plus[v]-1)
                    minx = max(minx,plus[v]-LIM)
                
                for nex,z in lis[v]:
                    
                    if state[nex] == 0: #新たに定義
                        state[nex] = -1 * state[v]
                        plus[nex] = z - plus[v]
                        q.append(nex)
                        
                    elif state[nex] != state[v]: #別側の場合、整合性チェック
                        if plus[v] + plus[nex] != z:
                            return 0
                    else: #同じ側の場合
                        if X == None: #Xが未定義の場合
                            if state[v] == 1: #2x+plus[v]+plus[nex] = z
                                x2 = z - plus[v] - plus[nex]
                                if x2 % 2 == 1 or x2 <= 0:
                                    return 0
                                else:
                                    X = x2 // 2
                            else: #-2x + p+p = z
                                x2 = plus[v] + plus[nex] - z
                                if x2 % 2 == 1 or x2 <= 0:
                                    return 0
                                else:
                                    X = x2 // 2
                        
                        else: #Xが定義済みの場合整合性チェック
                            if state[v] * X + plus[v] + state[nex] * X + plus[nex] != z:
                                return 0
            #可能性を書ける
            if X == None:
                ans *= max(0,maxx-minx+1)
            elif minx <= X <= maxx:
                ans *= 1
            else:
                ans *= 0
            ans %= mod
    return ans
    
N,M,K = map(int,stdin.readline().split())
lis = [ [] for i in range(N)]
for i in range(M):
    X,Y,Z = map(int,stdin.readline().split())
    X -= 1
    Y -= 1
    lis[X].append((Y,Z))
    lis[Y].append((X,Z))
print ((solve(K)-solve(K-1)) % mod)
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX