百度之星初赛2A 1001 ALL X（HDU 5690）

All X

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 889    Accepted Submission(s): 425

Problem Description

F(x,m) 代表一个全是由数字x组成的m位数字。请计算，以下式子是否成立：

F(x,m) mod k ≡ c

 

Input

第一行一个整数T，表示T组数据。
每组测试数据占一行，包含四个数字x,m,k,c

1≤x≤9

1≤m≤1010

0≤c<k≤10,000

 

思路：如xxxxx=x+10*x+100*x+1000*x+10000*x，可见这是一个以10为公比的等比数列，可构造矩阵进行运算

[x,x]*[10,0]

[0,0]  [1 ,1]

 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=;
int x,mod,c;
ll m;

struct Matrix
{
    int a[maxn][maxn];
    Matrix(){memset(a,,sizeof(a));}
    Matrix operator* (const Matrix &p)
    {
        Matrix res;
        for(int i=;i<maxn;i++)
        {
            for(int j=;j<maxn;j++)
            {
                for(int k=;k<maxn;k++)
                {
                    res.a[i][j]+=(a[i][k]*p.a[k][j]%mod);
                }
                res.a[i][j]%=mod;
            }
        }
        return res;
    }
}ans,base;

Matrix quick_pow(Matrix base,ll n)
{
    Matrix res;
    for(int i=;i<maxn;i++)
    {
        res.a[i][i]=;
    }
    while(n)
    {
        if(n&) res=res*base;
        base=base*base;
        n>>=;
    }
    return res;
}

void init_matrix()
{
    ans.a[][]=x;
    ans.a[][]=x;
    ans.a[][]=;
    ans.a[][]=;
    base.a[][]=;
    base.a[][]=;
    base.a[][]=;
    base.a[][]=;
}

int main()
{
    int t,cas=;
    scanf("%d",&t);
    while(t--)
    {
        cas++;
        scanf("%d%lld%d%d",&x,&m,&mod,&c);
        init_matrix();
        ans=ans*quick_pow(base,m-);
        printf("Case #%d:\n",cas);
        if(ans.a[][]==c) printf("Yes\n");
        else printf("No\n");
    }
    return ;
}