SRM Div2-145 200score

div2-145-200.cpp

// {{{ Boilerplate Code <--------------------------------------------------
//
// vim:filetype=cpp foldmethod=marker foldmarker={{{,}}}

#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <deque>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <utility>
#include <vector>

#define FOR(I,A,B)	for(int I = (A); I < (B); ++I)
#define REP(I,N)	FOR(I,0,N)
#define ALL(A)		(A).begin(), (A).end()

using namespace std;

// }}}

class ImageDithering
{
public:
    int count(string dithered, vector<string> screen)
    {
        int sum = 0;
        for (vector<string>::iterator str = screen.begin(); str != screen.end(); ++str) {
            for (string::iterator pos = str->begin(); pos != str->end(); ++pos) {
                for (string::iterator d = dithered.begin(); d != dithered.end(); ++d) {
                    if (*pos != *d) continue;
                    ++sum;
                }
            }
        }
        return sum;
    }
};

int main() {
    ImageDithering image;
    string screen1[] = {"AAAAAAAA",
        "ABWBWBWA",
        "AWBWBWBA",
        "ABWBWBWA",
        "AWBWBWBA",
        "AAAAAAAA"};
    vector<string> v(6);
    v.assign(screen1, screen1 + 6);
    cout << image.count("BW", v) << endl;
    return 0;
}