Cod sursa(job #2443506)

Utilizator RaduXD1Nicolae Radu RaduXD1 Data 28 iulie 2019 13:37:16
Problema 12-Perm Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include<fstream>
#include<cstring>
#define mod 1048576
using namespace std;
ifstream fin("12perm.in");
ofstream fout("12perm.out");
int n,sol,i,j,k;
int a[5][5]={1,0,0,1,0,0,0,0,1,0,0,0,1,0,0,0,0,0,1,1,0,1,1,0,0};
int aux[5][5],b[5][5];
void inm(int a[5][5], int b[5][5])
{
    for(i=0;i<=4;i++)
        for(j=0;j<=4;j++)
        {
            aux[i][j]=0;
            for(k=0;k<=4;k++)
                aux[i][j]+=(1LL*a[i][k]*b[k][j])%mod;
        }
    for(i=0;i<=4;i++)
        for(j=0;j<=4;j++)
            a[i][j]=aux[i][j]%mod;
}
int main(){
    fin>>n;
    if(n==1||n==2){fout<<n; return 0;}
    b[0][0]=2;b[0][3]=2;b[0][4]=2;
    n-=3;
    while(n)
    {
        if(n%2==1) inm(b,a);
        inm(a,a);
        n/=2;
    }
    for(i=0;i<=4;i++) sol+=b[0][i];
    fout<<sol%mod;
    return 0;
}