PTA甲级模拟第一弹:1132-1135

pta主要是打基础的哈,练完了考虑去刷leetcode,希望尽快练完叭~

本次模拟题目：甲级1132-1135
失分点：1133和1135

1. 1133考题

1133 Splitting A Linked List (25 分)
22/25 模拟得分

#include<bits/stdc++.h>
using namespace std;
//负数 0-k 其他
const int maxn=1000000;
struct Node{
	int addr;
	int value;
	int next=-1;
}nodes[maxn];
int n,k,head;
void show(){
	int p=head;
	while(p!=-1){
		printf("%05d %d %05d",nodes[p].addr,nodes[p].value,nodes[p].next);
		//顺序 
		p=nodes[p].next;
		printf("\n");
	}
}
int addr,value,nextnode;
//下面主要是重新排序
struct newNode{
	int id;
	int addr;
	int value;
	int next=-1;
	int kind;
}; 
vector<newNode>newnodes; 
bool cmp(newNode a,newNode b){
	if(a.kind!=b.kind){
		return a.kind<b.kind;
	}else{
		return a.id<b.id;
	}
}
void convert(){
	int p=head;
	int cnt=0;
	while(p!=-1){
		cnt++;
		newNode temp;
		temp.id=cnt;temp.addr=nodes[p].addr;temp.value=nodes[p].value;
		if(nodes[p].value<0)temp.kind=-1;
		else if(nodes[p].value<=k)temp.kind=0;
		else temp.kind=1;
		p=nodes[p].next;
		newnodes.push_back(temp);
	}
}
//22不清楚哪个细节没有扣到 
int main(){
	scanf("%d%d%d",&head,&n,&k);
	for(int i=0;i<n;i++){
		scanf("%d%d%d",&addr,&value,&nextnode);
		nodes[addr].addr=addr;
		nodes[addr].value=value;
		nodes[addr].next=nextnode;
	}
	//printf("\n");
	convert();
	sort(newnodes.begin(),newnodes.end(),cmp);
	//下面重新编辑地址
	for(int i=0;i<n;i++){
		printf("%05d %d ",newnodes[i].addr,newnodes[i].value);
		if(i!=n-1){
			printf("%05d\n",newnodes[i+1].addr);
		}else{
			printf("-1");
		}
	}
	return 0;
}

订正思路

利用vector本身的性质，由题意可知将list分成三段重新排序，那么第一步是按照list顺序重新输出list保存，注意，保存的时候调用三段list，这样之后合并就可以按照自己想要的顺序。😍

#include<bits/stdc++.h>
using namespace std;
const int maxn=101100;
struct Node{
	int addr;
	int value;
	int next;
}nodes[maxn];
int head,n,k;
int addr,value,nextid;
vector<Node>v1,v2,v3;
int main(){
	scanf("%d%d%d",&head,&n,&k);
	for(int i=0;i<n;i++){
		scanf("%d%d%d",&addr,&value,&nextid);
		nodes[addr].addr=addr;
		nodes[addr].value=value;
		nodes[addr].next=nextid;
	}
	int p=head;
	while(p!=-1){
		if(nodes[p].value<0)v1.push_back(nodes[p]);
		else if(nodes[p].value<=k)v2.push_back(nodes[p]);
		else v3.push_back(nodes[p]);
		p=nodes[p].next;
	}
	v1.insert(v1.end(),v2.begin(),v2.end());
	v1.insert(v1.end(),v3.begin(),v3.end());
	int len=v1.size();
	for(int i=0;i<len;i++){
		if(i!=len-1){
			printf("%05d %d %05d\n",v1[i].addr,v1[i].value,v1[i+1].addr);
		}else{
			printf("%05d %d -1",v1[i].addr,v1[i].value);
		}
	}
	return 0;
} 

2. 1135红黑树

1135 Is It A Red-Black Tree (30 分)
诚实来讲,不看任何资料的话,应该拿不到10+,悄咪咪看了模板才勉强上20

模板的话,涉及到中序遍历+()遍历的树结构的重构问题

21/30 模拟得分

#include<bits/stdc++.h>
using namespace std;
//红黑树的判别 :21分 
int k; 
int n;
vector<int>nodes;//前序遍历
vector<int>midnodes;//中序遍历
struct Node{
	int value;
	Node* lchild;
	Node* rchild;
}; 
Node* Create(int posL,int posR,int inL,int inR){
	//中序遍历和前序遍历确定二叉树
	if(posL>posR){
		return NULL;//递归边界 
	}
	Node * root=new Node;
	//根节点是前序遍历的第一个结点
	root->value=nodes[posL];
	
	int l;
	for(l=inL;l<=inR;l++){
		if(root->value==midnodes[l]){
			break;
		}
	} 
	//中序：左子树元素个数
	int left_num=l-inL;
	root->lchild=Create(posL+1,posL+left_num,inL,l-1);//left
	root->rchild=Create(posL+left_num+1,posR,l+1,inR);//right
	return root;
}
//前序遍历
int PreTravel(Node *root) {
	if(root==NULL)return 1;
	if(root->value<0&&(root->lchild!=NULL&&root->lchild->value<0||root->rchild!=NULL&&root->rchild->value<0)){
		return 0;
	}else if(root->lchild!=NULL&&root->rchild!=NULL&&root->lchild->value*root->rchild->value<0){
		return 0;
	}
	else return PreTravel(root->lchild)*PreTravel(root->rchild);
}

bool cmp(int a,int b){
	return abs(a)<abs(b);
} 
bool isok(Node* root){
	//判断是否为红黑树
	if(root==NULL)return true;
	//(1)root是黑色的 
	if(root->value<0) {
		return false;
	}
	//遍历：非黑即红，红色结点的两个孩子都是黑色的
	if(PreTravel(root)==0)return false;
	//对于每一个结点，从该节点到比它小的所有叶子结点所经过的黑色结点都相同 
	return true;
} 
int main(){
	scanf("%d",&k);
	while(k--){
		scanf("%d",&n);
		nodes.resize(n);
		midnodes.resize(n);
		for(int i=0;i<n;i++){
			scanf("%d",&nodes[i]);
			midnodes[i]=nodes[i];
		}
		sort(midnodes.begin(),midnodes.end(),cmp);
		//首先建立一棵树，
		Node*root=Create(0,n-1,0,n-1);
		PreTravel(root);
		//再判断是否为红黑树 
		if(isok(root)){
			printf("Yes");
		}else{
			printf("No");
		}
		if(k)printf("\n");
	}
	
	return 0;
}

订正思路

前序遍历不需要结合中序遍历求解,直接一边输入一遍插入建树,判断的方法这边首先判断是否存在红节点的子节点中出现红节点(非法),接着判断到达每一个结点时,该结点到叶子结点路线上的黑节点是否相等,不相等则判定为非法

参考了柳神的代码,觉得自己思路好冗余😓

#include<bits/stdc++.h>
using namespace std;
//首先定义树结点
struct Node{
	int value;
	Node *lchild;
	Node *rchild;
}; 
vector<int>vec;
int k,n;
//value<0 红色结点
Node *build(Node *root,int value){
	if(root==NULL){
		root=new Node();
		root->value=value;
		root->lchild=root->rchild=NULL;
	}else if(abs(value)<abs(root->value)){
		root->lchild=build(root->lchild,value);
	}else{
		root->rchild=build(root->rchild,value);
	}
	return root;
} 
bool isok1(Node*root){
	if(root==NULL)return true;
	if(root->value<0){
		if(root->rchild!=NULL&&root->rchild->value<0)return false;
		if(root->lchild!=NULL&&root->lchild->value<0)return false;
	}
	return isok1(root->lchild)&&isok1(root->rchild);
}
int getNum(Node*root){
	if(root==NULL)return 0;
	int l=getNum(root->lchild);
	int r=getNum(root->rchild);
	return root->value>0?max(l,r)+1:max(l,r); 
}
bool isok2(Node*root){
	//一个结点到叶子结点经过的黑色节点个数相同
	if(root==NULL)return true;
	int l=getNum(root->lchild);
	int r=getNum(root->rchild);
	if(l!=r)return false;
	return isok2(root->lchild)&&isok2(root->rchild);
}

int main(){
	scanf("%d",&k);
	for(int i=0;i<k;i++){
		scanf("%d",&n);
		vec.resize(n);
		Node*root=NULL;
		for(int j=0;j<n;j++){
			scanf("%d",&vec[j]);
			root=build(root,vec[j]);//插入建树 
		}
		if(vec[0]<0||isok1(root)==false||isok2(root)==false){
			printf("No\n");
		}else{
			printf("Yes\n");
		}
	}
	return 0;
}