20-nuc-oj-sq的日期问题

sq的日期问题

问题描述

sq觉得大家一会就写完签到题了一定很无聊，所以决定来个题给大家消磨一下时间。题目很简单，就是计算两个日期之间星期一到星期日各有多少天。例2020-12-5和2020-12-7之间有1个星期六，1个星期日，1个星期一。

输入描述

首先输入一行代表第一个日期，格式为year1 month1 day1。

第二行也输入一行代表第二个日期，格式为year2 month2 day2。

输出描述

输出两个日期之间星期一到星期日各有多少天。

    输出七行，第i(1<=i<=7)行代表星期i有多少天。

样例输入

2020 12 4
2020 12 5

样例输出

0
0
0
0
1
1
0

提示

已知2020年12月13日是星期日。日期均是现实世界中合法的日期。请考虑闰年所带来的影响。

闰年是能够被400整除或者被4整除且不被100整除的年份。

1000 <= year <= 9999。

#include<iostream>
#include<algorithm>
#include<set>
#include<vector>
#include<map>
#include<cstring>
#include<cmath>
#include<queue>
#include<stack>
using namespace std;
const int N = 1000*12*35;
int ye1 = 0, m1, d1, y2, m2, d2;
int month[13] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
int year[10005][13][35];
int vis[10];
void sw()
{
	swap(ye1, y2);
	swap(m1, m2);
	swap(d1, d2);
}

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

void init() // 咳，在这里我看了一眼日历，得知2000-01-01是周六
{
	int t = 6; // 周六
	for(int i=2000; i<10000; i++)
	{	
		if(run(i))
		{
			month[2] = 29;
		}
		else
		{
			month[2] = 28;
		}
		for(int j =1; j <=12; j++)
		{
		
			for(int k=1; k<=month[j]; k++)
			{
				year[i][j][k] = t;
				t++;
				if(t == 8)
				{
					t = 1;
				}
			}
		}
	}	
	t = 5; // 倒序，所以这里是周五
	for(int i=1999; i>=1000; i--)
	{
		if(run(i))
		{
			month[2] = 29;
		}
		else
		{
			month[2] = 28;
		}
		for(int j = 12; j>=1; j--)
		{
			for(int k=month[j]; k>=1; k--)
			{
				year[i][j][k] = t;
				t--;
				if(t == 0)
				{
					t = 7;
				}
			}
		}
	}
}

int main()
{
	scanf("%d%d%d", &ye1, &m1, &d1);
	scanf("%d%d%d", &y2, &m2, &d2);	
	if(ye1 > y2) 
	{
		sw();
	}
	else if(ye1 == y2 && m1 > m2)
	{
		sw();
	}
	else if(ye1 == y2 && m2 == m1 && d1 > d2)
	{
		sw();
	}
	int ans = 0, cnt = 0, t = 6;
	init();
	
	for(int i=ye1; i<=y2-1; i++) // 计算小年到大年-1年之间的时间
	{
		if(run(i))
		{
			month[2] = 29;
		}
		else
		{
			month[2] = 28;
		}
		for(int j=1; j <=12; j++)
		{
			for(int k=1; k<=month[j]; k++)
			{
				vis[year[i][j][k]]++;
			}
		}
	}		
	for(int i=ye1; i<=ye1; i++) // 计算小年从01-01开始到输入日期之间的时间
	{
		if(run(i))
		{
			month[2] = 29;
		}
		else
		{
			month[2] = 28;
		}
		for(int j=1; j <=m1; j++)
		{
			for(int k=1; k<=month[j]; k++)
			{
				if(i == ye1 && j == m1 && d1 == k)
				{
					break;
				}
				vis[year[i][j][k]]--;
			}
		}
	}
	for(int i=y2; i<=y2; i++) // 计算大年从01-01开始至当天的时间
	{
		if(run(i))
		{
			month[2] = 29;
		}
		else
		{
			month[2] = 28;
		}
		for(int j=1; j <=m2; j++)
		{
			for(int k=1; k<=month[j]; k++)
			{
				if(i == y2 && j == m2 && d2 == k)
				{
					vis[year[i][j][k]]++;
					break;
				}
				vis[year[i][j][k]]++;
			}
		}
	}				
	for(int i=1; i<=7; i++)
	{
		printf("%d\n", vis[i]);
	}
	return 0;
}

一道模拟题，据说只有一个坑，就是输入日期的大小问题，没踩到，思路就是把1000-01-01 – 9999-12-31的周几全部模拟出来，然后算就完了