CF1005A Tanya and Stairways 题解

看了大家的题解，感觉我的有些没必要，竟然要用 map。

我第一眼看这道题，感觉很简单，然后打着代码，发现了一个问题：

连续的1不好处理。

于是我采用了这个方法：

开一个map<int, bool>，每次读入一个数，如果是1，将当前 i 当作下标，变成 true，即 mp[i] = true，然后先输出 mp.sizs()，用一个迭代器，下一个是1的位置减去现在的这个位置就是当前的答案。

比如：

5
1 3 4 1 2

第一个1在下标1，第二个在4，4-1=3,3就是答案。

那么对于最后一个，也就是2，如何处理？

我们可以用总数n加上1减去4，也就是5+1-4，得到2。

代码：

#include <iostream>
#include <map>
using namespace std;

map<int, bool> mp;

int main()
{
	int n, cnt = 0, now_cnt = 1;
	cin >> n;
	for (register int i = 1; i <= n; i++)
	{
		int x;
		cin >> x;
		if (x == 1)
		{
			mp[i] = true;
		}
	}
	cout << mp.size() << endl;
	map<int, bool>::iterator it_end = mp.end(), real_it = mp.begin();
	++real_it;
	for (register map<int, bool>::iterator it = real_it; it != it_end; ++it)
	{
		map<int, bool>::iterator it_tmp = it;
		--it_tmp;
		cout << it -> first - it_tmp -> first << " ";
	}
	--it_end;
	cout << n - (it_end -> first) + 1 << endl;
	return 0;
}