AtCoder AGC033C Removing Coins (博弈论)

题目链接

https://atcoder.jp/contests/agc033/tasks/agc033_c

题解

终于会做点最简单的博弈论了……
首先题目中操作的含义就是选定一个点，把所有不是这个点的叶子删掉（如果这个点不是叶子就删所有叶子）。
对于任何一棵点数不少于\(3\)的树，一定存在一个点（比如非叶子节点），使得对该点操作之后直径减少\(2\)；同时一定存在一个点（比如直径的端点），使得对该点操作后直径减少\(1\)；同时不存在任何一种操作使得直径发生其他的变化。因此这是一个Bash博弈的模型，答案与直径长度模\(3\)的值有关。
设直径长度（点数）为\(l\). 若\(l=1\)则先手必胜，\(l=2\)则后手必胜。后面就变成了刚才讨论的情况，因此当且仅当直径长度\(\mod 3=2\)时后手必胜。
时间复杂度\(O(n)\).

代码

#include
#define llong long long
using namespace std;

inline int read()
{
    int x = 0,f = 1; char ch = getchar();
    for(;!isdigit(ch);ch=getchar()) {if(ch=='-') f = -1;}
    for(; isdigit(ch);ch=getchar()) {x = x*10+ch-48;}
    return x*f;
}

const int N = 2e5;
struct Edge
{
    int v,nxt;
} e[(N<<1)+3];
int fe[N+3];
int fa[N+3];
int len[N+3];
int n,en,mx;

void addedge(int u,int v)
{
    en++; e[en].v = v;
    e[en].nxt = fe[u]; fe[u] = en;
}

void dfs(int u)
{
    for(int i=fe[u]; i; i=e[i].nxt)
    {
        int v = e[i].v; if(v==fa[u]) continue;
        fa[v] = u; dfs(v);
        mx = max(mx,len[u]+len[v]+1);
        len[u] = max(len[u],len[v]+1);
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1; i