CSES - Datatähti 2023 alku - Results
Submission details
Task:Ruudukko
Sender:qanpi
Submission time:2022-11-13 12:53:39 +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.37 s3details
#8ACCEPTED0.63 s3details
#9ACCEPTED0.91 s3details

Code

#include<iostream>
//#include<chrono>
#include<algorithm>

using namespace std;
const int MOD = 1000000007;

int n;
int board[1000][1000];
bool visited[1000][1000];
int total_sum;

pair<int, int> sorted_rows[1000][1000];
pair<int, int> sorted_cols[1000][1000];

int tree_rows[1000][1000];
int tree_cols[1000][1000];

void recurse(int i, int j); 
int recursive_calls = 0;

int main() {
	cin.tie(0);
	//freopen("grid1000.txt", "r", stdin);
	cin >> n;

	for (int i=0; i<n; i++) {
		for (int j=0; j<n; j++) {
			int val;
			cin >> val;
			board[i][j] = val;
			sorted_rows[i][j] = {val, j};
			sorted_cols[j][i] = {val, i};
		}
	}

	//auto start = chrono::high_resolution_clock::now();
	for(int i=0; i<n; i++) {
		sort(sorted_rows[i], sorted_rows[i]+n);
		sort(sorted_cols[i], sorted_cols[i]+n);
	}


	//for future, maybe somehow sort the rows and cols and go from there
	for (int i=0; i<n; i++) {
		for (int j=0; j<n; j++) {
			recurse(i, j);
		}
	}

	//auto end = chrono::high_resolution_clock::now();
	//auto duration = chrono::duration_cast<chrono::microseconds>(end-start);

	cout << total_sum << endl;
	//cout << "calls: " << recursive_calls << endl;
	//cout << "dur: " << duration.count() << endl;
}

int sum_rows(int i, int k) {
	int s = 0;
	while (k >= 1) {
		s += tree_rows[i][k];
		s %= MOD;
		k -= k&-k;
	}
	return s;
}

int sum_cols(int j, int k) {
	int s = 0;
	while (k >= 1) {
		s += tree_cols[j][k];
		s %= MOD;
		k -= k&-k;
	}
	return s;
}

void add_rows(int i, int k, int x) {
	while(k <= n) {
		tree_rows[i][k] += x;
		tree_rows[i][k] %= MOD;
		k += k&-k;
	}
}

void add_cols(int j, int k, int x) {
	while(k <= n) {
		tree_cols[j][k] += x;
		tree_cols[j][k] %= MOD;
		k += k&-k;
	}
}
void recurse(int i, int j) {
	if (visited[i][j]) return;
	visited[i][j] = true;

	int current = board[i][j];
	pair<int, int> key = {current, 0};
	auto x_hi = lower_bound(sorted_rows[i], sorted_rows[i]+n, key) - sorted_rows[i];
	auto y_hi = lower_bound(sorted_cols[j], sorted_cols[j]+n, key) - sorted_cols[j];

	//if (x_hi == 0 && y_hi == 0) {
	//	add(tree_rows[i], 1, 1);
	//	add(tree_cols[j], 1, 1);
	//	return;
	//}
	//cout << current << " " << x_hi << " " << y_hi << endl;
	//cout << sorted_rows[i][x_hi-1].first << endl; 
	//cout << sorted_cols[j][y_hi-1].first << endl;

	pair<int, int> x_key = {sorted_rows[i][x_hi-1].first, 0};
	pair<int, int> y_key = {sorted_cols[j][y_hi-1].first, 0};
	auto x_low = lower_bound(sorted_rows[i], sorted_rows[i]+n, x_key) - sorted_rows[i]; 
	auto y_low = lower_bound(sorted_cols[j], sorted_cols[j]+n, y_key) - sorted_cols[j]; 


	for(int x = x_low; x<x_hi; x++) {
		recursive_calls++;
		int next = sorted_rows[i][x].second;
		//cout << current << " next x: " << next << endl;
		recurse(i, next);
	}

	for(int y = y_low; y<y_hi; y++) {
		recursive_calls++;
		int next = sorted_cols[j][y].second;
		//cout << current << " next y: " << next << endl;
		recurse(next, j);
	}

	int x_sum = sum_rows(i, x_hi);
	int y_sum = sum_cols(j, y_hi);
	int sum = (x_sum + y_sum) % MOD + 1;

	add_rows(i, x_hi + 1, sum);
	add_cols(j, y_hi + 1, sum);
	
	total_sum += sum;
	total_sum %= MOD;
//	cout << x_hi << " " << y_hi << endl;
//
//	int x_sum = memo_rows[i][x_hi-1];
//	cout << "res x: " << x_sum << endl;
//	int y_sum = memo_cols[j][y_hi-1];
//	cout << "res y: " << y_sum << endl;
//	int sum = x_sum + y_sum + 1;
//
//	memo_rows[i][x_hi] += x_sum + sum;
//	memo_cols[j][y_hi] += y_sum + sum;
//	cout << "res : " << x_sum + y_sum + 1 << endl;
	

}

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