#include "vision.h"

#include <bits/stdc++.h>

using namespace std;

void construct_network(int h, int w, int k) {
    int pos = h * w;
    vector<int> hs(h), ws(w);
    for (int i = 0; i < h; ++i) {
        vector<int> ns;
        for (int j = 0; j < w; ++j) {
            ns.push_back(i * w + j);
        }
        add_or(ns);
        hs[i] = pos++;
    }
    for (int i = 0; i < w; ++i) {
        vector<int> ns;
        for (int j = 0; j < h; ++j) {
            ns.push_back(j * w + i);
        }
        add_or(ns);
        ws[i] = pos++;
    }
    vector<int> hd(min(h, k + 1)), wd(min(w, k + 1));
    for (int i = 0; i < (int)hd.size(); ++i) {
        if (i == 0) {
            add_xor(hs);
            hd[i] = pos++;
        } else if (w != 1 || i == k) {
            vector<int> temp;
            for (int j = 0; j + i < h; ++j) {
                add_and({hs[j], hs[j + i]});
                temp.push_back(pos++);
            }
            add_or(temp);
            hd[i] = pos++;
        }
    }
    for (int i = 0; i < (int)wd.size(); ++i) {
        if (i == 0) {
            add_xor(ws);
            wd[i] = pos++;
        } else if (h != 1 || i == k) {
            vector<int> temp;
            for (int j = 0; j + i < w; ++j) {
                add_and({ws[j], ws[j + i]});
                temp.push_back(pos++);
            }
            add_or(temp);
            wd[i] = pos++;
        }
    }
    vector<int> temp;
    for (int i = 0; i < (int)hd.size(); ++i) {
        if (k - i < (int)wd.size()) {
            if (w != 1 && h != 1) {
                add_and({hd[i], wd[k - i]});
                temp.push_back(pos++);
            } else if (i == 0 || i == k) {
                add_and({hd[i], wd[k - i]});
                temp.push_back(pos++);
            }
        }
    }
    add_or(temp);
}