Submission #267121

#	Time	Username	Problem	Language	Result	Execution time	Memory
267121		peuch	Toys (CEOI18_toy)	C++17	79 / 100	5078 ms	7640 KiB

toy

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;
#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")


const int MAXN = 2e4;
 
int n;
int cnt;
int v[MAXN], qnt[MAXN];
 
vector<int> ans;
unordered_map<int, vector<int> > dp, divisor;
 
void getDiv(int num);
void calcDP(int num);
 
int main(){
	scanf("%d", &n);
	calcDP(n);
	sort(ans.begin(), ans.end());
	printf("%lu\n", ans.size());
	for(int i = 0; i < ans.size(); i++)
		printf("%d ", ans[i]);
	printf("\n");
}
 
void getDiv(int num){
	if(!divisor[num].empty()) return;
	if(num == 1){
		divisor[num].push_back(1);
		return;
	}
	map<int, bool> marc;
	for(int i = sqrt(num); i > 1; i--){
		if(num == i) continue;
		if(num % i) continue;
		if(divisor[i].empty() || divisor[num / i].empty()) continue;
		getDiv(i);
		getDiv(num / i);
		for(int j = 0; j < divisor[i].size(); j++){
			for(int k = 0; k < divisor[num / i].size(); k++){
				int val = divisor[i][j] * divisor[num / i][k];
				if(marc[val]) continue;
				marc[val] = 1;
				divisor[num].push_back(val);
			}
		}
		return;
	}
	for(int i = sqrt(num); i > 1; i--){
		if(num == i) continue;
		if(num % i) continue;
		if(divisor[i].empty() && divisor[num / i].empty()) continue;
		getDiv(i);
		getDiv(num / i);
		for(int j = 0; j < divisor[i].size(); j++){
			for(int k = 0; k < divisor[num / i].size(); k++){
				int val = divisor[i][j] * divisor[num / i][k];
				if(marc[val]) continue;
				marc[val] = 1;
				divisor[num].push_back(val);
			}
		}
		return;
	}
	for(int i = sqrt(num); i > 1; i--){
		if(num == i) continue;
		if(num % i) continue;
		getDiv(i);
		getDiv(num / i);
		for(int j = 0; j < divisor[i].size(); j++){
			for(int k = 0; k < divisor[num / i].size(); k++){
				int val = divisor[i][j] * divisor[num / i][k];
				if(marc[val]) continue;
				marc[val] = 1;
				divisor[num].push_back(val);
			}
		}
		return;
	}
	divisor[num].push_back(1);
	divisor[num].push_back(num);
}
 
void calcDP(int num){
	if(!dp[num].empty()){
		ans = dp[num];
		return;
	}
	getDiv(num);
	vector<int> aux = divisor[num];
	vector<int> ret;
	ret.push_back(num - 1);
	vector<int> aux1, aux2;
	map<int, bool> marc;
	for(int i = 0; i < aux.size(); i++){
		if(aux[i] == num || aux[i] == 1) continue;
		if(num / aux[i] < aux[i]) continue;
		calcDP(aux[i]);
		aux1 = ans;
		calcDP(num / aux[i]);
		aux2 = ans;
		for(int i1 = 0; i1 < aux1.size(); i1++){
			for( int i2 = 0; i2 < aux2.size(); i2++){
				if(marc[aux1[i1] + aux2[i2]]) continue;
				ret.push_back(aux1[i1] + aux2[i2]);
				marc[aux1[i1] + aux2[i2]] = 1;
			}
		}
	}
	ans = ret;
	dp[num] = ans;
}

Compilation message (stderr)

toy.cpp:5: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
    5 | #pragma GCC optimization ("O3")
      | 
toy.cpp:6: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
    6 | #pragma GCC optimization ("unroll-loops")
      | 
toy.cpp: In function 'int main()':
toy.cpp:26:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   26 |  for(int i = 0; i < ans.size(); i++)
      |                 ~~^~~~~~~~~~~~
toy.cpp: In function 'void getDiv(int)':
toy.cpp:44:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   44 |   for(int j = 0; j < divisor[i].size(); j++){
      |                  ~~^~~~~~~~~~~~~~~~~~~
toy.cpp:45:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |    for(int k = 0; k < divisor[num / i].size(); k++){
      |                   ~~^~~~~~~~~~~~~~~~~~~~~~~~~
toy.cpp:60:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   60 |   for(int j = 0; j < divisor[i].size(); j++){
      |                  ~~^~~~~~~~~~~~~~~~~~~
toy.cpp:61:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   61 |    for(int k = 0; k < divisor[num / i].size(); k++){
      |                   ~~^~~~~~~~~~~~~~~~~~~~~~~~~
toy.cpp:75:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   75 |   for(int j = 0; j < divisor[i].size(); j++){
      |                  ~~^~~~~~~~~~~~~~~~~~~
toy.cpp:76:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   76 |    for(int k = 0; k < divisor[num / i].size(); k++){
      |                   ~~^~~~~~~~~~~~~~~~~~~~~~~~~
toy.cpp: In function 'void calcDP(int)':
toy.cpp:100:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  100 |  for(int i = 0; i < aux.size(); i++){
      |                 ~~^~~~~~~~~~~~
toy.cpp:107:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  107 |   for(int i1 = 0; i1 < aux1.size(); i1++){
      |                   ~~~^~~~~~~~~~~~~
toy.cpp:108:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  108 |    for( int i2 = 0; i2 < aux2.size(); i2++){
      |                     ~~~^~~~~~~~~~~~~
toy.cpp: In function 'int main()':
toy.cpp:22:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   22 |  scanf("%d", &n);
      |  ~~~~~^~~~~~~~~~

• Subtask 1: 19 / 19
• Subtask 2: 20 / 20
• Subtask 3: 20 / 20
• Subtask 4: 20 / 20
• Subtask 5: 0 / 21