Ruudukko

#include <bits/stdc++.h>
using namespace std;
#define FOR(i, n) for(decltype(n) i=0;i<(n);++i)
#define FORI(i, s, n) for(decltype(n) i=(s);i<(n);++i)
#define BE(t) t.begin(), t.end()
#define RBE(t) t.rbegin(), t.rend()
#define evn(n) (!((n) & 1))
#define PB push_back
#define MP(f, s) make_pair((f), (s))
#define F first
#define S second
#define pmod(a, m) ((a)%(m)+(m))%(m)
typedef unsigned long long ull;
typedef long long ll;
#define MOD ((int)1e9+7)
#define CEILDIV(a, b) (((a)-1)/(b)+1)

int n;
ull yht = 0;
vector<vector<int>> ruu, rei;
vector<int> pysty, vaaka, uv, up;

typedef pair<int,pair<int,int>> ppi;
priority_queue<ppi, vector<ppi>, greater<ppi>> q;

void print_rei(){
	cout << '\n';
	cout << "vaaka: ";
	for(int i : vaaka){
		cout << i << ' ';
	}
	cout << '\n';
	cout << "pysty: ";
	for(int i : pysty){
		cout << i << ' ';
	}
	cout << '\n';
	FOR(y, n){
		FOR(x, n){
			cout << rei[y][x] << ' ';
		}
		cout << '\n';
	}
}

int main(){
	cin >> n;

	ruu.resize(n,   vector<int>(n));
	rei.resize(n+1, vector<int>(n));

	vaaka.resize(n);
	pysty.resize(n);

	FOR(y, n){
		FOR(x, n){
			int l;
			cin >> l;
			ruu[y][x] = l;
			if(ruu[y][x] == 1){
				rei[y][x] = 1;
				pysty[x] += 1;
				vaaka[y] += 1;
			} else {
				q.push(MP(l, MP(y, x)));
			}
		}
	}

	uv = vaaka;
	up = pysty;

	int lv;
	if(q.size())
		lv = q.top().F;
	else
		return 5;

	//print_rei();
	while(q.size()){
		auto p = q.top();
		int value = p.F;
		q.pop();

		if(value != lv){
			lv = value;
			pysty = up;
			vaaka = uv;
			//print_rei();
		}

		auto coord = p.S;
		int y = coord.F, x = coord.S;

		int av = (vaaka[y]+pysty[x]+1) % MOD;
		rei[y][x] = av;

		uv[y] += av;
		uv[y] %= MOD;

		up[x] += av;
		up[x] %= MOD;
	}
	//print_rei();

	FOR(y, n){
		FOR(x, n){
			yht += rei[y][x];
			yht %= MOD;
		}
	}

	cout << yht << '\n';
}

Test 1

Group: 1, 2, 3

Verdict: RUNTIME ERROR

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 2, 3

Verdict: RUNTIME ERROR

Test 5

Group: 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 3

Verdict: RUNTIME ERROR

Test 8

Group: 3

Verdict: ACCEPTED

Test 9

Group: 3

Verdict: ACCEPTED