社区讨论

求助替罪羊树

学术版参与者 1已保存回复 0

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
0 条
当前快照
1 份
快照标识符
@lo8x3d2n
此快照首次捕获于
2023/10/28 01:58
2 年前
此快照最后确认于
2023/10/28 01:58
2 年前
查看原帖
如何判断rank和kth输入数据不合法
CPP
//我的写法
int getRank(int x,int v)
{  
	if(x == 0)	return 0;
	if(v == tr[x].dat)
	{
		fl = 1;
		return tr[lc(x)].sz+1;
	}
	if(v > tr[x].dat) return getRank(rc(x),v)+tr[lc(x)].sz+tr[x].cnt; 
	if(v < tr[x].dat) return getRank(lc(x),v);
}
int getKth(int x, int k) 
{
  if(x == 0) return -1;
  if (tr[lc(x)].sum + 1 <= k && k <= tr[lc(x)].sum + tr[x].sz )
  	return tr[x].dat;
  if (k > tr[lc(x)].sum)
  	return getKth(rc(x),k - tr[lc(x)].sum - tr[x].cnt);
  else
  	return getKth(lc(x),k);
  
}
int main()
{
	int n;
	cin >> n;
	for(int i = 1;i <= n;i++)
	{
		char op;
		int r;
		cin >> op;
		cin >> r;
		if(op == 'I') insert(root,r);
		if(op == 'R')
		{
			int t = getRank(root,r);
			if(!fl)
				cout << -1 << endl;
			else
				cout << t << endl;
			fl = 0;
		}
		else if(op == 'K')
		{
			cout << getKth(root,r) << endl;
		}
	}
}

回复

0 条回复,欢迎继续交流。

正在加载回复...