刷题-1122

1. 蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。

例如，当输入5时，应该输出的三角形为：

1 3 6 10 15

2 5 9 14

4 8 13

7 12

11

import sys
def generate_snake_matrix(n):
    matrix = [[0]*n for _ in range(n)]
    curent_num = 1
    for i in range(n):
        for j in range(i+1):
            matrix[i-j][j] = curent_num
            curent_num += 1
    result = []
    for i in range(n):
        result.append([n_ for n_ in matrix[i] if n_ != 0])
    return result
for line in sys.stdin:
    a = line.split()
    result = generate_snake_matrix(int(a[0]))
    for line in result:
        print(' '.join(map(str, line)))

2. 

根据输入的日期，计算是这一年的第几天。

保证年份为4位数且日期合法。

 输入：2012 12 31

输出：366

import sys


def is_leap_year(year):
    # 判断是否为闰年
    return (year % 4 == 0 and year % 100 != 0) or (year % 400 == 0)


def days_in_month(month, year):
    # 每个月的天数
    days_in_months = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    if is_leap_year(year):
        days_in_months[1] = 29  # 2月在闰年为29天
    return days_in_months[month - 1]


def day_of_year(year, month, day):

    # 计算日期前所有月份的天数之和
    day_of_year = sum(days_in_month(m, year) for m in range(1, month)) + day
    return day_of_year


for line in sys.stdin:
    a = line.split()
    year = int(a[0])
    month = int(a[1])
    day = int(a[2])
    print(day_of_year(year, month, day))