ZOJ 3326 An Awful Problem

最新推荐文章于 2017-04-16 16:59:28 发布

原创最新推荐文章于 2017-04-16 16:59:28 发布 · 441 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法专栏收录该内容

338 篇文章

订阅专栏

本文提供ZOJ3326问题的解决方案，通过枚举法解决最大80万数据规模的问题，并使用C++实现质数判断及日期遍历逻辑。

题目链接：ZOJ 3326 An Awful Problem

最大八十来万的数据量，枚举就可以了。

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int MAX_N = 3000 + 100;

bool primes[MAX_N];

//primes[i] = 0表示i是素数，为1表示i不是素数
void get_primes()
{
    memset(primes,0,sizeof primes);
    primes[0]=primes[1]=1;
    for(int i=2;i<MAX_N;i++)
    {
        if(!primes[i])
        {
            for(int j=i+i;j<MAX_N;j+=i)
            {
                primes[j]=1;
            }
        }
    }
}

int main()
{
    //freopen("in.txt", "r", stdin);
    get_primes();
    int T;
    scanf("%d", &T);
    while(T--)
    {
        int ya, ma, da, yb, mb, db;
        scanf("%d%d%d%d%d%d", &ya, &ma, &da, &yb, &mb, &db);
        int res = 0;
        int m = ma, d = da, y = ya;
        while(y <= yb)
        {
            if(y == yb && m == mb && d > db)
                break;
            if(y == yb && m > mb)
                break;
            if(primes[d] == 0 && primes[m] == 0)
                res++;
            if(((y % 4) == 0 && (y % 100) != 0) || (y % 400) == 0)
            {
                if(m == 2 && d == 28)
                {
                    d = 29;
                    continue;
                }
                if(m == 2 && d == 29)
                {
                    d = 1;
                    m++;
                    continue;
                }
                if(m == 1 || m == 3 || m == 5 || m == 7 || m == 10 || m == 12)
                {
                    if(d == 30)
                    {
                        d++;
                        continue;
                    }
                    if(d == 31)
                    {
                        if(m == 12)
                        {
                            d = 1;
                            m = 1;
                            y++;
                        }
                        else
                        {
                            d = 1;
                            m++;
                        }
                        continue;
                    }
                }
                if(d == 30)
                {
                    d = 1;
                    m++;
                    continue;
                }
                d++;
            }
            else
            {
                if(m == 2 && d == 28)
                {
                    d = 1;
                    m++;
                    continue;
                }
                if(m == 2 && d == 28)
                {
                    d = 1;
                    m++;
                    continue;
                }
                if(m == 1 || m == 3 || m == 5 || m == 7 || m == 10 || m == 12)
                {
                    if(d == 30)
                    {
                        d++;
                        continue;
                    }
                    if(d == 31)
                    {
                        if(m == 12)
                        {
                            d = 1;
                            m = 1;
                            y++;
                        }
                        else
                        {
                            d = 1;
                            m++;
                        }
                        continue;
                    }
                }
                if(d == 30)
                {
                    d = 1;
                    m++;
                    continue;
                }
                d++;
            }
        }
        printf("%d\n", res);
    }
    return 0;
}