传送门: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();
}
}