2021-04-01から1ヶ月間の記事一覧

参考になったダイクストラ法

from heapq import heappush, heappopimport mathINF = 10 ** 18def dijkstra(s, n): # (始点, ノード数) dist = [INF] * n hq = [(0, s)] # (distance, node) dist[s] = 0 seen = [False] * n # ノードが確定済みかどうか while hq: v = heappop(hq)[1] # …

ダイクストラ法クラスで!

予めEdgeデータとdis(コスト)を用意してあげる必要があります。import heapqV,E,r = map(int,input().split())dis = [[float('inf')]*V for _ in range(V)]Edge = [[] for _ in range(V)]for _ in range(E): s,t,d = map(int,input().split()) dis[s][t] = d…

Atcoder come back quicklyをサンプルだけとけるやつ

atcoder.jp classでDFSを使って見たのですがReになってしまいました。(サンプルはいけた) 結構更新の考え方に慣れてきてよかったなあと思ったのですが肝心のところでTLEやREが取れない。。なお回答はダイクストラ法だったので頑張って習得します。(今週中…

pythonで機械学習用学習データ作成GUIをつくる!-2

# coding=utf-8from tkinter import filedialogfrom PIL import Imageimport matplotlib.pyplot as pltimport osfld = filedialog.askdirectory()file_path = []#include file_path in appointed directoryfor curDir, dirs, files in os.walk(fld): for fil…

pythonで機械学習用学習データ作成GUIをつくる!-1

まず実行フォルダと同じディレクトリに読み込んだファイルの0番目の画像を、新規ディレクトリ内に保存するコードを書きました! filedialogで選択する画像ディレクトリをゲットします。 次にそれらの画像のファイルパスをゲットし、インデックス番号に応じた…

二分探索の神みたいなコード

def meguru_bisect(ng, ok): ''' 初期値のng,okを受け取り,is_okを満たす最小(最大)のokを返す まずis_okを定義すべし ng ok は とり得る最小の値-1 とり得る最大の値+1 最大最小が逆の場合はよしなにひっくり返す ''' while (abs(ok - ng) > 1): mid = (ok …

Class で DFS

DFSを毎回自力で実装するのがめんどくさいのでAtcoder のKiの問題をclassで実装したのでメモとして残します 以下 利用したコード N,Q = map(int,input().split())Graph = [[]for _ in range(N)]operater = []for _ in range(N-1): a,b = map(int,input().spl…