【赛码网 牛客网】笔试 --＞ 输入输出写法 python版本

2020秋招陆续开始了。。。
找研发岗工作的学生党也陆续开始做各大公司的笔试题了。。。
然而曾经在力扣牛客等网站刷题时，我从未考虑过代码的输入与输出问题。。。
在首次笔试遇到白板题，代码写对了，但是输入输出不知如何写，真的是坑爹啊，特此记录一下。。。

这里以一例题为例，题目来源于赛码网上一道题
题目

题目描述

大学的同学来自全国各地，对于远离家乡步入陌生大学校园的大一新生来说，碰到老乡是多么激动的一件事，于是大家都热衷于问身边的同学是否与自己同乡，来自新疆的小赛尤其热衷。但是大家都不告诉小赛他们来自哪里，只是说与谁是不是同乡，从所给的信息中，你能告诉小赛有多少人确定是她的同乡吗？

输入描述

包含多组测试用例，对于每组测试用例：
第一行包括2个整数，N（1 <= N <= 1000），M(0 <= M <= N*(N-1)/2)，代表现有N个人（用1~N编号）和M组关系；
在接下来的M行里，每行包括3个整数，a，b, c，如果c为1，则代表a跟b是同乡；如果c为0，则代表a跟b不是同乡；
已知1表示小赛本人。

输出描述

对于每组测试实例，输出一个整数，代表确定是小赛同乡的人数

输入样例

3 1
2 3 1
5 4
1 2 1
3 4 0
2 5 1
3 2 1

样例输出

0
3

具体代码

def run(res):
    visited = set()
    visited.add(1)
    res.sort()
    # 将与1有关系的，即与小赛是老乡的，则记录在集合里，以免重复记录
    for i in res:
        if i[0] in visited:
            visited.add(i[1])
    # 集合的长度减去小赛本人则为他的老乡的人数
    return (len(visited) - 1)
def main():
    # while True的作用是连续输入多个测试样例，若题目没要求连续输入多个测试样例，则不需要使用while True
    while True:
        res = []
        N, M = map(int, input().split())
        for i in range(M):
            a, b, c = map(int, input().split())
            # a, b若为老乡，则记录下来
            if c == 1:
                res.append([a,b] if a < b else [b,a])
        result = run(res)
        print(result)
    return 
main()

输入输出不同点

1.这里的输入为自己键盘敲上去，当然你在本地测试需要自己敲，在网站上测试则不需要自己敲，
python3为input()，python2为raw_input()

2.若一行要求输入一个数字时，直接使用int(input())即可
若一行要求输入多个数字时，使用map(int, input().split())，分割符用split()表示，默认为所有的空字符，包括空格、换行(\n)、制表符(\t)等，若分割符为逗号，则为split(’,’)

3.若题目要求输入的是列表，则可以使用先定义个空列表，然后利用append将数据输入进去
或者可以直接nums = list(map(int, input().split()))

4.输出则必须使用print函数将答案输出出来，直接写return ,则不会返回答案

5.若自己定义函数，则最后切勿忘记调用该函数，
否则该函数不会被调用

6.若输出为列表，要求返回列表里的具体值，
则可以使用for循环将答案全部输出出来

先记录这些，后续遇到坑，再补充。。。