ZOJ-3939 The Lucky Week

ZOJ-3939 The Lucky Week

思路：读完题发现数据量好大，没法纯粹用日期增加来模拟，就想到一年内满足要求的日期个数是否有规律，所以打表，判断某个日期是星期几可以用公式来进行（感谢能带模板，不然公式死都记不住），第一次打出1753之后的2000年，暴力跑出循环长度是400，之后再打出更长的数据，发现循环长度就是400。于是开始敲代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<stack>
#include<set>
#include<map>
using namespace std;
#define ll long long 
#define mod 998244353
#define eps 1e-6
double pi=acos(-1);
const ll INF=1e12;
int CaculateWeekDay(int y,int m, int d)
{//判断y-m-d是星期几，0-6，星期一-星期天 
    if(m==1||m==2) {
        m+=12;
        y--; 
    } 
    int week=(d+2*m+3*(m+1)/5+y+y/4-y/100+y/400+7)%7;
    return week;
}
int CaculateYears(int y){
//打表出这一年的满足要求的日期的数量 
    int i,j,ans=0;
    for(i=1;i<13;i++){
        for(j=1;j<31;j+=10){
            if(CaculateWeekDay(y,i,j)==0){
                ans++;
            }
        }
    }
    return ans;
}
int CaculateThisYears(int y,int m,int d,int n){
//1.用于计算输入y-m-d开始到y这年结束有几个满足要求的数量
//2.计算并输出答案 
//懒得分开写，凑合着看看就好
    int i,j,ans=0;
    for(j=d;j<31;j+=10){
        if(CaculateWeekDay(y,m,j)==0){
            ans++;
            if(ans==n){
                printf("%d %d %d\n",y,m,j);
                return 0;
            }
        }
    }
    for(i=m+1;i<13;i++){
        for(j=1;j<31;j+=10){
            if(CaculateWeekDay(y,i,j)==0){
                ans++;
                if(ans==n){
                    printf("%d %d %d\n",y,i,j);
                    return 0;
                }
            }
        }
    }
    return ans;
}
int main()
{
    int ans[400],sum[400];//打表记录
    int T,y,m,d,n,i,j;
    for(i=0;i<400;i++){
        ans[i]=CaculateYears(i+1753);
        if(i==0){
            sum[0]=ans[0];
        }
        else{
            sum[i]=sum[i-1]+ans[i];
        }
    }
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d%d",&y,&m,&d,&n);
        int ans1=CaculateThisYears(y,m,d,n);//如果返回0说明已经找到，结束
        if(ans1){
            n-=ans1;//去掉输入日期并在同一年的数量
            y+=n/sum[399]*400;
            n%=sum[399];
            if(n==0){//说明答案在y这年
                ans1=CaculateThisYears(y,1,1,ans[(y-1753)%400]);
                continue;
            }
            y++;//在之后年份中
            while(n>ans[(y-1753)%400]){//多与这一年的总数量，跑到下一年
                n-=ans[(y-1753)%400];
                y++;
            }
            ans1=CaculateThisYears(y,1,1,n);
        }

    }
    return 0;
}