nyoj 312 20岁生日

20岁生日

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 1

描述

路过这的20岁生日就要到了，他当然很开心，可是他突然想到一个问题，是不是每个人从出生开始，到达20岁生日时所经过的天数都是一样的呢？似乎并不全都是这样，所以他想请你帮忙计算一下他和他的几个朋友从出生到达20岁生日所经过的总天数，让他好来比较一下。

输入

一个数T，后面T行每行有一个日期，格式是YYYY-MM-DD。如我的生日是1988-03-07。

输出

T行，每行一个数，表示此人从出生到20岁生日所经过的天数。如果这个人没有20岁生日，就输出-1。

样例输入

1
1988-03-07

样例输出

7305

超时代码

 
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f
using namespace std;

int a[20]={0,31,0,31,30,31,30,31,31,30,31,30,31};

int rr(int n){
    if((n%4==0&&n%100!=0)||n%400==0)
        return 1;
    return 0;
}

int yy(int y,int m){
    if(m!=2)
        return a[m];
    if(rr(y))
        return 29;
    return 28;
}

int main(){
    int y,m,d,t,s,sum,p,q;
    scanf("%d",&t);
    while(t--){
        s=0;
        sum=0;
        scanf("%d-%d-%d",&y,&m,&d);
        if(m==2&&d==29&&!rr(y+20)){
            printf("-1\n");
            continue;
        }
        p=m;
        q=d;
        while(1){
            if(s==20) break;
            d+=1;
            sum+=1;
            if(d>yy(y,m)){
                d=1;
                m+=1;
                if(m>12){
                    m=1;
                    y+=1;
                }
            }
            if(m==p&&d==q) s+=1;
        }
        printf("%d\n",sum);
    }
    return 0;
}
        

ac 代码

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f
using namespace std;

int a[20]={0,31,0,31,30,31,30,31,31,30,31,30,31};

int rr(int n){
    if((n%4==0&&n%100!=0)||n%400==0)
        return 1;
    return 0;
}

int yy(int y,int m){
    if(m!=2)
        return a[m];
    if(rr(y))
        return 29;
    return 28;
}

int main(){
    int i,y,m,d,t,s,sum,p,q;
    scanf("%d",&t);
    while(t--){
        sum=0;
        scanf("%d-%d-%d",&y,&m,&d);
        if(m==2&&d==29&&!rr(y+20)){
            printf("-1\n");
            continue;
        }

        sum+=yy(y,m)-d;
        for(i=m+1;i<=12;i++){
            sum+=yy(y,i);
        }

        for(i=y+1;i<y+20;i++){
            if(rr(i)) sum+=366;
            else sum+=365;
        }

        for(i=1;i<m;i++)
            sum+=yy(y+20,i);

        sum+=d;
        printf("%d\n",sum);
    }
    return 0;
}