POJ 2960 S-Nim（SG函数模板题）

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;


// sg值的计算方法：
//1.可选步数为1~m的连续整数，直接取模即可，SG(x) = x % (m+1);‘
//2.可选步数为任意步，SG(x) = x
//3.可选步数为一系列不连续的数，用打表或者dfs计算
const int maxn=105, maxm=10005;
int S[maxn], sg[maxm], vis[maxm];
int n, m, size;
void get_sg()
{
    memset(sg, 0, sizeof(sg));
    for(int i=1; i<=maxm; i++)
    {
        memset(vis, 0, sizeof(vis));
        for(int j=0; j<size; j++)
        {
            if(i<S[j]) break;
            vis[sg[i-S[j]]]=1;
        }
        for(int j=0; j<=maxn; j++)
            if(!vis[j]){
                sg[i]=j; break;
            }
    }
}

int dfs_sg(int x)
{
    if(sg[x]!=-1) return sg[x];
    memset(vis, 0, sizeof(vis));
    for(int i=0; i<n; i++)
    {
        if(x>=S[i])
        {
            dfs_sg(x-S[i]);
            vis[sg[x-S[i]]]=1;
        }
    }
    int res;
    for(int i=0; ;i++)
        if(!vis[i]){
            res=i; break;
        }
    return sg[x]=res;
}

int main()
{
    //freopen("in.txt", "r", stdin);
    while(scanf("%d", &size), size)
    {
        for(int i=0; i<size; i++)
            scanf("%d", &S[i]);
        sort(S, S+size);
        get_sg();
        scanf("%d", &n);
        for(int i=0; i<n; i++)
        {
            scanf("%d", &m);
            int ans=0, t;
            for(int i=0; i<m; i++)
            {
                scanf("%d", &t);
                ans^=sg[t];
            }
            printf(ans? "W":"L");
        }
        printf("\n");
    }
    return 0;
}