牛客 字节跳动笔试题-[编程题] 手串 思路及代码 C语言

题目

首先想到的解题思路

1. 将数据保存到二维数组，二维数组保存的纯粹是串珠的数据，也就是输入数据第二行第二列起的数组。这个数组可以动态分配。
2. 对二维数组逐行扫描，即 i 从0到n-1。对于每一个i，都累加 i 到 i+m-1 行中各种颜色出现的次数。
3. 步骤2中所述次数为一个map矩阵，map[index] 表示颜色 index出现了map[index] 次。

对以上思路的改进

以上思路中，如果 m 是一个比较大的数，会存在大量的重复扫描，如扫描第i行时，i+1行会被遍历一次；等待扫描第i+1行时，i+1行又被扫描了一次。
对此，我们可以每次只将上一行 （i-1）行中的颜色计数删除，然后只将 最后一行（i+m-1行）扫描计数即可。
当然，扫描第一行的时候，必须从第一行一直扫描到m-1行。

实现

#include<stdio.h>
#include<malloc.h>

int main() {
	int n, m, c;
	while (~scanf("%d%d%d", &n, &m, &c)) {
		// 二维数组，保存n个串珠，每个串珠为一行
		int **data = (int **)calloc(n, sizeof(int *));
		// 一位数组，表示每一行的串珠都有几种颜色
		int *datasize = (int *)malloc(sizeof(int) * n);
		// 循环输入每一行
		for (int i = 0; i < n; i++) {
			// 输入一行的第一个元素
			scanf("%d", datasize + i);
			data[i] = (int *)malloc(sizeof(int) * datasize[i]);
			// 输入一行的每一个元素
			for (int j = 0; j < datasize[i]; j++) {
				scanf("%d", &data[i][j]);
			}
		}

		// map[x]的		
		int map[51] = { 0 };
		int iis = 0;
		// 开始循环判断
		for (int i = 0; i < n; i++) {
			// 删除前一行的统计
			if (i > 0) {
				for (int j = 0; j < datasize[i - 1]; j++) {
					// 如果是1就删除，是2或者0就保留
					map[data[i - 1][j]] = map[data[i - 1][j]] == 1 ? 0 : map[data[i - 1][j]];
				}
				
				// 把最后一行统计入内
				iis = i + m - 1;
				iis = iis > n ? (iis%n) : iis;
				for (int j = 0; j < datasize[iis]; j++) {
					map[data[iis][j]]++;
				}

			}
			else {
				for (int ii = i; ii < i + m; ii++) {
					// 把iis限制在0-（n-1）范围内
					iis = ii< n ? ii : ii%n;
					for (int j = 0; j < datasize[iis]; j++) {
						map[data[iis][j]]++;
					}
				}
			}
		}

		// 统计
		int res = 0;
		for (int i = 1; i < 51; i++) {
			if (map[i] > 1) {
				res++;
			}
		}
		printf("%d\n", res);

		// free很重要
		for (int i = 0; i < n; i++) {
			free(data[i]);
		}
		free(data);
		free(datasize);
	}
	return 0;
}