摩尔斯电码字典序号

// 摩尔斯电码字典序号
// 原题参见
// https://bbs.csdn.net/topics/392691483

#include <iostream>
#include<string>

// 求组合数C(N, n)
int c(int N, int n)
{
	double p = 1.0;
	for (int i = 0; i <n; ++i)
	{
		p *= ((N - i)*1.0 /  (i+1));
	}

	return std::lround(p);
}

// 计算第一个遇到的'o'
int cx(int n, int k)
{
	if (n == k) return 0;
	if (n + 1 == k) return 1;
	return c(k-1, n);
}

// 计算整个字符串
int search(std::string s)
{
	int N = s.length();	// 总长度
	int n = 0;				// 'o' 的个数
	for (char c : s)
	{
		if (c == 'o')
		{
			n++;
		}
	}

	int t = 1; // 第几个(题目中是以1为基)
	for (int i = 0; i<N; ++i)
	{
		if (s[i] == 'o')
		{
			t += cx(n, N - i );
			n--;
		}

		if (N <= i + n)
		{
			break;
		}
	}

	return t;
}

int main()
{
	std::string s = "o-oo--";

	int t = search(s);
	std::cout << "index = " << t <<  std::endl;

	return 0;
}

/* 程序输出
index = 16

手工计算:
1. ---ooo
2. --o-oo
3. --oo-o
4. --ooo-
5. -o--oo
6. -o-o-o
7. -o-oo-
8. -oo--o
9. -oo-o-
10. -ooo--
11. o---oo
12. o--o-o
13. o--oo-
14. o-o--o
15. o-o-o-
16. o-oo--
*/