JL-05 字符串题解

#JL-05 字符串题解
原题：
字符串
描述

现在有一个长度为 N 的字符串，它的字典集为 M .也就是说每一位有 M 种不同的字母可以选。对于这个字符串所有连续的长度为 K 的子串都必 须是回文串，请问有多少种不同的方案。

由于答案可能很大，请将答案 mod 1e9 + 7 .

输入
一共一行，三个整数表示 N,M,K .

N,M,K ≤ 2000

输出
一共一行，一个整数表示最后答案。

输入样例 1

5 2 4
输出样例 1

2

思路题解：
回文串的实现

回文串长度为K
外层for循环i从1遍历到n
内层while循环Union i—K，i+1—K-1，i+2—K-2…

数据结构——并查集（一开始不知道并查集，想了好久，LGJNB）

Find：确定元素属于哪一个子集。这个确定方法就是不断向上查找找到它的根节点，它可以被用来确定两个元素是否属于同一子集。
Union：将两个子集合并成同一个集合。
Init：并查集初始化。（我下面的题解初始化在主函数中，这样其实不太好，想了想下次应该单独写出来，这样主函数的逻辑会更加清晰，也方便调用）

核心代码：

int find(int k)
{
    if(k!=f[k]) f[k]=find(f[k]);
    return f[k];
}

void Union(int x1, int x2) 
{
    int p1 = find(x1);
    int p2 = find(x2);
    f[p1] = p2;
}

void Init()
{
	for(int i = 1; i <= n; i++)
		f[i] = i;
}

AC代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <string>
#include <queue>
using namespace std;

const int modnum=1000000007;

int f[10000],n,m;

int find(int k)
{
    if(k!=f[k]) f[k]=find(f[k]);
    return f[k];
}

void Union(int x1, int x2) //令x1的根节点为x2的根节点
{
    int p1 = find(x1);
    int p2 = find(x2);
    f[p1] = p2;
}

int main(){
	int n,m,k;
	long long ans=0;
	int sum=0;
	cin>>n>>m>>k;
	if(n<=0||m<=0){
		cout<<"0"<<endl;
		return 0;
	}
	if(k==1||k==0){
		ans=1;
		for(int i=1;i<=n;i++){
			ans*=m;
			ans%=modnum;
		}
		cout<<ans<<endl;
		return 0;
	}
	else{
		for(int i=1;i<=n;i++){//并查集初始化(下次写的时候把这个写到外面当自定义函数好了
	        f[i]=i;
        }
		for(int i=1;i<=n;i++){
			int dk=k-1;
			int di=0;
		    while(i+dk>i+di){
			   	if(i+dk>n) break;
			   	Union(i+dk,i+di); //令i+dk的根节点为i+di
			    dk--;
			    di++;
		   }
		}
		for(int i=1;i<=n;i++){
			if(i==f[i]) sum++;//记录有多少个不同的根节点 代表有多少个位置的数可以取不同的数（当然也可取相同
		}
		ans=1;
		for(int i=1;i<=sum;i++){//模拟 sum个位置每个位置可以放m种 取余 
			ans*=m;
			ans%=modnum;
		}
		cout<<ans<<endl;	
	}
	return 0;
}