CSES - Datatähti 2023 alku - Results
Submission details
Task:Ruudukko
Sender:Niilo
Submission time:2022-12-06 12:22:47 +0200
Language:C++ (C++11)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED28
#2ACCEPTED33
#3ACCEPTED39
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2, 3details
#2ACCEPTED0.00 s1, 2, 3details
#3ACCEPTED0.00 s1, 2, 3details
#4ACCEPTED0.01 s2, 3details
#5ACCEPTED0.01 s2, 3details
#6ACCEPTED0.01 s2, 3details
#7ACCEPTED0.34 s3details
#8ACCEPTED0.40 s3details
#9ACCEPTED0.48 s3details

Code

#include <iostream>
#include <algorithm>
#include <stack>

using namespace std;

typedef unsigned int intu;

int main() {
	const int MOD = 1e9 + 7;
	// size
	int size;
	cin >> size;
	const int size2 = size * size;
	// order of boxes
	auto *order = new pair<int,pair<int,int>>[size2];
	// input to boxes
	int i;
	int cord = 0;
	for (int y=0; y < size; y++) {
		for (int x=0; x < size; x++) {
			cin >> i;
			order[cord++] = {i, {x, y}};
		}
	}
	// preparation
	intu total_paths = 0;
	intu row1[size] = {};
	intu row2[size] = {};
	sort(order, order + size2);
	// helping stuff
	intu cpath;
	stack<pair<intu,int>> s1;
	stack<pair<intu,int>> s2;
	pair<int,pair<int,int>> p;
	pair<intu,int> s;
	// main check
	for (int i=0; i < size2 - 1; i++) {
		p = order[i];
		cpath = row1[p.second.first] + row2[p.second.second] + 1;
		// same number next -> save new rows
		if (order[i + 1].first == p.first) {
			s1.push({cpath, p.second.first});
			s2.push({cpath, p.second.second});
		}
		// new number next -> update every row
		else {
			row1[p.second.first] = (row1[p.second.first] + cpath) % MOD;
			row2[p.second.second] = (row2[p.second.second] + cpath) % MOD;
			while (!s1.empty()) {
				s = s1.top();
				row1[s.second] = (row1[s.second] + s.first) % MOD;
				s = s2.top();
				row2[s.second] = (row2[s.second] + s.first) % MOD;
				s1.pop();
				s2.pop();
			}
		};
		// add to total count
		total_paths += cpath;
		total_paths %= MOD;
	}
	// last path
	p = order[size2 - 1];
	total_paths += row1[p.second.first] + row2[p.second.second] + 1;
	total_paths %= MOD;
	// result
	cout << total_paths << "\n";
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
1 1 1
1 1 1
1 1 1

correct output
9

user output
9

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
1 2 3
6 5 4
7 8 9

correct output
135

user output
135

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
7 8 1
4 5 4
3 9 6

correct output
57

user output
57

Test 4

Group: 2, 3

Verdict: ACCEPTED

input
100
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
10000

user output
10000

Test 5

Group: 2, 3

Verdict: ACCEPTED

input
100
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
187458477

user output
187458477

Test 6

Group: 2, 3

Verdict: ACCEPTED

input
100
2995 8734 1018 2513 7971 5063 ...

correct output
964692694

user output
964692694

Test 7

Group: 3

Verdict: ACCEPTED

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
1000000

user output
1000000

Test 8

Group: 3

Verdict: ACCEPTED

input
1000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
229147081

user output
229147081

Test 9

Group: 3

Verdict: ACCEPTED

input
1000
520283 805991 492643 75254 527...

correct output
951147313

user output
951147313