GYM 100801 C.Concatenation（组合数学）

Description

给出两个字符串，要求用第一个字符串的前缀和第二个字符串的后缀组成新的字符串，问至多可以组成多少个不同的字符串

Input

输入两个长度不超过$10^5$的只由小写字母组成的字符串

Output

输出方案数

Sample Input

cat
dog

Sample Output

9

Solution

假设第一个字符串$S$的长度为$n$，第二个字符串$T$的长度为$m$，首先第一个字符串的第一个字符和第二个字符串的最后一个字符串必然在最后的字符串中，而根据第一个字符串前缀的长度有$n$种可能和第二个字符串后缀的长度有$m$种可能，故总方案数为$n\times m$，但其中有重复的，即如果取$S[1,...,x]+T[y,...,m]$组成一个字符串$(x>1,y<m)$，但是$S[x]=T[y]$，这样就有两种情况得到同样的结果，故统计$S$字符串后$n-1$个字符中第$i$个字母出现的次数$x[i]$，统计$T$的前$m-1$个字符中第$i$个字母出现的次数$y[i]$，那么重复的方案数即为$\sum\limits_{i=0}^{25}x[i]\cdot y[i]$，从总方案数种减去即可

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;
int x[33],y[33];
char a[maxn],b[maxn];
int main()
{
    freopen("concatenation.in","r",stdin);
    freopen("concatenation.out","w",stdout);
    while(~scanf("%s%s",a,b))
    {
        int n=strlen(a),m=strlen(b);
        memset(x,0,sizeof(x));
        memset(y,0,sizeof(y));
        for(int i=1;i<n;i++)x[a[i]-'a']++;
        for(int i=0;i<m-1;i++)y[b[i]-'a']++;
        ll ans=(ll)n*m;
        for(int i=0;i<26;i++)ans-=(ll)x[i]*y[i];
        printf("%I64d\n",ans);
    }
    return 0;
}