【每日一题006】字节跳动2017后端工程师实习生笔试题-3

题目

题目来源
给出 n 个字符串，对于每个 n 个排列 p，按排列给出的顺序(p[0] , p[1] … p[n-1])依次连接这 n 个字符串都能得到一个长度为这些字符串长度之和的字符串。所以按照这个方法一共可以生成 n! 个字符串。

一个字符串的权值等于把这个字符串循环左移 i 次后得到的字符串仍和原字符串全等的数量，i 的取值为 [1 , 字符串长度]。求这些字符串最后生成的 n! 个字符串中权值为 K 的有多少个。

注：定义把一个串循环左移 1 次等价于把这个串的第一个字符移动到最后一个字符的后面。

思路

用map实现对各个字符串及其数量的统计，然后按照每个字符串、每个偏移量、对应偏移量的每个字母是否相等进行对比。

相关思考

1.对于题目的理解不到位，刚开始理解成k为对应的偏移量，而实际上k其实是满足条件的字符串的数量。
2.全排列借助于一个int数组，使用next_permutation(，)函数进行相应的排列验证。
3.对于map的应用，int实现的并不是标号的作用，而是对相应的字符串的数量进行计数，从而在最后通过每个字符串实现相应偏移量的验证后，能够直接在总数上加对应的字符串数量。

代码（C++/牛客）

#include <iostream>
#include<map>
#include <algorithm> 
using namespace std;
string s[9];
map<string,int>m;//int其实是用来计数的 
int main(){
    int n,k;
	cin>>n>>k;
    for(int i=1;i<=n;i++)
	cin>>s[i];
    int a[10];
    for(int i=1;i<=n;i++)
	a[i]=i;
    do{
        string k;
        for(int i=1;i<=n;i++)
		k+=s[a[i]];
        m[k]++;
    }while(next_permutation(a+1,a+1+n));//全排列 （利用数组a实现对map中元素中的全排列） 
    
    int ans=0;
    for(map<string,int>::iterator it=m.begin();it!=m.end();it++){ //map中的所有元素遍历 
        string x=(*(it)).first;
        int now=0;//now计数一个字符串有几个
        for(int i=0;i<x.size();i++)   //所有偏移量遍历 
		{//偏移量
            int f=1;
            for(int j=0;j<x.size();j++){  
                if(x[j]!=x[(j+i)%x.size()]) //一旦遇到未匹配的立刻修改f值 并退出对比循环 
				{f=0;
				break;}
            }
            now+=f;    
            if(now>k)break;
        }
        if(now==k)
		ans+=m[x];  
    }
    cout<<ans;
}