一天就训了两个并查集的题。。。
第一个水题题意:给一个字符串s和一个排列p,每次s会按照p变成s’,问需要多少次能变成和原来一样的串。
传送门
思路:先按并查集把字符串分成几个部分,设每个部分需要p次操作保持原样,则总的串s只需lcm(p1,p2,p3…pn)次即可。p1,p2…pn暴力模拟即可求出来。(p1,p2的值不会大于其并查集的size)
代码
#include <bits/stdc++.h>
#define PII pair<int,int>
#define int long long
#define PI acos(-1)
using namespace std;
const int N=2e5;
const int M=10000;
const int mod=998244353;
int n,k,t,ans;
char s[N+5],ss[2][N+5];
int a[N+5],fa[N+5];
int Find(int x){
return x==fa[x]? x:fa[x]=Find(fa[x]);
}
vector<int> vec[N+5];
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
int t;cin>>t;
while(t--){
int n;cin>>n;
cin>>s+1;
for(int i=1;i<=n;++i)
cin>>a[i];
for(int i=1;i<=n;++i){
fa[i]=i;
ss[0][i]=s[i];
vec[i].clear();
}
for(int i=1;i<=n;++i){
fa[Find(i)]=Find(a[i]);
}
for(int i=1;i<=n;++i){
vec[Find(i)].push_back(i);
}
int ans=1;
for(int i=1;i<=n;++i){
int tmp=0,op=0;
while(1){//至少变换一次
op^=1;
tmp++;
int fl=1;
for(auto x:vec[i]){
ss[op][x]=ss[op^1][a[x]];
if(ss[op][x]!=s[x]){
fl=0;
}
}
if(fl) break;
}
ans=ans*tmp/__gcd(ans,tmp);
}
cout<<ans<<endl;
}
return 0;
}
/*
*/
第二个水题题意:给一个数组A,(a[i]<=n),输出一个排列P满足p[i]!=a[i]。如果不存在就输出NO。
思路:选出A中的不同元素构成一个集合,然后让集合内元素位置左移一下。其余的元素就可以填不在{P}-{A}中的任意元素。演了队友一手^ ^!。
代码
#include <bits/stdc++.h>
#define PII pair<int,int>
//#define int long long
#define PI acos(-1)
using namespace std;
const int N=2e5;
int a[N+5],vis[N+5];
int ans[N+5];
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
int t;cin>>t;
while(t--){
int n;cin>>n;
map<int,int> mp;
for(int i=1;i<=n;++i)
vis[i]=0;
int pre=0;
for(int i=1;i<=n;++i){
cin>>a[i];
mp[a[i]]=i;
vis[a[i]]=1;
ans[i]=-1;
pre=max(pre,a[i]);
}
if(mp.size()==1){
cout<<"NO"<<endl;
continue;
}
for(auto x:mp){
ans[x.second]=pre;
pre=x.first;
}
for(int i=1,j=1;i<=n;++i){
if(ans[i]!=-1) continue;
while(j<=n&&vis[j]){
j++;
}
ans[i]=j;
vis[j]=1;
}
cout<<"YES"<<endl;
for(int i=1;i<=n;++i){
cout<<ans[i]<<" ";
}
cout<<endl;
}
return 0;
}
/*
*/
后续再加补个dp题。。。
题目:计数问题。有多少个长为n的数组满足a[i]<=n,且区间和是k的倍数的区间个数有t个。(n<=64)
思路:一眼dp,一开始以为是数位dp,考虑到第pos位置时余数为i有几个,然后空间复杂度会G。正解是把区间和的问题转换成前缀和,设dp[i][j][m]表示选择了j个位置放数,当前放的数满足前缀和为i(模k意义下),且区间和是k的倍数的区间个数有m个。则每次只需考虑前缀和为i的位置我们选择了多少个,其对m贡献值有多少。(这里利用了前缀数组的唯一性,即假设我们选择了f个位置让该位置前缀和为i(模k意义下),那么再其他位置确定时这些位置放的数也唯一确定,所以我们不必管这些位置到底放了什么数 )
代码
#include <bits/stdc++.h>
#define PII pair<int,int>
#define int long long
#define PI acos(-1)
#define LL long long
using namespace std;
const int N=64;
const int mod=998244353;
int c[N+5][N+5];
int dp[N+5][N+5][N*N+5];
//dp[i][j][k]表示前缀和为i,(前缀和为0———i-1)放到了j个格子里,贡献为k的方案数
LL fac[N+5];
LL qpow(LL a,LL b)
{
LL res=1;
while(b)
{
if(b&1)
res=res*a%mod;
b/=2;
a=a*a%mod;
}
return res;
}
LL inv(LL a)
{
return qpow(a,mod-2);
}
LL C(LL n, LL m)
{
if(m>n)
return 0;
return ( fac[n]*inv(fac[m])%mod )*inv( fac[n-m] )%mod;
}
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
fac[0]=fac[1]=1;
for(int i=2;i<=N;++i)
fac[i]=fac[i-1]*i%mod;
for(int i=0;i<=N;++i){
for(int j=0;j<=N;++j)
c[i][j]=C(i,j);
}
int n,k,t;cin>>n>>k>>t;
for(int i=0;i<=n;++i){
dp[0][i][i*(i+1)/2]=c[n][i];
//前缀和为0的位置有i个,贡献为i*(i+1)/2
}
for(int i=1;i<=k-1;++i){//当前前缀和
for(int j=0;j<=n;++j){//已经放了j个格子
for(int f=0;f+j<=n;++f){//放到f个格子里
for(int g=0;g<=t;++g){
(dp[i][j+f][g+f*(f-1)/2]+=dp[i-1][j][g]*c[n-j][f])%=mod;//贡献为f*(f-1)/2,
//显然若选出了f个位置前缀和为i,则满足这种放法的方案唯一
}
}
}
}
cout<<dp[k-1][n][t]%mod<<endl;
return 0;
}
/*
*/