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;
}