CodeForces 150 B.Quantity of Strings（组合数学+并查集）

最新推荐文章于 2021-05-16 16:34:09 发布

v5zsq

最新推荐文章于 2021-05-16 16:34:09 发布

阅读量372

点赞数

分类专栏： Code Forces 组合数学并查集

本文链接：https://blog.csdn.net/V5ZSQ/article/details/78943329

版权

Code Forces 同时被 3 个专栏收录

606 篇文章 6 订阅

订阅专栏

组合数学

190 篇文章 1 订阅

订阅专栏

并查集

38 篇文章 0 订阅

订阅专栏

Description

要求用 $m$ 种字母构造一个长度为 $n$ 的字符串，使得该字符串的所有长度为 $k$ 的子串均为回文串，问方案数

Input

三个整数 $n,m,k(1\le n,m,k\le 2000)$

Output

输出满足条件的字符串数量，结果模 $10^9+7$

Sample Input

5 2 4

Sample Output

Solution

对于每一个位置，考虑其在一个长度为 $k$ 的子串中的位置，然后找到与其对称的位置，那么这两个位置的字符应该相同，将相同的位置合并，假设最后有 $cnt$ 个集合，同一个集合中的位置字符相同，不同集合中的位置字符互不相关，故答案为 $m^{cnt}$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
#define mod 1000000007
int Pow(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1)ans=(ll)ans*a%mod;
        a=(ll)a*a%mod;
        b>>=1;
    }
    return ans;
}
int n,m,k,fa[maxn];
int find(int x)
{
    if(fa[x]==x)return x;
    return fa[x]=find(fa[x]);
}
void unite(int x,int y)
{
    x=find(x),y=find(y);
    if(x==y)return ;
    fa[x]=y;
}
int main()
{
    int n,m,k;
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        for(int i=1;i<=n;i++)fa[i]=i;
        for(int i=1;i<=n;i++)
            for(int j=1;j<k&&i+k-j<=n;j+=2)
                unite(i,i+k-j);
        int ans=0;
        for(int i=1;i<=n;i++)
            if(fa[i]==i)ans++;
        ans=Pow(m,ans);
        printf("%d\n",ans);
    }
    return 0;
}