235. 魔法珠

题目链接：235. 魔法珠

算法分析

有明显的sg函数特征，因为每堆有数字，可以入sg函数的下标。类似于尼姆博弈的取石子。双人、相同规则、必胜或必败，符合公平组合游戏规则。可以用sg函数。

有$n$堆魔法珠，这是有向图游戏和的问题，需要求出来单堆的sg函数，最后异或和即可。所以，需要预处理出$[1,1000]$的sg函数。明显地，$sg[1]=0$是边界。

对于任意的数量r，假设其约数有t1、t2、t3三个。因为要随机消掉一堆，所以r的后继状态总共有三个：{t1,t2}、{t1,t3}、{t2,t3}。所以：
s g [ r ] = m e x ( { s g ( t 1 , t 2 ) , s g ( t 1 , t 3 ) , s g ( t 2 , t 3 ) } ) sg[r] =mex(\lbrace sg(t1,t2),sg(t1,t3),sg(t2,t3) \rbrace) sg[r]=mex({sg(t1,t2),sg(t1,t3),sg(t2,t3)})

如何求$sg(t1,t2)$呢？这又是一个有向图游戏的和的问题：
$$sg(t1,t2)=sg(t1)xorsg(t2)$$

到此，问题就解决了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
using namespace std;
int n, a[110];
int sg[1010], vis[1010], sav[1010];
vector<int> g[1010];
void init()
{
	for (int i = 1; i <= 1000; ++i)  // 倍数法  
		for (int j = 2; i * j <= 1000; ++j) // 不能包含本身，从2倍开始  
			g[i*j].push_back(i);
			
	sg[1] = 0;
	for (int i = 2; i <= 1000; ++i)
	{
		int t = 0;
		for (int j = 0; j < g[i].size(); ++j)
		{
			int r = 0;
			for (int k = 0; k < g[i].size(); ++k)
			{
				if (j == k) continue;  // 消掉一堆  
				r ^= sg[g[i][k]];
			}
			vis[r] = 1;
			sav[++t] = r;
		}
		for (int k = 0; k <= 1000; ++k)
		if (!vis[k])
		{
			sg[i] = k;
			break;
		}
		// 还原，比memset更快些，因为每次只有少量的r被标记 
		for (int k = 1; k <= t; ++k) vis[sav[k]] = 0;  
	}
}
int main()
{	
	init();
	while (~scanf("%d", &n))
	{
		for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);		
		int ans = 0;
		for (int i = 1; i <= n; ++i)
			ans ^= sg[a[i]];
		if (ans) printf("freda\n");
		else printf("rainbow\n");
	}
	return 0;
} 

反思与总结

1. 求每个数的约数，可以用倍数法，效率高。

2. 推sg函数，要结合具体的策略，如本题中是随意消掉一堆，如果不消的话，则难度也会降低。只要符合公平组合游戏，都可以尝试用sg函数解决。