【编程笔试】美团2021校招笔试-通用编程题第6场（附思路及C++代码）

练习地址

点此前往练习

小团的装饰物2

小团需要购买m样装饰物。商店出售n种装饰物，按照从小到大的顺序从左到右摆了一排。对于每一个装饰物，小团都给予了一个美丽值a_i。

小团希望购买的装饰物有着相似的大小，所以他要求购买的装饰物在商店中摆放的位置是连续的一段。小团还认为，一个装饰物的美丽值不能低于k，否则会不好看。

现在，请你计算小团有多少种不同的购买方案。

输入描述:

输入第一行包含三个数n, m, k接下来一行n个数，空格隔开，表示商店从左到右摆放的每个装饰物的美丽值。

输出描述:

输出一个数，表示小团购买的方案数。

输入例子1:

8 2 5

5 5 5 4 5 5 5 5

输出例子1:

5

例子说明1:

有[1,2][2,3][5,6][6,7][7,8] 共5段

思路：

两重循环，判断当美丽值小于k时计数+1，并跳出循环。最后用减法的形式输出符合条件的方案数。

代码：

#include<bits/stdc++.h>
using namespace std;

int main()
{
	int n,m,k;
	cin>>n>>m>>k;
	vector<int> a(n+1,0);
	int count=0,out=0;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
	}
	for(int i=0;i<n-m;i++)
	{
		for(int j=i;j<m+i;j++)
		{
			if(a[j]<k)
				count++;
				break;
		}
	}
	out=n-m-count;
	cout<<out;
	return 0;
}

填数游戏

小团和小美正在玩一个填数游戏，这个游戏是给一个等式，其中有一些数被挖掉了，你需要向其中填数字，使得等式成立。

比如 ___+12=34，那么横线填的一定是22

现在，这个游戏到了最后一关，这一关的等式很奇特：+++…+=n

这里可以填任意多个正整数（甚至可能是1个），只要这些数的和等于n即可。

但是，有一个额外的限制，填入的所有数必须小于等于k，大于等于1，填入的数的最大值必须大于等于d。

请你计算，有多少个不同的等式满足这些限制。由于答案可能很大，请将答案mod(998244353)后输出。

输入描述:

输入包含三个数n, k, d

输出描述：

输出包含一行，即方案数。

输入例子1:

5 3 2

输出例子1:

12

例子说明1:

2+3=5

3+2=5

1+1+3=5

1+3+1=5

3+1+1=5

1+2+2=5

2+1+2=5

2+2+1=5

1+1+1+2=5

1+1+2+1=5

1+2+1+1=5

2+1+1+1=5

共12种填法

思路：

动态规划，设dp[n][r]是把n做Composition且和数属于[1,r]范围的方案数目。

转移方程： d p [ n ] [ r ] = ∑ j ∈ A d p [ n − j ] [ r ] A = { 1 , 2 , . . . , r } dp[n][r]=\sum_{j∈A}dp[n-j][r] A=\{1,2,...,r\} dp[n][r]=j∈A∑​dp[n−j][r]A={1,2,...,r}

边界条件：$$dp[0][r]=1$$​

Ref：参考文章

#include<bits/stdc++.h>
#define mod 998244353
typedef long long ll;
using namespace std;
ll window=0;
ll dp[100005][105];
ll n,k,d;

void Calculatedp(ll n,ll k)
{
	dp[0][k]=1;
	window=0;
	for(ll i=1;i<=n;i++)
	{
	
		if(i>=k+1) 
		{
		  window=(window+dp[i-1][k]-dp[i-k-1][k]+mod)%mod;
		  dp[i][k]=window;
		}
		else 
		{
		  window=(window+dp[i-1][k])%mod;	
		  dp[i][k]=window;
		}
	}
}

int main()
{
	cin>>n>>k>>d;
	memset(dp,0,sizeof(dp));
	Calculatedp(n,k);
	Calculatedp(n,d-1);
	if(d<1||k<0||d>k) 
		cout<<"0"<<endl;
	else 
		cout<<(dp[n][k]+mod-dp[n][d-1])%mod<<endl;
	return 0;
}

照镜子

小团有一个n×m的矩阵A， 他知道这是小美用一种特殊的方法生成的，具体规则如下:

小美首先写下一个n’×m的矩阵，然后小美每一次将这个矩阵上下翻转后接到原矩阵的下方。小美重复这个过程若干次（甚至可能是0次，也就是没有进行过这一操作），然后将操作后的矩阵交给小团。

小团想知道，小美一开始写下的矩阵是什么。因为小美可能有多种一开始的矩阵，小团想得到最小的矩阵（这里的最小指矩阵即n’×m的面积最小）。

输入描述:

输入包含两个整数n,m，表示小团矩阵的大小。接下来n行，每行m个正整数，第i行第j列表示矩阵第i行第j列的数。

输出描述:

输出包含一个矩阵，一共n’行m列，表示小美一开始最小的矩阵。

输入例子1:

8 3

1 0 1

0 1 0

0 1 0

1 0 1

1 0 1

0 1 0

0 1 0

1 0 1

输出例子1:

1 0 1
0 1 0

例子说明1:

小美一开始的矩阵可能有以下3种：
1.

1 0 1
0 1 0
2.

1 0 1
0 1 0
0 1 0
1 0 1
3.

1 0 1
0 1 0
0 1 0
1 0 1
1 0 1
0 1 0
0 1 0
1 0 1

其中最小的矩阵为第一种。

思路：

采用评论区中零葬的思路：

(1) 原始矩阵是奇数行：那小美肯定就没有进行过操作，输出现在的矩阵就好。

(2) 原始矩阵是偶数行：先检查原始矩阵上半部分和下半部分是不是镜像对称的。如果是，矩阵规模就折半再检查（因为下半部分是翻转得到的，所以再检查上半部分），否则输出当前的矩阵。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n,m;
ll a[100005][55];

bool Judge(ll n){
	if(n%2) 
		return false;
	ll midle=n/2;
	for(ll i=1;i<=midle;i++){
		for(ll j=1;j<=m;j++){
			if(a[i][j]!=a[2*midle-i+1][j]) 
				return false;
		}
	}
	return true;
}


int main()
{
	cin>>n>>m;
	for(ll i=1;i<=n;i++)
	{
		for(ll j=1;j<=m;j++)
			cin>>a[i][j];
	}
	if(n%2) ;
	else 
	{
	  while(n)
	  {
		if(Judge(n)) n/=2;
		else break;
	  }	
	}
	
	for(ll i=1;i<=n;i++){
		for(ll j=1;j<=m;j++){
			if(j==1) ;else cout<<" ";
			cout<<a[i][j];
		}
		cout<<endl;
	}
	return 0;
}

密室闯关

小团和小美正在密室中解密。他们现在来到了一个新的关卡面前。这一关是一个配合关卡，有n个巨大的齿轮摆成一排，每个齿轮上有两个按钮和按顺时针排成一环的26个大写字母。在齿轮的最上面有一个孔，透过孔可以看到齿轮最上方的字母。

小团发现，每次他可以按住一个齿轮的一个按钮，小美就可以顺时针移动这个齿轮，使得孔里看到的字母变为其对应的下一个字母（比如A变为B，Y变为Z），并且如果小团按下的第一个按钮，则齿轮与上一个齿轮咬合，上一个齿轮的能看见的字母会变为其减1的字母（即B变为A，Z变为Y），进行这个操作的时候，不会影响上一个齿轮之前的齿轮。如果小团按下的第二个按钮，则下一个齿轮能看见的字母会变为其减1的字母，同样，这个操作不会影响下一个齿轮之后的齿轮。

如果这个齿轮是第一个齿轮，或者上一个齿轮的字母为A，小团按下第一个按钮后小美将不能移动。同理，如果这个齿轮是最后一个齿轮，或者下一个齿轮的字母为A，小团按下第二个按钮后小美将不能移动。

如果该齿轮上的字母是Z，该齿轮按下按钮后也不能移动。这个齿轮组的某个状态所组成的字符串将会是通关密码。

现在，小团想计算出可以变化出多少种齿轮的组合，他会依据这个数字来计算是否可以暴力计算出密码。请你帮助他。

输入描述:

输入包含多组数据。对于每组数据，将会输入两行。第一行会输入一个数n，代表齿轮的个数。接下来一行n个大写字母，代表每个齿轮最开始能看见的字母。

输出描述:

对于每组数据，输出一行一个数字，代表密码可能的组合数，对998244353取模。

输入例子1:

2

BB

3

ABA

输出例子1:

3
3

例子说明1:

对于样例1,有以下三种最终可能的组合：AC,BB,CA

对于样例2,有以下三种最终可能的组合：BAA,ABA,AAB

对于样例2，一开始的状态如图第一行所示，其中，红色箭头指向的是透过小孔看得见的字母。

小团选择第1个齿轮的第2个按钮按下，小美旋转后，会变为图第二行的状态

小团选择第3个齿轮的第1个按钮按下，小美旋转后，会变为图第三行的状态

思路：

参考评论区中JCoder的思路：

将 a-z 视为 0-25 的数字，只有两种操作：

(1) a[i] += 1 并且 a[i - 1] -= 1

(2) a[i] -= 1 并且 a[i - 1] += 1

前提是不能让 a[i] 小于 0 或者 大于 25.

无论执行多少次操作，都不会对整个串的权值和造成影响。

也就是状态 x 能转移到的 状态集合 s 具有权值和相同的特点。

为了统计 x 能达到多少个状态，也就是统计状态集合 s 的大小，同样也是统计 s 相对应的 sum 可能出现多少次 。

这个问题就变成了用 n 个 0-25 的数字，凑成和为 sum 的方案数。

和上述的填数问题类似，是一个 DP 问题。

代码：

#include <bits/stdc++.h>
#define mod 998244353
using namespace std;

int n;
long long dp[105][2505];
char a[100005];

void init_dp() {
    int maxn = 100, maxm = 25 * maxn;
    dp[0][0] = 1LL;
    for (int i = 1; i <= maxm; i++) 
        dp[0][i] = 0LL;
    for (int i = 1; i <= maxn; i++) 
	{
        dp[i][0] = 1LL;
        for (int j = 1; j <= maxm; j++) 
		{
            for (int k = 0; k < 26; k++) 
			{
                if (j - k >= 0) 
				{
                    dp[i][j] += dp[i - 1][j - k];
                    dp[i][j] %= mod;
                }
            }
        }
    }
}

int main() {
    init_dp();
    while(scanf("%d", &n) != EOF) 
	{
        scanf("%s", a);
        int sum = 0;
        for (int i = 0; i < n; i++) 
            sum += (a[i] - 'A');
        printf("%lld\n", dp[n][sum] % mod);
    }
    return 0;
}