【CVTE】兔子藏洞

题目描述

一只兔子藏身于20个圆形排列的洞中（洞从1开始编号），一只狼从x号洞开始找,下次隔一个洞找(即在x＋2号洞找)，在下次个两个洞找(即在x＋5号洞找)，它找了n次仍然没有找到。问兔子可能在那些洞中。

输入描述:

输入有多组数据，每组数据一行两个整数分别为x和n(x <= 20,n <= 100000)

输出描述:

每组数据一行按从小到大的顺序输出兔子可能在的洞，数字之间用空格隔开。若每个洞都不肯能藏着兔子，输出-1。

#include "stdio.h"
#include <math.h>
#include <vector>
#include <iostream>
using namespace std;

int main()
{
	int x, n;
	while (cin >> x >> n)
	{
		vector<bool> flag(20, 1);
		for (int i = 1; i <= n; i++)
		{
			if (i == 1)
				flag[x - 1] = 0;
			else
			{
				x += i;
				flag[(x - 1)%20] = 0;
			}
		}

		vector<int> ans;
		for (int i = 0; i < 20; i++)
		{
			if (flag[i])
				ans.push_back(i + 1);
		}
		if (ans.size() < 1)
			cout << -1 << endl;
		else
		{
			for (int i = 0; i < ans.size(); i++)
			{
				cout << ans[i] << ' ';
			}
			cout<< endl;
		}
	}
	return 0;
}