コード例 #1
0
def test_run():
    with open("./tests/02", "r") as f:
        n_tables, n_queries = map(int, f.readline().split())
        print('test run results, output in map ./tests')
        counts = list(map(int, f.readline().split()))
        db = Set.Special_Set(counts)

        with open("./tests/02.b", "w") as g:
            for i in range(n_queries):
                dst, src = map(int, f.readline().split())
                # print('c: ', dst, src)
                db.Union(dst - 1, src - 1)
                print(db.max_row_count, file=g)
コード例 #2
0
def main():
    n_tables, n_queries = map(int, input('give input: \n').split())
    counts = list(map(int, input().split()))
    assert len(counts) == n_tables
    database = Set.Special_Set(counts)

    # test files and examples all are 1-index based; convert to 0-index base.
    for i in range(n_queries):
        dst, src = map(int, input().split())
        database.Union(dst - 1, src - 1)
        print(database.max_row_count)

    # run a file in the test directory
    test_run()