传送门:AtCoder Regular Contest 167 - AtCoder
再次感谢樱雪喵大佬的题解,讲的很详细,Orz。
大佬的博客链接如下:Atcoder Regular Contest 167 - 樱雪喵 - 博客园 (cnblogs.com)
第一题很签到,就省略掉了。
第二题其实也不算难,要想清楚因子之间的关系(可是本人没长脑子被卡了俩小时)。通过分解质因数来得出最后的数有多少个因子,然后两两匹配,如果因子个数是奇数,说明存在完全平方数因子的情况,于是单独计算出这样的贡献。
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PII;
const int N=998244353;
const int MX=0x3f3f3f3f3f3f3f3f;
// void read(__int128 &x)
// {
// x=0;
// int f=1;
// char ch;
// if((ch=getchar())=='-')
// f=-f;
// else
// x=x*10+ch-'0';// while((ch=getchar())>='0'&&ch<='9')
// x=x*10+ch-'0';
// x*=f;
// }
// void print(__int128 x){// if(x<0){
// putchar('-');
// x=-x;
// }
// if(x>9)
// print(x/10);// putchar(x%10+'0');
// }
int n,m;
int an;
int su[1000005];
bool c[1000005];void suu(int x){for(int i=2;i<=x;i++){if(!c[i])su[++an]=i;for(int j=1;j<=an&&su[j]*i<=x;j++){c[su[j]*i]=1;if(i%su[j]==0)break;}}
}
int kuai(int a,int b){int ans=1;while(b){if(b&1)ans=ans*a%N;b>>=1;a=a*a%N;}return ans%N;
}
void icealsoheat(){cin>>n>>m;int bn=n;if(m==0){cout<<0;return;}vector<PII>ve;for(int i=1;i<=an&&su[i]<=sqrt(n);i++){if(n%su[i]==0){ve.push_back({su[i],0});while(n>1&&n%su[i]==0){ve.back().second++;n/=su[i];}}}if(n>1){ve.push_back({n,1});}int sum=1;int cnt=0;for(auto [i,j]:ve){if(m%2==1&&j%2==1)cnt=1;sum=sum*(m%N*j%N+1ll)%N;}int ans=0;// ans=sum*kuai(2ll,N-2)%N*(m%N)%N;// // if(cnt==0)ans=(ans+m/2)%N;// if(cnt==0){// ans=((ans-1)%N+N)%N;// ans=ans=(ans+m/2)%N;// }if(cnt==0){sum=((sum-1)%N+N)%N;}ans=sum*kuai(2ll,N-2)%N*(m%N)%N;if(cnt==0)ans=(ans+m/2)%N;cout<<ans;}
signed main(){ios::sync_with_stdio(false);cin.tie();cout.tie();suu(1000000);int _;_=1;// cin>>_;while(_--){icealsoheat();}
}
C - MST on Line++
c,写着题的时候真的很想骂娘。没想到要在限制下标并且在各种顺序的情况下,还得考虑最小生成树的贡献。。。。。。。脑子快炸了,看大佬的代码,好不容易才磕出来。同时也学到了一种新思路。首先,计算这种排列组合题,一般都会想到求出每一个值对答案的贡献,然后相加。在用kruskal算法求最小生成树的时候,我们发现我们只用考虑边全值的大小,对其优先排列。那我们只要求出每一个Ai对应的有几个边的长度就好了。因为我们需要求的是最小值,所以,尽可能的让所有数小才是最优的,按照数值顺序来说,相邻的会尽可能的小。这里看Atcoder Regular Contest 167 - 樱雪喵 - 博客园 (cnblogs.com)
佬的博客吧,解释的特别清楚。
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PII;
const int N=998244353;
const int MX=0x3f3f3f3f3f3f3f3f;
int n,k;
int an;
int a[500005];
int c[5005][5005];
int f[500005];
int be[500005];
void init(int mx)
{for(int i=0;i<=mx;i++)for(int j=0;j<=i;j++) c[i][j]=j?(c[i-1][j-1]+c[i-1][j])%N:1;
}
void icealsoheat(){cin>>n>>k;for(int i=1;i<=n;i++){cin>>a[i];}sort(a+1,a+1+n);be[0]=1;for(int i=1;i<=n;i++){be[i]=be[i-1]*i%N;}for(int i=1;i<=n;i++){for(int j=1;j<=k;j++){f[i]=(f[i]+(i-1)*c[n-j][i-1]%N)%N;}f[i]=be[i]*be[n-i]%N*f[i]%N;}int ans=0;for(int i=1;i<=n;i++){ans=(ans+((f[i]-f[i-1])%N+N)%N*a[i]%N)%N;}cout<<ans;}
signed main(){ios::sync_with_stdio(false);cin.tie();cout.tie();int _;_=1;init(5000);// cin>>_;while(_--){icealsoheat();}
}
D - Good Permutation
这道题最开始我是用优先队列来维护的,因为我们要找在改变次数最小的基础上,要求词序也要最小。我们不妨把所有的序列都涂上各自的颜色,看看有几种颜色,然后每个都取最小的那个,进行不断的替换。但出现了问题。因为存在会误删一些边和点的情况。
后来看了佬的思路,感觉很奇妙,通过并查集来找所有所有的环,然后用set去维护这个环的最小值,如果当前的最小值小于后面环的最小值的话,就替换并且将两个环合并。否则我们不希望字典序变大,尽量不换。但如果这是它所在连通块的最后一个位置,必须要换,那就找后面最小的环值来替换这个值。
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PII;
const int N=998244353;
const int MX=0x3f3f3f3f3f3f3f3f;
int n,k;
int an;
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
int col,tot,top,num;
// int co[200010];
// int dfn[200010];
// int low[200010];
// int a[200010];
int b[200005];
int pre[200005];
int fa[200005];
int c[200005];
int siz[200005];
int mn[200005];
set<PII>q;
int find(int x){if(fa[x]==x)return x;return fa[x]=find(fa[x]);
}
void icealsoheat(){cin>>n;q.clear();for(int i=1;i<=n;i++){cin>>b[i];pre[b[i]]=i;fa[i]=i;siz[i]=1;mn[i]=i;q.insert({i,i});}auto add=[&](int x,int y)->void{x=find(x);y=find(y);if(x==y)return;q.erase({mn[x],x});q.erase({mn[y],y});fa[x]=y;siz[y]+=siz[x];mn[y]=min(mn[x],mn[y]);q.insert({mn[y],y});};for(int i=1;i<=n;i++){add(i,b[i]);}// for(int i=1;i<=n;i++){// cout<<fa[i]<<"+++\n";// cout<<mn[fa[i]]<<"---\n";// cout<<pre[b[fa[i]]]<<"\n";// }for(int i=1;i<=n;i++){if(q.size()==1)break;auto it=q.begin();if(find(it->second)==find(i))it++;if(it->first<b[i]||siz[find(i)]==1){int j=pre[it->first];// cout<<i<<":"<<j<<"\n";swap(b[j],b[i]);swap(pre[b[j]],pre[b[i]]);add(i,j);}siz[find(i)]--;}for(int i=1;i<=n;i++)cout<<b[i]<<" ";cout<<"\n";
}
signed main(){ios::sync_with_stdio(false);cin.tie();cout.tie();int _;_=1;cin>>_;while(_--){icealsoheat();}
}