Acwing每日一题

#include <iostream>
#include <cstring>
#include <algorithm>
#include <set>

using namespace std;
const int N = 10;
char a[N][N];
set<set<char>> ans[2];

void insert(vector<vector<int>> ps){
    set<char> S;
    for(auto p : ps){
        int x = p[0], y = p[1];
        S.insert(a[x][y]);
    }
    if(S.size() == 1)ans[0].insert(S);
    else if(S.size() == 2)ans[1].insert(S);
}


int main()
{
    int n = 3;
    for(int i = 0; i < n; i ++)
            cin >> a[i]; 
    for(int i = 0; i < 3 ; i ++)insert({{i,0},{i,1},{i,2}});
    for(int i = 0; i < 3 ; i ++)insert({{0,i},{1,i},{2,i}});
    insert({{0,0},{1,1},{2,2}});
    insert({{2,0},{1,1},{0,2}});
    cout << ans[0].size() << endl;
    cout << ans[1].size() << endl;
    return 0;
}