ZOJ - 3950 How Many Nines 年份大模拟

大模拟

大体思路：

给定 y1:m1:d1 ~ y2:m2:d2，

可以分为三个阶段，分别计算：

① y1+1 ~ y2-1  这其中的整年份可以直接算出来：打表预处理

② y1:m1:d1 ~ y1年年末；  多少个月，多少天，分别考虑

③ y2年年初 ~ y2:m2:d2； ==

闰年的关系处理清楚就行了；

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<set>
#include<queue>
#include<stack>
#include<map>
#define PI acos(-1.0)
#define in freopen("in.txt", "r", stdin)
#define out freopen("out.txt", "w", stdout)
#define kuaidian ios::sync_with_stdio(0);

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 100 + 7, maxd = 1e8;
const ll mod = 1e9 + 7;
const int INF = 0x7f7f7f7f;


int n1[15] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
int n2[15] = {0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
int sum[10000];

bool is_run(int n) {
    if( (n % 4 == 0 && n % 100 != 0) || (n % 400 == 0)) return true;
    else return false;
}

int num_(int n) {
    int cnt = 0;
    while(n) {
        int t = n % 10;
        if(t == 9) cnt++;
        n /= 10;
    }
    return cnt;
}

void init() {
    sum[1999] = 0;
    for (int i = 2000; i <= 9999; i ++) {
        if (is_run(i)) {
            sum[i] = sum[i-1] + ( 36 + 366*num_(i) + 30);
        } else {
            sum[i] = sum[i-1] + (35 + 365*num_(i) + 30);
        }
    }
}


int main() {

    int n;
    int y1, m1, d1, y2, m2, d2;
    init();

    scanf("%d", &n);
    for(int ii = 0; ii < n; ++ii) {
        int ans = 0;
        scanf("%d %d %d %d %d %d", &y1, &m1, &d1, &y2, &m2, &d2);
        ans += max(0, (sum[y2-1] - sum[y1]));
        int tt1 = 0, tt2 = 0;
        if(is_run(y1)) {
            int t1 = num_(y1);
            int t2 = 0;
            for(int i = 1; i < m1; ++i)
                t2 += (n2[i]);
            int t3 = (366 - t2 - d1+1) * t1; // 年 贡献
            int t4 = (12 - m1)*3;
            int t5 = 0;
            for(int i = d1; i <= 30; ++i) { // 日 贡献
                if(i % 10 == 9) t5 ++;
            }
            int t6 = 0;
            if(m1 == 9) {
                t6 += (30 - d1 + 1);
            } else if(m1 < 9) {
                t6 += (30);
            }
            tt1 = (t3 + t4+t5 + t6);
        } else {
            int t1 = num_(y1);
            int t2 = 0;
            for(int i = 1; i < m1; ++i)
                t2 += (n1[i]);
            int t3 = (365 - t2 - d1+1) * t1; // 年 贡献
            int t4 = (12 - m1)*3;
            int t5 = 0;
            for(int i = d1; i <= 30; ++i) { // 日 贡献
                if(i % 10 == 9) t5 ++;
            }
            if(m1 <= 2) t5--;

            int t6 = 0;
            if(m1 == 9) {
                t6 += (30 - d1 + 1); // 月 贡献
            } else if(m1 < 9) {
                t6 += (30);
            }
            tt1 = (t3 + t4+t5 + t6);
        }

        if(is_run(y2)) {
            int t1 = num_(y2);
            int t2 = 0;
            for(int i = 1; i < m2; ++i)
                t2 += (n2[i]);
            int t3 = (t2 + d2) * t1;  // 年

            int t4 = (m2 - 1) * 3;
            int t5 = 0;
            for(int i = 1; i <= d2; ++i)
                if(i % 10 == 9) t5 ++;

            int t6 = 0;
            if(m2 > 9) {
                t6 += 30;
            } else if(m2 == 9) {
                t6 += (d2);
            }
            tt2 = (t3 + t4+t5 + t6);
        } else {
            int t1 = num_(y2);
            int t2 = 0;
            for(int i = 1; i < m2; ++i)
                t2 += (n1[i]);
            int t3 = (t2 + d2) * t1;  // 年

            int t4 = (m2 - 1) * 3;
            int t5 = 0;
            for(int i = 1; i <= d2; ++i)
                if(i % 10 == 9) t5 ++;
            if(m2 > 2) {
                t5--;
            }
            int t6 = 0;
            if(m2 > 9) {
                t6 += 30;
            } else if(m2 == 9) {
                t6 += (d2);
            }
            tt2 = (t3 + t4+t5 + t6);
        }

        if(y1 == y2) {

            int tt3 = 0;
            if(is_run(y1)) {
                tt3 = ( 36 + 366*num_(y1) + 30);
            } else tt3 = ( 35 + 365*num_(y1) + 30);

            ans += (tt2 + tt1 - tt3);
        } else {
            ans += (tt1 + tt2);
        }
        printf("%d\n", ans);
    }

    return 0;
}