download:PyTorch入门到进阶 实战计算机视觉与自然语言处理项目
PyTorch是目前深度学习的主流框架之一，它有着成熟的生态、大量开源的源码以及最新的模型，无论学术研究还是工程落地，PyTorch都是主流选择。同时，PyTorch比其他深度学习框架更易学，也是新手入门的好选择。本课程将算法、模型和基础理论知识进行有机结合，结合多个不同的CV与NLP实战项目，帮助大家掌握PyTorch框架的基础知识和使用方法，并学会利用PyTorch框架解决实际问题。通过本课程，可以较平稳地快速入门深度学习领域，初步掌握解决深度学习基础问题的关键性技能。
适合人群
想转行到深度学习方向的工程师；
学习了理论但欠缺实践的深度学习starter；
有志于、感兴趣深度学习的爱好者们；
想快速做实验完成研究任务毕设任务的同学们；
技术储备要求
机器学习相关基本概念；
Python3编程语言；
Ubuntu基本使用知识；
![]()![]()```
1 #include<bits/stdc++.h>
2 using namespace std;
3 const int maxn=3e5+10;
4 const int mod=1e9+7;
5 #define ll long long
6 #define pi pair<int,int>
7 #define fi first
8 #define sc second
9 #define pb push_back
10 int cnt[maxn][2];
11 int a[maxn];
12 vectorE[maxn];
13 int ans,n;
14 int cntr,cntb;
15
16 void dfs(int u,int fa)
17 {
18 if(a[u]==1) cnt[u][1]++;
19 if(a[u]==2) cnt[u][0]++;
20 for(auto &v:E[u])
21 {
22 if(v==fa) continue;
23 dfs(v,u);
24 cnt[u][0]+=cnt[v][0];
25 cnt[u][1]+=cnt[v][1];
26 if(cnt[v][1]==cntr&&cnt[v][0]==0)
27 ans++;
28 if(cnt[v][0]==cntb&&cnt[v][1]==0)
29 ans++;
30 }
31 }
32
33
34 int main()
35 {
36 ios::sync_with_stdio(0);
37 cin.tie(0);
38 cin>>n;
39 for(int i=1;i<=n;i++) cin>>a[i],cntr+=a[i]==1,cntb+=a[i]==2;
40 for(int i=1;i<n;i++)
41 {
42 int x,y;
43 cin>>x>>y;
44 E[x].pb(y);
45 E[y].pb(x);
46 }
47 if(cntr&&cntb)
48 dfs(1,0);
49 else
50 ans=n-1;
51 cout<<ans<<'\n';
52
53
54
55
56
57 }
