常用代码模板1——基础算法(来自ACWING)

快速排序

https://www.acwing.com/problem/content/787/

void quick_sort(int q[], int l, int r)
{
    if (l >= r) return;

    int i = l - 1, j = r + 1, x = q[l + r >> 1];
    while (i < j)
    {
        do i ++ ; while (q[i] < x);
        do j -- ; while (q[j] > x);
        if (i < j) swap(q[i], q[j]);
    }
    quick_sort(q, l, j), quick_sort(q, j + 1, r);
}

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<queue>
#include<stack>
#include<cstring>
using namespace std;
const int Maxn=1e6+5;
int a[Maxn];
int n;
void quick_sort(int a[],int l,int r){
    if(l>=r) return;
    int i=l-1,j=r+1,x=a[l+r>>1];
    while(i<j){
        do i++;while(a[i]<x);
        do j--;while(a[j]>x);
        if(i<j) swap(a[i],a[j]);
    }
    quick_sort(a,l,j);
    quick_sort(a,j+1,r);
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    quick_sort(a,0,n-1);
    for(int i=0;i<n;i++){
        cout<<a[i]<<" ";
    }
  return 0;
}

//  freopen("testdata.in", "r", stdin);

整数二分算法模板

https://www.acwing.com/problem/content/791/

bool check(int x) {/* ... */} // 检查x是否满足某种性质

// 区间[l, r]被划分成[l, mid]和[mid + 1, r]时使用:
int bsearch_1(int l, int r)
{
    while (l < r)
    {
        int mid = l + r >> 1;
        if (check(mid)) r = mid;    // check()判断mid是否满足性质
        else l = mid + 1;
    }
    return l;
}
// 区间[l, r]被划分成[l, mid - 1]和[mid, r]时使用:
int bsearch_2(int l, int r)
{
    while (l < r)
    {
        int mid = l + r + 1 >> 1;
        if (check(mid)) l = mid;
        else r = mid - 1;
    }
    return l;
}
#include <iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<queue>
#include<stack>
#include<cstring>
using namespace std;
int n,q;
int a[100005],b[10005];
//题意在数组中查找某元素,找不到就输出-1,找到了就输出不小于该元素的最小位置和不大于该元素的最大位置。
//不小于
//区间是mid+1,mid不用加1
int fun1(int l,int r,int k){
    while(l<r){
        int mid=l+r>>1;
        if(a[mid]<k) l=mid+1;
        else r=mid;
    }
    if (a[l]!=k) return -1;
    return l;
}
//不大于
//区间是mid-1 所以mid+1
int fun2(int l,int r,int k){
    while(l<r){
        int mid=l+r+1>>1;
        if(a[mid]>k) r=mid-1;
        else l=mid;
    }
    if (a[l]!=k) return -1;
    return l;
}
int main(){
    cin>>n>>q;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<q;i++){
        cin>>b[i];
    }

    for(int i=0;i<q;i++){
        cout<<fun1(0,n-1,b[i])<<" "<<fun2(0,n-1,b[i])<<endl;
    }
  return 0;
}

滑动窗口

常见模型:找出滑动窗口中的最大值/最小值
int hh = 0, tt = -1;
for (int i = 0; i < n; i ++ )
{
    while (hh <= tt && check_out(q[hh])) hh ++ ;  // 判断队头是否滑出窗口
    while (hh <= tt && check(q[tt], i)) tt -- ;
    q[ ++ tt] = i;
}

思路可以看这个题解:https://www.luogu.com.cn/problem/solution/P1886
例题:https://www.luogu.com.cn/problem/solution/P1886

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cmath>
using namespace std;
int n,m;
int q1[1000001],q2[1000001];
int a[1000001];
int min_deque()
{	
    int h=1,t=0;
    //h为首元素,t为队尾元素
	//t>=h时说明有元素 
    for(int i=1;i<=n;i++)
    {	
        while(h<=t&&q1[h]+m<=i){
        	h++;
		} 
        //跟随循环的次数 将头递进 
        while(h<=t&&a[i]<a[q1[t]]) {
        	t--;
		}
        //队列里有元素,并且新来的元素比队尾的小 那么就直接踢掉 
        q1[++t]=i;//入队 
        if(i>=m) printf("%d ",a[q1[h]]);
    }
    cout<<endl;
}
int max_deque()
{
    int h=1,t=0;
    for(int i=1;i<=n;i++)
    {
        while(h<=t&&q2[h]+m<=i) h++;
        while(h<=t&&a[i]>a[q2[t]]) t--;
        q2[++t]=i;
        if(i>=m) printf("%d ",a[q2[h]]);
    }
}
int main()
{
	
    cin>>n>>m;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    min_deque();
    max_deque();
    return 0;
}

posted @ 2021-03-11 19:46  一个经常掉线的人  阅读(130)  评论(0)    收藏  举报