今天包括昨天带数学的树形dp基本算是3个数学题
题意:给一棵树每个节点有一个数 要求求出最长的一条路径 这个路径满足条件 所以在上面的点gcd>1
思路:树形dp便利到每个节点的时候维护一个子树的前缀最大值(对每个质因数所能形成的长度)Ans = pref[divsor] + len[divisor].然后用这个孩子下面的长度更新一下前缀最大值区。当整个这个节点和子树更新完后把自己的这个前缀最大值用len数组返回回去。dfs回溯的一个用法。此题注意复杂度,并且因子取质因子否则炸空间。(自闭)
#include <bits/stdc++.h>
using namespace std;
int a[200001];
vector<int> G[200001];
vector<int> div0[200001];
map<int,int> div1[200001];
int vis[200001];
int ans;
void init(){
int n = 2e5;
for(int i=2;i<=n;i++){
if(!vis[i]){
for(int j=i;j<=n;j+=i){
vis[j] = 1;
div0[j].push_back(i);
}
}
}
return;
}
void dfs(int v,int fa){
for(auto to : G[v]){
if( to == fa) continue;
dfs(to,v);
int num = a[to];
for(int i=0;i<div0[num].size();i++){
ans = max(ans,1);
int now = div0[num][i];
if(a[v] % now == 0 ){
ans = max(ans,div1[v][now] + vis[now] + 1);
div1[v][now] = max(div1[v][now],vis[now]);
}
vis[now] = 0;
}
}
for(int i=0;i<div0[a[v]].size();i++){
int now = div0[a[v]][i];
vis[now] = div1[v][now] +1;
}
div1[v].clear();
return;
}
int main()
{
int n;
cin >> n;
init();
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
if(n==1 && a[1]>1){
cout<<"1";
return 0;
}
for(int i=0;i<n-1;i++){
int u,v;
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1,-1);
cout <<ans;
return 0;
}
473E
链接:https://codeforc.es/contest/959/problem/E
题意:给n个点分别为[0,N) 每个点对连一条边 权值为他们的 异或和。求这个图的最小生成树大小。
思路:利用克鲁斯卡尔的思想知道每次连接能连并且消费最小的边。而这里边权为异或和,故第一步肯定是所有第一位为0的数连比他大1的数。这样就连了n/2次。然后把连好的两点看成一个点再次连相邻的集合。此时消费为1*2.边权每次乘二以此递归下去。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n;
ll ans;
void dfs(ll v,ll num){
if(v<=1) return;
//cout << v <<" " << num <<" ";
//cout << (v/2) * num <<"\n";
ans += (v/2) * num;
dfs(v/2 + ((v%2)>0) ,num*2);
}
int main()
{
cin>>n;
dfs(n,1);
cout << ans;
return 0;
}
473D
链接:https://codeforc.es/contest/959/problem/D
题意:给定n个数,要你构造另外n个数 字典序大于给定的。并且n个数字互质。
思路:由于互质很容易知道就是n个数没有共同因子,故我们从头到尾构造每次构造完一个数把他的质因子标记一下,一旦与a数组不同的时候,可以从2尝试着取。(因为前面以前构造成比给定的大了。不需要再从a[i]开始往后取了。因为1e5-1e6这个数量级的质因数应该不多而且题目给了4000ms 所以复杂度是完全ok的 。
官方题解证明的复杂度:
#include <bits/stdc++.h>
using namespace std;
int used[2000500];
int vis[2000500];
int a[100050];
int b[100050];
vector<int> G[2000500];
void init(){
for(int i=2;i<=2e6;i++){
if(!used[i]){
for(int j=i;j<=2e6;j+=i){
G[j].push_back(i);
used[j] = 1;
}
}
}
}
int check(int num){
for(auto t:G[num]){
if(vis[t]){
return 1;
}
}
return 0;
}
int main()
{
int n;
cin>>n;
init();
memset(used,0,sizeof(used));
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
int f = 0;
int p = 2;
for(int i=1;i<=n;i++){
int num = a[i];
if(f){
while(check(p)) p++;
b[i] = p;
for(auto t : G[p]){
vis[t] = 1;
}
p++;
continue;
}
if(check(num)) f=1;
while(check(num)) num++;
b[i] = num;
for(auto t : G[num]){
vis[t] = 1;
}
}
for(int i=1;i<=n;i++){
printf("%d ",b[i]);
}
return 0;
}