HDU 4288线段树开动脑筋
2016-04-06 14:00
501 查看
点击打开链接
题意:有几种操作,add加入一个数,del删除一个数,问所有下标%5==3的数的和
思路:看题就知道是线段树,又有加入又有删除又有求和的,线段树无疑,可是问的是下标%5==3的数的和,我们可以令开一个数组cnt,记录区间数字的个数,这样我们开一个num数组,二维的,后面的[5]分别记录区间的下标%5==0,1,2,3,4的数的和,在更新的时候,如果要求num[][3]的值,则左儿子的[3]加上右儿子那部分,右儿子怎么算呢,因为左儿子的数字个数已经知道了cnt嘛,这样我们可以通过它求右儿子的,但不一定是[3]的了,如果cnt==4,则右儿子应该为[4]的值,画一画一目了然,对了,这题数太大,所以要离散化,比较简单#include <math.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=100010;
ll num[maxn*4][5];
int a[maxn],n,m,cnt[maxn*4];
struct edge{
char c;
int x;
}q[maxn];
void pushup(int node){
cnt[node]=cnt[node<<1]+cnt[node<<1|1];
for(int i=0;i<5;i++){
num[node][i]=num[node<<1][i]+num[node<<1|1][(i-cnt[node<<1]%5+5)%5];
}
}
void add(int pos,int val,int le,int ri,int node){
if(le==ri){
cnt[node]=1;
num[node][1]=val;
return ;
}
int t=(le+ri)>>1;
if(pos<=t) add(pos,val,le,t,node<<1);
else add(pos,val,t+1,ri,node<<1|1);
pushup(node);
}
void del(int pos,int le,int ri,int node){
if(le==ri){
num[node][1]=cnt[node]=0;
return ;
}
int t=(le+ri)>>1;
if(pos<=t) del(pos,le,t,node<<1);
else del(pos,t+1,ri,node<<1|1);
pushup(node);
}
int main(){
char str[10];
while(scanf("%d",&n)!=-1){
m=0;
for(int i=0;i<n;i++){
scanf(" %s",str);
q[i].c=str[0];
if(str[0]!='s'){
scanf("%d",&q[i].x);
a[m++]=q[i].x;
}
}
sort(a,a+m);
m=unique(a,a+m)-a;
memset(cnt,0,sizeof(cnt));
memset(num,0,sizeof(num));
int pos;
for(int i=0;i<n;i++){
if(q[i].c=='s') printf("%I64d\n",num[1][3]);
else{
pos=lower_bound(a,a+m,q[i].x)-a+1;
if(q[i].c=='a') add(pos,q[i].x,1,m,1);
else del(pos,1,m,1);
}
}
}
return 0;
}
题意:有几种操作,add加入一个数,del删除一个数,问所有下标%5==3的数的和
思路:看题就知道是线段树,又有加入又有删除又有求和的,线段树无疑,可是问的是下标%5==3的数的和,我们可以令开一个数组cnt,记录区间数字的个数,这样我们开一个num数组,二维的,后面的[5]分别记录区间的下标%5==0,1,2,3,4的数的和,在更新的时候,如果要求num[][3]的值,则左儿子的[3]加上右儿子那部分,右儿子怎么算呢,因为左儿子的数字个数已经知道了cnt嘛,这样我们可以通过它求右儿子的,但不一定是[3]的了,如果cnt==4,则右儿子应该为[4]的值,画一画一目了然,对了,这题数太大,所以要离散化,比较简单#include <math.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=100010;
ll num[maxn*4][5];
int a[maxn],n,m,cnt[maxn*4];
struct edge{
char c;
int x;
}q[maxn];
void pushup(int node){
cnt[node]=cnt[node<<1]+cnt[node<<1|1];
for(int i=0;i<5;i++){
num[node][i]=num[node<<1][i]+num[node<<1|1][(i-cnt[node<<1]%5+5)%5];
}
}
void add(int pos,int val,int le,int ri,int node){
if(le==ri){
cnt[node]=1;
num[node][1]=val;
return ;
}
int t=(le+ri)>>1;
if(pos<=t) add(pos,val,le,t,node<<1);
else add(pos,val,t+1,ri,node<<1|1);
pushup(node);
}
void del(int pos,int le,int ri,int node){
if(le==ri){
num[node][1]=cnt[node]=0;
return ;
}
int t=(le+ri)>>1;
if(pos<=t) del(pos,le,t,node<<1);
else del(pos,t+1,ri,node<<1|1);
pushup(node);
}
int main(){
char str[10];
while(scanf("%d",&n)!=-1){
m=0;
for(int i=0;i<n;i++){
scanf(" %s",str);
q[i].c=str[0];
if(str[0]!='s'){
scanf("%d",&q[i].x);
a[m++]=q[i].x;
}
}
sort(a,a+m);
m=unique(a,a+m)-a;
memset(cnt,0,sizeof(cnt));
memset(num,0,sizeof(num));
int pos;
for(int i=0;i<n;i++){
if(q[i].c=='s') printf("%I64d\n",num[1][3]);
else{
pos=lower_bound(a,a+m,q[i].x)-a+1;
if(q[i].c=='a') add(pos,q[i].x,1,m,1);
else del(pos,1,m,1);
}
}
}
return 0;
}
相关文章推荐
- C#数据结构之顺序表(SeqList)实例详解
- Lua教程(七):数据结构详解
- 解析从源码分析常见的基于Array的数据结构动态扩容机制的详解
- C#数据结构之队列(Quene)实例详解
- C#数据结构揭秘一
- C#数据结构之单链表(LinkList)实例详解
- 数据结构之Treap详解
- 用C语言举例讲解数据结构中的算法复杂度结与顺序表
- C#数据结构之堆栈(Stack)实例详解
- C#数据结构之双向链表(DbLinkList)实例详解
- JavaScript数据结构和算法之图和图算法
- Java数据结构及算法实例:冒泡排序 Bubble Sort
- Java数据结构及算法实例:插入排序 Insertion Sort
- Java数据结构及算法实例:考拉兹猜想 Collatz Conjecture
- java数据结构之java实现栈
- java数据结构之实现双向链表的示例
- Java数据结构及算法实例:选择排序 Selection Sort
- Java数据结构及算法实例:朴素字符匹配 Brute Force
- Java数据结构及算法实例:汉诺塔问题 Hanoi
- Java数据结构及算法实例:快速计算二进制数中1的个数(Fast Bit Counting)