UVa 10855 - Rotated square Solution

 #include <bits/stdc++.h>
using namespace std;
int n,m;
vector<vector<char> > a,b;
void rotate(){
    vector<vector<char> > c(m,vector<char>(m));
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++)
            c[j][m-i-1]=b[i][j];
    }
    b=c;
}
int compare(int ii,int jj){
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            if(b[i][j]!=a[ii+i][jj+j])return 0;
        }
    }
    return 1;
}
int count(){
    int ans=0;
    for(int i=0;i<n-m+1;i++){
        for(int j=0;j<n-m+1;j++){
            if(compare(i,j))ans++;
        }
    }
    return ans;
}
int main(){
    ios::sync_with_stdio(0);
    while(cin>>n>>m){
        if(n==0&&m==0)return 0;
        a.assign(n,vector<char>(n));
        b.assign(m,vector<char>(m));
        cin>>ws;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                cin>>a[i][j];
            }
            cin>>ws;
        }
        for(int i=0;i<m;i++){
            for(int j=0;j<m;j++){
                cin>>b[i][j];
            }
            cin>>ws;
        }
        cout<<count()<<' ';
        rotate();
        cout<<count()<<' ';
        rotate();
        cout<<count()<<' ';
        rotate();
        cout<<count()<<endl;
    }
return 0;
}

Comments