方块栈游戏，“算法训练营”--陈小玉著

最新推荐文章于 2024-05-30 20:15:33 发布

微分生活，积分梦想！

最新推荐文章于 2024-05-30 20:15:33 发布

阅读量737

点赞数

文章标签：算法数据结构 matlab

本文链接：https://blog.csdn.net/qq_48080245/article/details/127184935

版权

华子正在玩方块游戏，方块编号为1~N，开始时每个方块都相当于一个栈。华子执行P个操作，操作有两种类型：第一种：（0，X），查询X方块下有多少个方块。第二种：（X,Y），将含有X方块的栈整体移动到含有Y方块栈的顶部。请对于华子每个第一种操作，都输出统计结果。

函数文件：

function y=findss_(fa,x)%寻找祖宗法
if x~=fa(x)
y=findss_(fa,fa(x));%函数嵌套
else
y=fa(x);
end
end

脚本文件：

clc
clear all
N=input('请输入方块总数：');
fa=zeros(1,N);
d=zeros(1,N);%某一方块下面的方块个数
cnt=zeros(1,N);%含有该方块的栈中总方块个数
for i=1:N%初始化
fa(i)=i;
d(i)=0;
cnt(i)=1;
end
p=input('请输入总操作数：');
A=zeros(2,p);%存储操作，第一种操作：将X移到Y上，X为上层，Y为下层；
A(1,:)=input('请输入上层：');%第二种操作：查某一方块下方块个数，默认0 为其上层，该方块编号为下层；
A(2,:)=input('请输入下层：');
for i=1:p
if A(1,i)~=0
a=findss_(fa,A(1,i));b=findss_(fa,A(2,i));%寻找该方块祖宗
fa(a)=b;%调整相关信息
d(a)=cnt(b);
cnt(b)=cnt(b)+cnt(a);
else
d(A(2,i))=d(A(2,i))+d(fa(A(2,i)));%此处只需运用父节点
disp(d(A(2,i)));
end
end