Cod sursa(job #2547758)

Utilizator marius004scarlat marius marius004 Data 15 februarie 2020 17:27:48
Problema Matrix Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.44 kb
#include <iostream>
#include <fstream>

std::ifstream f("matrix.in");
std::ofstream g("matrix.out");

const int NMAX = 1005;
char genom[NMAX][NMAX],virus[NMAX][NMAX];
int n,m,s[NMAX][NMAX],fr[30],sol;
bool infectat[NMAX][NMAX];

// infectat[i][j] - daca submatricea care incepe din (i,j) e infectata

int suma(int is,int js,int ij,int jj){
    return s[ij][jj] - s[is - 1][jj] - s[ij][js - 1] + s[is - 1][js - 1];
}

int main(){
    
    f >> n >> m;
    
    for(int i = 1;i <= n;++i){
        for(int j = 1;j <= n;++j){
            f >> genom[i][j];
            infectat[i][j] = true;
        }
    }
    
    for(int i = 1;i <= m;++i){
        for(int j = 1;j <= m;++j){
            f >> virus[i][j];
            fr[ virus[i][j] - 'a']++;
        }
    }
    
    for(char litera = 'a';litera <= 'z';++litera){
        
        for(int i = 1;i <= n;++i)
            for(int j = 1;j <= n;++j)
                s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + (genom[i][j] == litera);
    
        for(int i = 1;i <= n - m + 1;++i){
            for(int j = 1;j <= n - m + 1;++j){
                
                int is = i,js = j,ij = i + m - 1,jj = j + m - 1;
                
                if(suma(is,js,ij,jj) != fr[litera - 'a'])
                    infectat[i][j] = false;
            }
        }
    }
    
    for(int i = 1;i <= n - m + 1;++i)
        for(int j = 1;j <= n - m + 1;++j)
            sol += infectat[i][j];
    
    g << sol;
    
    return 0;
}