题目:
给定n个A和2n个B,用这些字符拼成一个字符串,要求这个串的所有前缀和后缀B的个数始终不少于A。
(一个字符串的前缀是只从开头到某个位置为止的子串,后缀是只从某个位置到结尾的子串)。
输入格式
多组数据,每组数据只有一行,包含一个正整数n。(n<=10^17)。
输出格式
每组数据输出一行,最终结果对99991取余数的结果。
分析:
简单的想法是建立一个二叉树,深度遍历一下即可。但是效率太低,对比较大的n不太适用,暂时没想到什么好办法,先简单的做做看。
代码:
<span style="font-size:18px;">#include <stdio.h> #include <stdlib.h> struct Node { int value; struct Node *left; struct Node *right; }; #define n 3 static int result = 0; static int index = 0; static int array[3 * n] = {0}; struct Node* initNode() { struct Node* temp = (struct Node *)malloc(sizeof(struct Node)); temp->value = 0; temp->left = NULL; temp->right = NULL; return temp; } void creatTree(struct Node* parent, int a, int b) { if (a > 0) { struct Node* left = initNode(); left->value = -1; parent->left = left; creatTree(left, a - 1, b); } if (b > 0) { struct Node* right = initNode(); right->value = 1; parent->right = right; creatTree(right, a, b - 1); } if (a <= 0 && b <=0) { return; } } //static int array[100] = {0}; void dfs(struct Node* root, int sum) { int i = 0; if (root == NULL) { //result++; return; } if (root->value == 1 && root->left == NULL && root->right == NULL) { result++; printf("%d\n", root->value); printf("\nget one\n"); return; } sum += root->value; if (sum < 0 || sum > n) { return; } else { printf("%d\n", root->value); //index = 0; dfs(root->left, sum); dfs(root->right, sum); } } void freeNode(struct Node* root) { if (root != NULL) { //printf("%d\n", root->value); //printf("%d\n", root->left->value); freeNode(root->left); freeNode(root->right); free(root); } return; } int main() { int a = 0, b = 0, sum = 0; a = n; b = n * 2; /* 根节点 B */ struct Node *root = initNode(); root->value = 1; /* 建立二叉树 */ creatTree(root, a, b-1); /* 深度搜索遍历法得到结果 */ dfs(root, sum); printf("result = %d\n", result); /* 释放内存 */ freeNode(root); return 0; } </span>
AB串
时间: 2024-10-09 22:28:02