CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:_Ahrou
Submission time:2020-10-01 21:04:52 +0300
Language:C++ (C++11)
Status:READY
Result:11
Feedback
groupverdictscore
#1ACCEPTED11
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2, 3details
#2ACCEPTED0.01 s1, 2, 3details
#3ACCEPTED0.01 s1, 2, 3details
#4ACCEPTED0.01 s1, 2, 3details
#5ACCEPTED0.01 s1, 2, 3details
#6--2, 3details
#7ACCEPTED0.01 s2, 3details
#8ACCEPTED0.01 s2, 3details
#9--2, 3details
#10--2, 3details
#11--3details
#12ACCEPTED0.09 s3details
#13ACCEPTED0.04 s3details
#14--3details
#15--3details
#16--3details
#17--3details

Compiler report

input/code.cpp: In function 'std::vector<std::vector<int> > get_all_sub_tables(std::vector<int>&)':
input/code.cpp:87:32: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int n_digit = 1; n_digit <= table.size(); n_digit++)
                        ~~~~~~~~^~~~~~~~~~~~~~~
input/code.cpp: In function 'std::vector<std::vector<int> > get_sub_tables_perdigit(std::vector<int>&, int)':
input/code.cpp:123:32: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int n_digit = 1; n_digit <= table.size(); n_digit++)
                        ~~~~~~~~^~~~~~~~~~~~~~~
input/code.cpp:146:24: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    if (sub_nums.size() == n_digit)
        ~~~~~~~~~~~~~~~~^~~~~~~~~~

Code

#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
#include <random>

int abs(int val)
{
	if (val < 0)
		return -val;
	else
		return val;
}
int max_e(const std::vector<int>& elements)
{
	int max;
	max = elements[0];
	for (auto e : elements)
	{
		if (e > max)
			max = e;
	}
	return max;
}
int min_e(const std::vector<int>& elements)
{
	int min;
	min = elements[0];
	for (auto e : elements)
	{
		if (e < min)
			min = e;
	}
	return min;
}

std::vector<int> generate_table(int n)
{
	std::vector<int> table;
	table.reserve(n);
	std::random_device rd;
	std::mt19937 gen(rd());
	std::uniform_int_distribution<int> dist(1, 99999999);
	for (int i = 0; i < n; i++)
	{
		table.emplace_back(dist(gen));
	}
	return table;
}
std::vector<std::vector<int>> get_sub_tables(std::vector<int>& table, int difference)
{
	std::vector<std::vector<int>> sub_tables;

	for (auto start_e = table.begin(); start_e != table.end(); ++start_e)
	{
		sub_tables.emplace_back(std::vector<int>{*start_e});
		auto max = start_e;
		for (auto cur_e = start_e + 1; cur_e != table.end(); ++cur_e)
		{
			int d = *max - *cur_e;
			if (abs(d) <= difference)
			{
				std::vector<int> temp;
				for (auto i = start_e; i != cur_e + 1; ++i)
				{
					temp.emplace_back(*i);
				}
				if ((max_e(temp) - min_e(temp) <= difference))
				{
					sub_tables.emplace_back(temp);
				}
					
				if (d < 0)
					max = cur_e;
			}
			else
			{
				break;
			}
		}
	}
	return sub_tables;
}
std::vector<std::vector<int>> get_all_sub_tables(std::vector<int>& table)
{
	std::vector<std::vector<int>> sub_tables;
	for (int n_digit = 1; n_digit <= table.size(); n_digit++)
	{
		for (auto left_i = table.begin(); left_i != table.end(); ++left_i)
		{
			if (table.end() - left_i < n_digit)
				break;
			std::vector<int> sub_nums;
			for (auto right_i = left_i; right_i != left_i + n_digit; ++right_i)
			{
				if (right_i == table.end())
					break;
				sub_nums.emplace_back(*right_i);
			}
			sub_tables.emplace_back(sub_nums);
		}
	}
	return sub_tables;
}
std::vector<std::vector<int>> get_sub_tables_d(const std::vector<std::vector<int>>& sub_tables, int difference)
{
	std::vector<std::vector<int>> sub_tables_d;
	for (auto table = sub_tables.begin(); table != sub_tables.end(); ++table)
	{
		int min = *std::min_element(table->begin(), table->end());
		int max = *std::max_element(table->begin(), table->end());
		if ((max - min) <= difference)
		{
			sub_tables_d.emplace_back(*table);
		}
	}
	return sub_tables_d;
}

std::vector<std::vector<int>> get_sub_tables_perdigit(std::vector<int>& table, int difference)
{
	std::vector<std::vector<int>> sub_tables;
	for (int n_digit = 1; n_digit <= table.size(); n_digit++)
	{
		int min;
		for (auto left_i = table.begin(); left_i != table.end(); ++left_i)
		{
			if (table.end() - left_i < n_digit)
				break;
			min = *left_i;
			std::vector<int> sub_nums;
			for (auto right_i = left_i; right_i != left_i + n_digit; ++right_i)
			{
				int d = min - (*right_i);
				if (std::abs(d) <= difference)
				{
					sub_nums.emplace_back(*right_i);
					if (d > 0)
						min = *right_i;
				}
				else
					break;
				if (right_i == table.end())
					break;
			}
			if (sub_nums.size() == n_digit)
			{
				int d = max_e(sub_nums) - min_e(sub_nums);
				if (abs(d) <= difference)
				{
					sub_tables.emplace_back(sub_nums);
				}
				
			}
				
		}
	}
	return sub_tables;
}

void print_sub_tables(const std::vector<std::vector<int>>& source)
{
	for (auto table : source)
	{
		std::cout << "[ ";
		for (auto element : table)
		{
			std::cout << element << " ";
		}
		std::cout << "]";
	}
}

int main()
{
	//int d = 50000000;
	//for (int n = 0; n < 200; n++)
	//{
	//	std::vector<int> table = generate_table(n);
	//	std::cout << "results for n: " << n << '\n';
	//	std::vector<std::vector<int>> correct = get_sub_tables_d(get_all_sub_tables(table), d);
	//	std::vector<std::vector<int>> first = get_sub_tables_first_alg(table, d);
	//	std::cout << "correct: " << correct.size() << '\n';
	//	std::cout << "first: " << first.size() << '\n';
	//}
	int n;
	int d;
	std::vector<int> table;
	std::cin >> n;
	std::cin >> d;
	for (int temp, i = 0; i < n; i++)
	{
		std::cin >> temp;
		table.emplace_back(temp);
	}
	auto s_t = get_sub_tables(table, d);
	std::cout << s_t.size();

	return 0;
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

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

correct output
5050

user output
5050

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 2
5 5 2 4 3 5 3 4 3 2 3 4 5 4 4 ...

correct output
317

user output
317

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 10
71 60 61 96 25 10 10 9 84 85 1...

correct output
119

user output
119

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 990000000
111122929 961821360 578238211 ...

correct output
4006

user output
4006

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 1000000000
553190572 453407680 667300705 ...

correct output
5050

user output
5050

Test 6

Group: 2, 3

Verdict:

input
2000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
2001000

user output
(empty)

Test 7

Group: 2, 3

Verdict: ACCEPTED

input
2000 2
4 4 1 4 2 3 1 2 1 3 5 2 2 4 4 ...

correct output
6340

user output
6340

Test 8

Group: 2, 3

Verdict: ACCEPTED

input
2000 10
65 88 33 88 41 10 17 38 22 3 8...

correct output
2413

user output
2413

Test 9

Group: 2, 3

Verdict:

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
(empty)

Test 10

Group: 2, 3

Verdict:

input
2000 1000000000
621947980 510355354 756705418 ...

correct output
2001000

user output
(empty)

Test 11

Group: 3

Verdict:

input
100000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5000050000

user output
(empty)

Test 12

Group: 3

Verdict: ACCEPTED

input
100000 2
3 3 1 3 3 1 1 5 1 2 5 4 1 3 1 ...

correct output
317066

user output
317066

Test 13

Group: 3

Verdict: ACCEPTED

input
100000 10
10 3 6 3 43 60 5 48 15 27 86 4...

correct output
123292

user output
123292

Test 14

Group: 3

Verdict:

input
100000 999990000
460235639 963048588 47270983 3...

correct output
4946886742

user output
(empty)

Test 15

Group: 3

Verdict:

input
100000 1000000000
885457070 18257718 927615960 3...

correct output
5000050000

user output
(empty)

Test 16

Group: 3

Verdict:

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

correct output
3750075000

user output
(empty)

Test 17

Group: 3

Verdict:

input
100000 50000
100000 99999 99998 99997 99996...

correct output
3750075000

user output
(empty)