【bzoj5110】[CodePlus2017]Yazid 的新生舞会 Treap
题目描述
输入
输出
样例输入
5 0
1 1 2 2 3
样例输出
10
题解
Treap
做麻烦了...
先把所有相同的数出现的位置存到一个vector里,然后考虑这个数作为众数的贡献。
考虑枚举贡献段的最后一个数的位置,这样对于某个左边的该数出现的位置,区间长度应该满足:区间长度小于2*这段中该数的个数。
如果设 x 表示右端点位置, y 表示右端点在某位置下满足条件的区间个数。那么由于区间长度一定,可以得到一个 y=-x+b 类型的直线(实际上有意义的只有第一象限的部分)。
但是这样会有一个问题:某些区间会计算重复。
由于考虑的是这段该数出现位置的贡献,因此区间不能包含其它的该数,即贡献区间的最端点不能达到上一个出现位置。
因此可以得到如下的图:
(好,以上只是理论部分...)
考虑一下我们需要做的:
我们枚举该数出现位置的右端点,并计算 [上一个出现的位置,当前位置) 区间的贡献。我们维护所有前面的这样的折线,统计所有折线在这段区间的面积和。
然后,由于多了当前位置,相当于以前的区间中选择区间的长度加了2,因此这些折线需要向右平移2个单位。
最后,需要把当前位置对应的折线维护起来。
一条折线可以看作是两条直线作差,因此我们需要做的有:添加直线、维护直线面积、支持直线集体向右平移。可以使用Treap来维护。
按照直线与x轴交点来维护,维护直线的斜率、截距、面积,以及子树中这三者的和。插入直接正常插入,整体平移则直接打标记。
统计面积的过程可以转化为前缀相减。在Treap上查询,如果其与左边的位置都小于当前位置,那么左边的答案就是面积总和,递归右子树;否则右边的贡献都是梯形面积(转化为等差数列求和),递归左子树。
然后就可以了,注意开long long。
细节贼多...
时间复杂度$O(n\log n)$
然后我考试时写的SBT,由于没有在旋转时pushdown(震惊!SBT竟然要在旋转时pushdown)导致爆0,再也不写SBT了...
附上Treap的代码:
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 500010
using namespace std;
typedef long long ll;
vector<int> v[N];
int l[N] , r[N] , rnd[N] , tot , root , a[N];
ll vp[N] , tag[N] , wk[N] , wb[N] , ws[N] , sk[N] , sb[N] , ss[N];
inline void add(int k , ll x)
{
vp[k] += x;
ws[k] += (2 * wb[k] - (x + 1) * wk[k]) * x / 2;
ss[k] += (2 * sb[k] - (x + 1) * sk[k]) * x / 2;
wb[k] -= wk[k] * x;
sb[k] -= sk[k] * x;
tag[k] += x;
}
inline void pushup(int k)
{
sk[k] = sk[l[k]] + sk[r[k]] + wk[k];
sb[k] = sb[l[k]] + sb[r[k]] + wb[k];
ss[k] = ss[l[k]] + ss[r[k]] + ws[k];
}
inline void pushdown(int k)
{
if(tag[k]) add(l[k] , tag[k]) , add(r[k] , tag[k]) , tag[k] = 0;
}
inline void zig(int &k)
{
int t = l[k];
l[k] = r[t] , r[t] = k , pushup(k) , pushup(t) , k = t;
}
inline void zag(int &k)
{
int t = r[k];
r[k] = l[t] , l[t] = k , pushup(k) , pushup(t) , k = t;
}
void insert(int &k , ll p , ll xk , ll xb)
{
if(!k)
{
k = ++tot , vp[k] = p , wk[k] = sk[k] = xk , wb[k] = sb[k] = xb , ws[k] = ss[k] = xb * (p + 1) / 2 , rnd[k] = rand();
return;
}
pushdown(k);
if(p < vp[k])
{
insert(l[k] , p , xk , xb) , pushup(k);
if(rnd[l[k]] < rnd[k]) zig(k);
}
else
{
insert(r[k] , p , xk , xb) , pushup(k);
if(rnd[r[k]] < rnd[k]) zag(k);
}
}
ll query(int k , ll p)
{
if(!k) return 0;
pushdown(k);
if(p < vp[k]) return (2 * (wb[k] + sb[r[k]]) + (wk[k] + sk[r[k]]) * p) * (p + 1) / 2 + query(l[k] , p);
else return ws[k] + ss[l[k]] + query(r[k] , p);
}
void clear(int k)
{
if(!k) return;
clear(l[k]) , clear(r[k]) , l[k] = r[k] = vp[k] = tag[k] = wk[k] = wb[k] = ws[k] = sk[k] = sb[k] = ss[k] = 0;
}
int main()
{
int n , i , pos;
ll ans = 0;
unsigned j;
scanf("%d%*d" , &n);
for(i = 1 ; i <= n ; i ++ ) scanf("%d" , &a[i]) , v[++a[i]].push_back(i);
for(i = 1 ; i <= n ; i ++ ) v[i].push_back(n + 1);
for(i = 1 ; i <= n ; i ++ )
{
pos = 0;
for(j = 0 ; j < v[i].size() ; j ++ )
{
ans += query(root , v[i][j] - 1) - query(root , pos - 1);
add(root , 2);
insert(root , v[i][j] + 1 , -1 , v[i][j] + 1) , insert(root , pos + 1 , 1 , -pos - 1);
pos = v[i][j];
}
clear(root) , root = tot = 0;
}
printf("%lld\n" , ans);
return 0;
}
【bzoj5110】[CodePlus2017]Yazid 的新生舞会 Treap的更多相关文章
- bzoj5110: [CodePlus2017]Yazid 的新生舞会
Description Yazid有一个长度为n的序列A,下标从1至n.显然地,这个序列共有n(n+1)/2个子区间.对于任意一个子区间[l,r] ,如果该子区间内的众数在该子区间的出现次数严格大于( ...
- 【BZOJ5110】[CodePlus2017]Yazid 的新生舞会 线段树
[BZOJ5110][CodePlus2017]Yazid 的新生舞会 Description Yazid有一个长度为n的序列A,下标从1至n.显然地,这个序列共有n(n+1)/2个子区间.对于任意一 ...
- 【bzoj5110】Yazid的新生舞会
这里是 $THUWC$ 选拔时间 模拟赛的时候犯 $SB$ 了,写了所有的部分分,然后直接跑过了 $4$ 个大样例(一个大样例是一个特殊情况)…… 我还以为我非常叼,部分分都写对了,于是就不管了…… ...
- 【BZOJ5110】[CodePlus2017]Yazid 的新生舞会
题解: 没笔的时候我想了一下 发现如果不是出现一半次数而是k次,并不太会做 然后我用前缀和写了一下发现就是维护一个不等式: 于是就可以随便维护了
- BZOJ.5110.[CodePlus2017]Yazid 的新生舞会(线段树/树状数组/分治)
LOJ BZOJ 洛谷 又来发良心题解啦 \(Description\) 给定一个序列\(A_i\).求有多少个子区间,满足该区间众数出现次数大于区间长度的一半. \(n\leq5\times10^5 ...
- [loj 6253] Yazid的新生舞会
(很久之前刷的题现在看起来十分陌生a) 题意: 给你一个长度为n的序列A,定义一个区间$[l,r]$是“新生舞会的”当且仅当该区间的众数次数严格大于$\frac{r-l+1}{2}$,求有多少子区间是 ...
- [BZOJ5110]Yazid的新生舞会
题目大意: 给你一个长度为$n(n\leq 5\times 10^5)$的序列$A_{1\sim n}$.求满足区间众数在区间内出现次数严格大于$\lfloor\frac{r-l+1}{2}\rflo ...
- 「CodePlus 2017 11 月赛」Yazid 的新生舞会(树状数组/线段树)
学习了新姿势..(一直看不懂大爷的代码卡了好久T T 首先数字范围那么小可以考虑枚举众数来计算答案,设当前枚举到$x$,$s_i$为前$i$个数中$x$的出现次数,则满足$2*s_r-r > 2 ...
- bzoj 5110 Yazid的新生舞会
题目大意: 一个数列,求有多少个区间$[l,r]$满足该区间的众数出现次数大于$\lceil \frac{r-l}{2} \rceil$ 思路: 对于一个区间满足条件的众数明显是唯一的 所以设该数的前 ...
随机推荐
- PHP实现识别带emoji表情的字符串
function have_special_char($str) { $length = mb_strlen($str); $array = []; for ($i=0; $i<$length; ...
- tcpdump使用
1. tcpdump选项 它的命令格式为: tcpdump [ -DenNqvX ] [ -c count ] [ -F file ] [ -i interface ] [ -r file ][ -s ...
- Win10 远程服务器版
朋友的电脑刚装了1803版的Win10,然后他用KMS_VL_ALL6.9激活了一下,竟然变成了一个奇怪的版本:“远程服务器版”!第一次见这玩意,还真稀罕.帮他研究了一下,发现KMS_VL_ALL在激 ...
- libevent学习二(Working with an event loop)
Runing the loop #define EVLOOP_ONCE 0x01 #define EVLOOP_NONBLOCK 0x02 #define EV ...
- java对于Redis中jedis的操作
package com.answer.redis; import java.util.HashMap; import java.util.List; import java.util.Map; imp ...
- 【isJson( jsonObj )】判断是否是JSON实例
判断是否是JSON实例: 原型:isJson( jsonObj ) 说明:判断对象是否是JSON实例 返回:[true | false] 示例: <% Set jsonObj1 = toJson ...
- springMVC怎么改变form的提交方式为put或者delete
想着练习一下创建restful风格的网站呢,结果发现在jsp页面上并不能灵活使用put和delete提交方式.下面我的解决办法 一. form 只支持post和get两种提交方式,只支持get提交方式 ...
- 创新手机游戏《3L》开发点滴(3)——道具、物品、装备表设计 V2(最终版)
我们正在开发一款新手游,里面有道具,之前也写了一篇博文记录了下我们的设计思路,但是国庆到了,于是我有了时间继续纠结这个问题. 其实我主要是在到底是用mysql还是mongodb上纠结.这个复杂.痛苦. ...
- 十 Writing YARN Applications
本节介绍: 使用yarn 高级提交写yarn应用程序.其实已经yarn底层API.MR计算框架对底层的API实现了封装. 高级提交指直接使用yarn的三种接口来提交应用程序: 1)YarnCl ...
- ViewPager的简单使用说明
前提:工程中使用ViewPager,需要导入google提供的jar包(android-support-v4.jar). 要学习ViewPager的使用,建议直接看官方文档 Creating Swip ...