CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:_Ahrou
Submission time:2020-10-02 12:53:44 +0300
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2, 3details
#20.01 s1, 2, 3details
#3ACCEPTED0.01 s1, 2, 3details
#40.01 s1, 2, 3details
#5ACCEPTED0.01 s1, 2, 3details
#6--2, 3details
#70.01 s2, 3details
#80.01 s2, 3details
#9--2, 3details
#10--2, 3details
#11--3details
#120.03 s3details
#130.03 s3details
#14--3details
#15--3details
#16--3details
#17--3details

Code

#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
#include <random>
#include <chrono>
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;
}
int absolute(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;
}
int get_n_sub_tables(std::vector<int>& table, int difference)
{
	int counter = 0;
	for (auto start_e = table.begin(); start_e != table.end(); ++start_e)
	{
		counter++;
		auto max = start_e;
		for (auto cur_e = start_e + 1; cur_e != table.end(); ++cur_e)
		{
			int d = *max - *cur_e;
			if (absolute(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))
				{
					counter++;
				}
				if (d < 0)
					max = cur_e;
			}
			else
			{
				break;
			}
		}
	}
	return counter;
}
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 (absolute(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_sub_tables_debug(std::vector<int>& table, std::vector<std::vector<int>>& correct, 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 (absolute(d) <= difference)
			{
				std::vector<int> temp;
				for (auto i = start_e; i != cur_e + 1; ++i)
				{
					temp.emplace_back(*i);
				}
				if (std::find(correct.begin(), correct.end(), temp) == correct.end())
				{
					std::cout << "table: [";
					for (auto e : temp)
					{
						std::cout << e << " ";
					}
					std::cout << "] does not match \n";	
				}
					
				sub_tables.emplace_back(temp);

				if (d < 0)
					max = cur_e;
			}
			else
			{
				break;
			}
		}
	}
	return sub_tables;
}
int n_subs(const std::vector<int>& table, int differece)
{
	int n_sub_tables = 0;
	for (auto left_i = table.begin(); left_i != table.end(); ++left_i)
	{
		n_sub_tables++;
		int max = *left_i;
		int n_digits = 1;
		for (auto right_i = left_i + 1; right_i != table.end(); )
		{
			int d = max - *right_i;
			if (d < 0)
				max = *right_i;
			if (absolute(d) <= differece)
			{
				if (right_i - left_i >= n_digits)
				{
					n_sub_tables++;
					if (right_i - left_i >= n_digits)
					{
						n_digits++;
						right_i = left_i + 1;
						max = *left_i;
						continue;
					}
				}
				else
					++right_i;
			}
			else
				break;
		}
	}
	return n_sub_tables;
}
//int main()
//{
//	int d = 99999999;
//	for (int n = 0; n < 2000; n++)
//	{
//		auto table = generate_table(n);
//		std::cout << "results for n: " << n << '\n';
//		int correct = get_sub_tables(table, d).size();
//		int new_alg = n_subs(table, d);
//		std::cout << "correct: " << correct << '\n';
//		std::cout << "new: " << new_alg << '\n';
//		if (new_alg != correct)
//			std::cout << "bad";
//	}
//	return 0;
//}
int main()
{
	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);
	}
	std::cout << n_subs(table, d);
	return 0;
}
































/*
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_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 << "]";
	}
}
*/

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:

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

correct output
317

user output
442

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:

input
100 990000000
111122929 961821360 578238211 ...

correct output
4006

user output
4258

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:

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

correct output
6340

user output
8137

Test 8

Group: 2, 3

Verdict:

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

correct output
2413

user output
2432

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:

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

correct output
317066

user output
410909

Test 13

Group: 3

Verdict:

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

correct output
123292

user output
124778

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)