【华为机试】计算日期到天数转换

描述

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

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

进阶：时间复杂度：O(n)，空间复杂度：O(1)。

输入描述：

输入一行，每行空格分割，分别是年，月，日

输出描述：

输出是这一年的第几天

示例1

输入：

2012 12 31

输出：

366

示例2

输入：

1982 3 4

输出：

63

#include <iostream>
using namespace std;

int main()
{
    int date[3] = {0, 0, 0}; //年月日初始化
    //从0月到12月初始化天数数组（为了方便后面的1月计算）
    int days[13] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
    int sum = 0;

    //输入日期按照年月日赋值给数组
    for(int i = 0; i < 3; i++)
        cin >>date[i];
    
    //是闰年且月份>=2则天数加1(百年需要除以400才算闰年)
    if(date[0]%4 == 0 && date[0]%100 != 0 ||  date[0]%400 == 0)
        if(date[1] > 2 || date[1] == 2 && date[2] == 29)//如果是二月则日必须为29才能++
            sum++;
    
    //从第0月到前一个月的总天数（考虑1月特殊情况）
    for(int i = 0; i <= date[1]-1; i++)
        sum += days[i];
    sum += date[2]; //在加上当月天数
    
    cout << sum << endl;
    return 0;
}