【bzoj5055】膜法师 离散化+树状数组
2024-10-19 14:28:16
题目描述
给定一个序列$a$,求满足$i<j<k$且$a_i<a_j<a_k$的三元组$(i,j,k)$的个数。
输入
第一行1个数 n
第二行n个数 a_i
输出
一个数,表示能为长者续几秒,由于长者是不朽的,
所以能活很久,不妨将答案对**19260817**取模吧
样例输入
4
1 2 3 4
样例输出
50
题解
离散化+树状数组
枚举中间的数,然后只需要求出左边小于它的数的个数和右边大于它的数的个数即可。
离散化后对于左边和右边分别建立权值树状数组,然后随着指针移动插入/删除元素即可。
时间复杂度$O(n\log n)$
#include <cstdio>
#include <algorithm>
#define N 300010
#define mod 19260817
using namespace std;
int n , a[N] , v[N];
struct bit
{
int f[N];
inline void add(int x , int a)
{
int i;
for(i = x ; i <= n ; i += i & -i) f[i] = (f[i] + a + mod) % mod;
}
inline int query(int x)
{
int i , ans = 0;
for(i = x ; i ; i -= i & -i) ans = (ans + f[i]) % mod;
return ans;
}
}A , B;
int main()
{
int i , ans = 0;
scanf("%d" , &n);
for(i = 1 ; i <= n ; i ++ ) scanf("%d" , &a[i]) , v[i] = a[i];
sort(v + 1 , v + n + 1);
for(i = 1 ; i <= n ; i ++ ) a[i] = lower_bound(v + 1 , v + n + 1 , a[i]) - v;
for(i = 1 ; i <= n ; i ++ ) v[i] %= mod;
for(i = 2 ; i <= n ; i ++ ) B.add(n - a[i] + 1 , v[a[i]]);
for(i = 2 ; i <= n ; i ++ ) B.add(n - a[i] + 1 , -v[a[i]]) , A.add(a[i - 1] , v[a[i - 1]]) , ans = (ans + (long long)A.query(a[i] - 1) * B.query(n - a[i]) % mod * v[a[i]]) % mod;
printf("%d\n" , ans);
return 0;
}
最新文章
- PDB调试Python程序
- mysql安装出现error Nr.1045 (转)
- Stanford机器学习---第六讲. 怎样选择机器学习方法、系统
- XCode属性面板使用说明
- linq.js
- linux 配置 Samba 服务器实现文件共享
- 蓝桥杯--Quadratic Equation
- 【转】Linux(ubuntu14.04)上编译Android4.4源码的环境搭建及编译全过程
- Appium移动自动化测试(三)--安装Android模拟器(转)
- Python 实现 Html 转 Markdown(支持 MathJax 数学公式)
- linux系统mysql主从配置
- Eslint使用(webpack中使用)
- Keras.applications.models权重:存储路径及加载
- [C语言]使用函数
- [转]KMP算法
- PAT 1029 Median[求中位数][难]
- Spring单例 和 Scope注解
- RSA 每次公钥加密不同结果
- ArchLinux 下 OpenSSH 高级运用
- Unity自动打包工具