ACWing例题

数组中第k小的数:
https://www.acwing.com/problem/content/submission/code_detail/9193612/
主要思路:
我们采用类似快排的算法,每次给函数指定一个起点l和终点r,我们选取一个哨兵(a[l]),然后进行快排,将它放在合适的位置上,使得它左边的数都比他小,右边的数都比他大。 然后我们判断k与a[l]的位置关系,选择递归处理左半部分/右半部分即可。

每次将数组分半,因此复杂度=(n/2+n/4+n/8…)=2n

#include<iostream>
using namespace std;
#define N 100010
int a[N];
int qsort(int l,int r,int k){
    if(l>=r) return a[l];
    
    int i=l-1,j=r+1,x=a[l];
    while(i<j){
       while(a[++i]<x);
       while(a[--j]>x);
       if(i<j) swap(a[i],a[j]);
    }
    
    if(k<=j) return qsort(l,j,k);
    else return qsort(j+1,r,k);
}

int main(){
    int n,k;
    cin>>n>>k;

    for(int i=0;i<n;i++)
    scanf("%d",&a[i]);
    
    cout<<qsort(0,n-1,--k);
}

---

逆序对的数量:
https://www.acwing.com/problem/content/790/

设msort(l,r)的返回值为区间 [l,r]的逆序对的个数
则msort(l,r) =(msort(l,mid)+msort(mid+1,r)+[l,mid]和[mid+1,r]两个区间的逆序对的个数)
现在来考虑如何统计两个区间之间的逆序对的个数
首先左区间内的每个数都在右区间内每个数的左边(这可不是废话)
也就是说,对于右区间的一个数b,我们需要找到在左区间中找到比b大的数的个数即可。

采用双指针算法,两个指针i,j,假设j已经放置好了,i为比j大的第一个数的位置,则ans+=mid-i+1,枚举每一个j,然后循环的加ans就可以了!

#include<iostream>
#define N 100010
#define ll long long
using  namespace std;
int a[N],temp[N];
int n;
ll msort(int l,int r){
    if(l>=r) return 0;
    
    int mid=l+r >>1;
    ll res=0;
    res+=msort(l,mid)+msort(mid+1,r);
    
    int i=l,j=mid+1,k=l;
    while(i<=mid&&j<=r){
        if(a[i]<=a[j]){
            temp[k++]=a[i++];
        }
        else{
            // 此时i停留的数一定大于j
            res+=(mid-i)+1;
            temp[k++]=a[j++];
        }
    }
    
    while(i<=mid) temp[k++]=a[i++];
    while(j<=r) temp[k++]=a[j++];
    
    for(i=l;i<=r;i++)
    a[i]=temp[i];
    
    return res;
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    cout<<msort(0,n-1);
    
}

---

数的三次方根:
https://www.acwing.com/problem/content/792/

设置一个EPS,当l和r之间的差距大于EPS时就不断运行即可

#include<iostream>

using namespace std;
#define EPS 1e-8

int main(){
    
    double l=-10000,r=10000;
    double n;
    cin>>n;
    
    while(r-l>=EPS){
        double mid=(l+r)/2;
        if(mid*mid*mid>n){
            r=mid;
        }else{
            l=mid;
        }
    }
    
    printf("%.6lf",l);
        
}

---

前缀和:
https://www.acwing.com/problem/content/797/
sum(l,r)= pre[r]-pre[l-1]

#include<iostream>
#include<cstring>
using namespace std;

#define N 100010

int a[N],pre[N];

int main(){
    memset(pre,0,sizeof(pre));
    int l,r,n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);

    for(int i=1;i<=n;i++)
    pre[i]=pre[i-1]+a[i]; //统计出前缀和
    
    while(m--){
        scanf("%d %d",&l,&r);
		printf("%d\n",pre[r]-pre[l-1]);
    }
    
}

---

二维前缀和:
https://www.acwing.com/problem/content/798/
sum((x1,y1),(x2,y2))=pre[x2][y2]-pre[x1-1][y2]-pre[x2][y1-1]+pre[x1-1][y1-1]

#include<iostream>
using namespace std;
#define N 1010
int a[N][N],pre[N][N];

int main(){
    int n,m,q,x1,y1,x2,y2;
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    scanf("%d",&a[i][j]);
    
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    pre[i][j]=a[i][j]+pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1];
    
    while(q--){
        cin>>x1>>y1>>x2>>y2;
        
        cout<<pre[x2][y2]-pre[x1-1][y2]-pre[x2][y1-1]+pre[x1-1][y1-1]<<endl;
    }
    
}

---

一维差分数组:
https://www.acwing.com/problem/content/799/

#include<iostream>
using namespace std;

#define N 100010

int n,m;
int sub[N];
void add(int l,int r,int k){
    sub[l]+=k;
    sub[r+1]-=k;
} //为l-r之间的所有数字加上K

int main(){
 
cin>>n>>m;
for(int i=1;i<=n;i++)
    {
        int t;
    scanf("%d",&t);
    add(i,i,t);
    }

while(m--){
    int l,r,k;
    cin>>l>>r>>k;
    add(l,r,k);
}

for(int i=1;i<=n;i++){
    sub[i]+=sub[i-1];
    printf("%d ",sub[i]);
    }
    
}

---

二维差分数组:
https://www.acwing.com/problem/content/800/

#include<iostream>
#define N 1010
int b[N][N];
using namespace std;

int n,m,q;

void add(int x1,int y1,int x2,int y2,int v){
    b[x1][y1]+=v;
    b[x1][y2+1]-=v;
    b[x2+1][y1]-=v;
    b[x2+1][y2+1]+=v;
}

int main(){
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        int t;
        scanf("%d",&t);
        add(i,j,i,j,t);
    }
    
    while(q--){
        int a,b,c,d,v;
        cin>>a>>b>>c>>d>>v;
        add(a,b,c,d,v);
    }
    
    for(int i=1;i<=n;i++)
    {
    for(int j=1;j<=m;j++){
         b[i][j]+=b[i-1][j]+b[i][j-1]-b[i-1][j-1];
        printf("%d ",b[i][j]);
    }
    printf("\n");
    }
}