2019河北省大学生程序设计竞赛E.Paper Plane Fly Away

这玩意谁能看出来是树状数组来,好吧我承认我是菜鸡。

题意:现在有n nn个男孩从1 11到n nn按照左右顺序坐成一排,同样有n nn个女孩坐在男孩的前面,第i ii个男孩前面坐的是编号a[i] a[i]a[i]的女孩。每个男孩都有一个喜欢的女孩,他要做一个情书飞机飞给喜欢的女孩,但是如果飞机的轨迹在空中相交飞机就会相互碰撞,现在你需要得到每个男孩的飞机要和多少个其他男孩的飞机相撞。

问题抽象成一个匹配好的二分图,求每条边和多少条边相交。考虑第i个男孩和第j个女孩配对,那么对于i左边的男孩中,目标女孩的编号大于j的会和它相交,i右边的男孩中目标女孩编号小于j会和它相交。从左到右,从右到左分别树状数组存一下出现过的目标女孩编号就可以算出上述两个东西。copy这位大佬的_ 泛白

#include<bits/stdc++.h>
#define lowbit(x) (x&-x)
using namespace std;
const int maxn = 1e5 + 50;
int a[maxn];
void add(int i, int x){
	while(i < maxn) a[i] += x, i += lowbit(i);
}
int sum(int i){
	int res = 0;
	while(i) res += a[i], i -= lowbit(i);
	return res;
}
int to[maxn];
int id[maxn];
int ans[maxn];
int main(){
	int n;cin>>n;
	for(int i = 1; i <= n ; ++i){
		int c;
		scanf("%d%d", &c, &to[i]);
		c -= n;
		id[c] = i;
		to[i] -= n;
	}
	for(int i = 1; i <= n; ++i) to[i] = id[to[i]];
	for(int i = 1; i <= n; ++i){
		add(to[i], 1);
		ans[i] += sum(n+1) - sum(to[i]);
	}
	memset(a, 0, sizeof a);
	for(int i = n; i > 0; --i){
		ans[i] += sum(to[i]);
		add(to[i], 1);
	}
	for(int i = 1; i <= n; ++i) printf("%d\n", ans[i]);
}
发布了183 篇原创文章 · 获赞 31 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/mlm5678/article/details/91910305
今日推荐