HDU6021：MG loves string（思维 & 容斥）

MG loves string

 

 Accepts: 30

 

 Submissions: 67

 Time Limit: 2000/1000 MS (Java/Others)

 

 Memory Limit: 262144/262144 K (Java/Others)

问题描述

$MG$是一个很忙碌的男孩子。今天他沉迷于这样一个问题：

对于一个长度为$N$的由小写英文字母构成的随机字符串，当它进行一次变换，所有字符$i$都会变成$a[i]$。

$MG$规定所有$a[i]$构成了$26$个字母组成的排列。

$MG$现在需要知道这个随机串变换到自身的期望变换次数。请你输出期望答案乘上26^n26​n​​以后模 $1000000007$ 的结果。

$MG$认为这件事非常容易，不屑于用计算机解决，于是运用他高超的人类智慧开始进行计算。作为一名旁观者，你也想挑战$MG$智慧，请你写个程序，计算答案。

输入描述

第一行一个整数$T$，代表数据组数（$1<=T<=10$）。

接下来，对于每组数据——

第一行一个整数$N$，表示给定的随机串长度（$1<=N<=1000000000$）。

第二行$26$个字母，表示a_ia​i​​序列

输出描述

对于每一组数据，输出一行。

显然，这个期望是一个实数。请你输出它乘上26^N26​N​​以后模 $1000000007$ 的结果

输入样例

2
2
abcdefghijklmnpqrstuvwxyzo
1
abcdefghijklmnopqrstuvwxyz

输出样例

5956
26

思路：对于每个字符，不断地跳肯定能跳回自己本身，定义a[i]为第i个字符跳回自身的次数，易知不同的a[i]最多只有6种，因为7+6+5+4+3+2+1 > 26，那么先预处理a[i]的种数及其对应的字符个数，如果将每个a[i]种类看成不同的集合。那么题目就变成计算从不同集合中取若干个数的所有情况，用容斥可以解决。

# include <iostream>
# include <cstring>
# include <cstdio>
using namespace std;
typedef long long LL;
int cnt, n, a[28], b[28], c[28], vis[28];
LL mod = 1000000007, ans;
char s[28];

LL qmod(LL a, LL b)
{
    LL ans = 1, pow = a%mod;
    while(b)
    {
        if(b&1) ans = (ans*pow)%mod;
        pow = (pow*pow)%mod;
        b >>= 1;
    }
    return ans;
}
LL gcd(LL a, LL b) {return b==0?a:gcd(b, a%b);}
LL lcm(LL a, LL b) {return a/gcd(a, b)*b;}
void solve()
{
    memset(a, 0, sizeof(a));
    memset(b, 0, sizeof(b));
    memset(c, 0, sizeof(c));
    memset(vis, 0, sizeof(vis));
    for(int i=1; i<=26; ++i)
        s[i] = s[i]-'a'+1;
    for(int i=1; i<=26; ++i)
    {
        if(!vis[i])
        {
            vis[i] = 1;
            int j=s[i], k=1;
            while(j != i) ++k, vis[j]=1, j=s[j];
            c[k] += k;
        }
    }
    ans = (LL)0;
    cnt = 0;
    for(int i=1; i<=26; ++i)
        if(c[i])
            a[++cnt] = i, b[cnt] = c[i];//a[i]是第i种跳数，b[i]是a[i]跳数有多少个字母。
    for(int i=0; i<(1<<cnt); ++i)
    {
        int one = __builtin_popcount(i);
        LL tmp = 0, tmp2=1;
        for(int j=0; j<(1<<cnt); ++j)//枚举子状态，待改良。
        {
            if((j|i) == i)
            {
                LL tmp1 = 0, tmp3 = 0, tmp4=0;
                for(int k=0; k<cnt; ++k)
                {
                    if((1<<k)&j)
                    {
                        tmp2 = lcm(tmp2, a[k+1]);
                        ++tmp3;
                        tmp4 += b[k+1];
                    }
                }
                if((one&1) != (tmp3&1))//奇偶不同。
                    tmp = (tmp - qmod(tmp4, n) + mod) % mod;
                else
                    tmp = (tmp + qmod(tmp4, n)) % mod;
            }
        }
        ans = (ans + tmp*tmp2)%mod;
    }
    printf("%lld\n",ans);
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        scanf("%s",s+1);
        solve();
    }
    return 0;
}