[USACO20FEB] Clock Tree S

一开始读错题意了。
然后以为自己做法假了，实际并没有。
有 \(O(n)\) 做法但我写的是 \(O(n^2)\) 的。
换根dp就行了，每次判断一下能不能成功。
注意如果到最后根是 \(1\) 也是可以成功的。

点击查看代码

#include<bits/stdc++.h>
#define p_b push_back
#define e_b emplace_back
#define ll long long
#define pii pair<int,int>
#define fir first
#define sec second
#define il inline
#define ios ios::sync_with_stdio(0),cin.tie(0)
using namespace std;
const int N=2501,inf=1e9;
int n,a[N],t[N],ans;
vector<int>g[N];
int dfs(int u,int fa){int res=0;for(int v:g[u])if(v!=fa){dfs(v,u);res+=12-t[v],t[v]=12;}//cout<<u<<' '<<res<<'\n';return t[u]=(res+t[u]-1)%12+1;
}
void solve(){cin>>n;for(int i=1;i<=n;i++)cin>>a[i];for(int i=1,u,v;i<n;i++){cin>>u>>v;g[u].e_b(v),g[v].e_b(u);}for(int i=1;i<=n;i++){for(int j=1;j<=n;j++)t[j]=a[j];dfs(i,0);if(t[i]==1||t[i]==12)ans++;}cout<<ans;
}
signed main(){ios;solve();return 0;
}