Codeforces Round #826 (Div. 3)（D. Masha and a Beautiful Tree）线段树解法

题目大意：输入包含 t ( 1 ≤ t ≤ 1e4 ) 组测试样例，每组样例先输入 m 大小的数组，构建一棵满二叉树。问是否可以通过交换根节点的子树让序列变成升序。

思路

我们在树的每个节点记录下区间的最小值 v 和 递归的方向 d 。规定向左递归为 0，向右为 1。

当序列本身就是一个升序序列时，每个节点的 d 都会是 0 ，也就是说如果你要询问每个点，一定是先找左子树再找右子树。

​​​​

那么 d==1 的地方一定是需要执行一次交换的点 ，加上 1 次即可，当L==R时，把当前值push 进一个数组里，代表排序后的顺序，最后再验证一遍顺序是否符合即可。

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string>
#include <vector>
#include <string.h>
using namespace std;
int  t;
int n;
const int LL=3e5+5;
int arr[LL];
vector<int>vt;
struct node{
	int v,d;//最小值，方向
}w[LL<<2];

void pushup(int u){
	w[u].v=min(w[u<<1].v,w[u<<1|1].v);
	if(w[u<<1].v<w[u<<1|1].v){
		w[u].d=0;
	}
	else w[u].d=1;
}

void build(int u,int L,int R){
	if(L==R){
		w[u].v=arr[L];w[u].d=0;return;
	}
	int mid=L+R>>1;
	build(u<<1,L,mid);build(u<<1|1,mid+1,R);
	pushup(u);
}

int query(int u,int L,int R){
	int res=0;
	res+=w[u].d;
	if(L==R){
		vt.push_back(w[u].v);//把当前值放进去，最后查看是否符合顺序
		return res;
	}
	int mid=L+R>>1;
	if(w[u].d==0){
		return res+=query(u<<1,L,mid)+query(u<<1|1,mid+1,R);
	}
	else{
		return res+=query(u<<1|1,mid+1,R)+query(u<<1,L,mid);
	}
}

int main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		for(int i=1;i<=n;i++){
			scanf("%d",&arr[i]);
		}
		build(1,1,n);
		int flg=0;//等于1说明不可以，输出-1
		int ans=query(1,1,n);//交换的次数
		int lens=vt.size();
		for(int i=0;i<lens;i++){//查看是否符合顺序
			if(vt[i]!=i+1){
				flg=1;break;
			}
		}
		if(flg)printf("-1\n");
		else printf("%d\n",ans);
		vt.clear();
	}


	//system("pause");
	return 0;
}

这道题解法挺多的www