2021-05-13 树状数组（补）

242. 一个简单的整数问题

1.题意：

2.题解：

3.ac代码：

#include<bits/stdc++.h>
#include<string>
using namespace std;
typedef long long ll;
const int N=1e5+10;
ll tr[N];
int a[N];
int n,m;
int lowbit(int x){
    return x&(-x);
}
void add(int l,int x){
    for(int i=l;i<=n;i+=lowbit(i)){
        tr[i]+=x;
    }
}
ll sum(int l){
    ll res=0;
    for(int i=l;i;i-=lowbit(i)){
        res+=tr[i];
    }
    return res;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        add(i,a[i]-a[i-1]);
    }
    string s;
    int l,r,x;
    for(int i=1;i<=m;i++){
        cin>>s;
        if(s=="C"){
            cin>>l>>r>>x;
            add(l,x);add(r+1,-x);
        }else{
            cin>>x;
            cout<<sum(x)<<endl;
        }
    }
}

243. 一个简单的整数问题2

1.题意：

2.题解：

3.ac代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
ll tr1[N];
ll tr2[N];
int a[N];
int n,m;
int lowbit(int x){
    return x&(-x);
}
void add(ll tr[],int l,ll x){//!!!!!注意ll x
    for(int i=l;i<=n;i+=lowbit(i)) tr[i]+=x;
}
ll sum(ll tr[],int l){
    ll res=0;
    for(int i=l;i;i-=lowbit(i)) {
        res+=tr[i];
        
    }
    return res;
}
ll prefix_sum(int x){
    return sum(tr1,x)*(x+1)-sum(tr2,x);
}
int main(){
    string s;
    cin>>n>>m;
    int l,r,d;
    for(int i=1;i<=n;i++) {
        cin>>a[i];
        add(tr1,i,a[i]-a[i-1]);
        add(tr2,i,(ll)i*(a[i]-a[i-1]));
    }
    for(int i=1;i<=m;i++) {
        cin>>s;
        if(s=="C"){
            cin>>l>>r>>d;
            add(tr1,l,d);add(tr1,r+1,-d);
            add(tr2,l,l*d);add(tr2,r+1,-(r+1)*d);
        }else{
            cin>>l>>r;
            cout<<prefix_sum(r)-prefix_sum(l-1)<<endl;
        }
    }
}

241. 楼兰图腾

1.题意：

2.题解：

omg！！！看了几天终于看懂这个题了！！！

3.ac代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e6+10;
ll tr[N];
int a[N];
int n,m;
int Greater[N],Lower[N];
int lowbit(int x){
    return x&(-x);
}
void add(int l,int x){
    for(int i=l;i<=n;i+=lowbit(i)) tr[i]+=x;
}
ll sum(int x){
    ll res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}
int main(){
    cin>>n;
    ll res1=0,res2=0;
    for(int i=1;i<=n;i++){
         cin>>a[i];
        
    }
    for(int i=1;i<=n;i++){
         Greater[i]=sum(n)-sum(a[i]);//从左到右遍历，找到在i左边为大于a[i]的个数
         Lower[i]=sum(a[i]-1);//找到在i左边为小于a[i]的个数
         add(a[i],1);//把a[i]加到树状数组中
    }
    memset(tr,0,sizeof tr);
    //再将树状数组清空，从右边开始，计算右边大于a[i]的
    for(int i=n;i>=1;i--){
        int y=a[i];
        res1+=Greater[i]*(ll)(sum(n)-sum(y));//计算
        res2+=Lower[i]*(ll)(sum(y-1));
        add(a[i],1);
    }
    cout<<res1<<" "<<res2<<endl;
}

AcWing 244. 谜一样的牛

1.题意：

2.题解：

建议有空再做一遍。。。

确实很迷！
以下为参考

由于每头牛的高度各不相同且取遍区间[1,n][1,n]，因此，对于倒数第二头牛而言，它应该在除去上述x+1x+1的区间[1,n][1,n]中，选取An−1+1An−1+1小的数。其他的牛以此类推。

假如建立一个全部元素为1的数列，某个位置的数为1代表这个高度还不知道是哪头牛的，那么就用树状数组维护该数列的前缀和，若某个位置的前缀和等于Ai+1Ai+1，此时的下标就是要找的数。选择这个数后，将相应位置的1置0.可以二分这个位置。

3.ac代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
ll tr[N];
int a[N];
int ans[N];
int n;
int lowbit(int x){
    return x&(-x);
}
void add(int l,int x){
    for(int i=l;i<=n;i+=lowbit(i)) tr[i]+=x;
}
ll sum(int l){
    ll res=0;
    for(int i=l;i;i-=lowbit(i)){
        res+=tr[i];
    }
    return res;
}
int main(){
    cin>>n;
    for(int i=2;i<=n;i++) {
        cin>>a[i];
       
    }
    for(int i=1;i<=n;i++) tr[i]=lowbit(i);//每一个数原来都是1，就意味着前缀和是他的区间长
    for(int i=n;i;i--){//对范围是1~n的数，每次找到他第k大的数字，利用求前缀和。。找到这个数字后，其下标就是答案，
        int k=a[i]+1;
        int l=1,r=n;
        while(l<r){
            int mid=l+r>>1;
            if(sum(mid)>=k) r=mid;
            else l=mid+1;
        }
        ans[i]=r;
        add(r,-1);
    }
    for(int i=1;i<=n;i++) cout<<ans[i]<<endl;
    // cout<<endl;
}