Cod sursa(job #3261371)

Utilizator Silviu643Dumitrescu Silviu Florian Silviu643 Data 5 decembrie 2024 17:20:37
Problema Matrix Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream in("matrix.in");
ofstream out("matrix.out");
long long s1[1005][1005],s2[1005][1005],s1k,s2k;
int x,sol,n,m;
char y;
int main()
{
    in>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
    {
        in>>y;
        x=y-'a'+1;
        s1[i][j]=s1[i-1][j]+s1[i][j-1]-s1[i-1][j-1]+x;
        s2[i][j]=s2[i-1][j]+s2[i][j-1]-s2[i-1][j-1]+x*x;
    }
    for(int i=1;i<=m;i++)
        for(int j=1;j<=m;j++)
        {in>>y;
        x=y-'a'+1;
        s1k+=x;
        s2k+=x*x;
        }
    for(int i=m;i<=n;i++)
        for(int j=m;j<=n;j++)
    {
        if(s1k==s1[i][j]-s1[i-m][j]-s1[i][j-m]+s1[i-m][j-m] &&
           s2k==s2[i][j]-s2[i-m][j]-s2[i][j-m]+s2[i-m][j-m]) sol++;
    }
    cout<<sol;
    return 0;
}