【腾讯音乐0413】春招实习笔试练习

实习摸鱼、刷一刷题和八股。

1. 小红的好数组

小红定义一个数组为“好数组”，当且仅当该数组满足以下条件
1. 数组仅由0、1、2三种元素组成
2. 数组相邻的元素不想等

例如：1，2，1，2，0，1是好数组。小红定义一个数组的“陡峭值”为该数组相邻元素的差的绝对值之和。例如，[2,1,2,0,1]的陡峭值为|2-1|+|1-2|+|2-0|+|0-1|=5，小红想知道，长度为n的所有好数组的陡峭值之和是多少？由于答案过大请对10^9+7取模，数据范围2<=n<=10^9

示例1：

输入

2

输出

8

思路：dp思路比较好想，但是n的数据范围太大了，只能过很少一部分用例。原因出现求幂的过程中，应该使用快速幂的方式，逐步进行取模

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
using namespace std;
const int MOD = 10e9 + 7;

long long fastMi(long long x, int n, int m) {
	int res = 1;
	while (n > 0) {
		if (n % 2 == 1)	res = (res * x) % m;
		x = (x * x) % m;
		n /= 2;
	}
	return res;
}

int main() {
	long long n;
	cin >> n;
	vector<vector<long long>> tail(3, vector<long long>(n + 5, 0));
	tail[0][2] = 3;
	tail[1][2] = 2;
	tail[2][2] = 3;
	for (long long i = 3; i <= n; i++) {
		long long count = fastMi(2, i - 2, MOD);

		tail[0][i] = ((tail[1][i - 1] + count) % MOD + (tail[2][i - 1] + 2 * count) % MOD) % MOD;
		tail[1][i] = ((tail[0][i - 1] + count) % MOD + (tail[2][i - 1] + count) % MOD) % MOD;
		tail[2][i] = ((tail[1][i - 1] + count) % MOD + (tail[0][i - 1] + 2 * count) % MOD) % MOD;
	}
	cout << (tail[0][n] + tail[1][n] + tail[2][n]) % MOD;
	return 0;
}

 另外，也能通过打表找到规律，从而推导出函数直接进行求解。

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
using namespace std;
const int MOD = 10e9 + 7;

long long fastMi(long long x, int n, int m) {
	int res = 1;
	while (n > 0) {
		if (n % 2 == 1)	res = (res * x) % m;
		x = (x * x) % m;
		n /= 2;
	}
	return res;
}

int main() {
	long long n;
	cin >> n;
	long long ans = ((n - 1) * fastMi(2, n + 1, MOD)) % MOD;
	cout << ans;
	return 0;
}

2. 小红的二叉树

小红拿到了一颗二叉树，定义每个节点的价值为子树节点乘积的末尾0的数量。现在请你返回一颗二叉树，树的结构和给定二叉树相同，将每个节点的权值替换为该节点的价值。二叉树的节点不超过10^5。二叉树每个节点的权值都是不超过10^9的正整数。

示例：

输入

{2,5,10,#,#,4,5}

输出

{3,0,2,#,#,0,0}

思路：核心代码如下，首先二叉树权值过大，无法将所有子树节点相乘，因此计算子树内包含因子2和5的最小数量，则为包含尾数0的个数。剩下的就是简单的dfs，dfs从顶向上更新，不会影响父节点结果。

vector<int> counts(int n) {
	int count_2 = 0, count_5 = 0;
	int temp = n;
	while (temp % 2 == 0) {
		count_2++;
		temp /= 2;
	}
	temp = n;
	while (temp % 5 == 0) {
		count_5++;
		temp /= 5;
	}
	return { count_2, count_5 };
}

vector<int> dfs(TreeNode* root) {
	if (root == nullptr)	return { 0, 0 };
	vector<int> cur = counts(root->val);
	vector<int> l = dfs(root->left);
	vector<int> r = dfs(root->right);
	cur[0] += l[0] + r[0];
	cur[1] += l[1] + r[1];
	node->val = min(cur[0], cur[1]);
	return cur;
}

总结：题意不难思路也不难，难在如何处理大规模数据，尽量选择简单的办法先把数据规模降低或者进一步将问题转换为简单问题。简单的dp问题一般来说都可以打表找出规律。