2022.7.25刷题合集

目录

1、单源最短路

2、并查集

3、快速幂

4、快速排序

5、堆

6、线性筛质数

7、最小生成树

---

1、单源最短路

 

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N=1e4+10,M=5e5+10;
typedef pair<int,int> PII;
int n,m,s;
int h[N],e[M],ne[M],w[M],idx;
int dist[N];
void add(int a,int b,int c){
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void dijkstra(){
    memset(dist,0x3f,sizeof dist);
    dist[s]=0;
    priority_queue<PII,vector<PII>,greater<PII>> heap;
    heap.push({0,s});
    while(heap.size()){
        auto t=heap.top();
        heap.pop();
        int var=t.second;
        for(int i=h[var];i!=-1;i=ne[i]){
            int j=e[i];
            if(dist[j]>dist[var]+w[i]){
                dist[j]=dist[var]+w[i];
                heap.push({dist[j],j});
            }
        }
    }
}

int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n>>m>>s;
    memset(h,-1,sizeof h);
    while(m--){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }
    dijkstra();
    for(int i=1;i<=n;i++){
        if(dist[i]==0x3f3f3f3f) cout<<"2147483647 ";
        else cout<<dist[i]<<' ';
    }
    return 0;
}

2、并查集

 

#include<iostream>
using namespace std;
const int N=1e4+10,M=2e5+10;
int p[N];
int n,m;

int find(int x){
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++) p[i]=i;
    while(m--){
        int z,x,y;
        cin>>z>>x>>y;
        x=find(x),y=find(y);
        if(z==1){
            if(x!=y) p[x]=y;
        }else{
            if(x==y) puts("Y");
            else puts("N");
        }
    }
    return 0;
}

3、快速幂

 

#include<iostream>
using namespace std;
typedef long long LL;
LL qmi(int a,int b, int p){
    LL res=1%p;
    while(b){
        if(b&1) res=res*a%p;
        a=a*(LL)a%p;
        b>>=1;
    }
    return res;
}
int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int a,b,p;
    cin>>a>>b>>p;
    printf("%d^%d mod %d=%lld",a,b,p,qmi(a,b,p));
    return 0;
}

4、快速排序

 

#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int arr[N];
void quicksort(int arr[],int l,int r){
    if(l>=r) return;
    int x=arr[(l+r)>>1];
    int i=l-1,j=r+1;
    while(i<j){
        while(arr[++i]<x);
        while(arr[--j]>x);
        if(i<j) swap(arr[i],arr[j]);
    }
    quicksort(arr,l,j);
    quicksort(arr,j+1,r);
}
int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int n;
    cin>>n;
    for(int i=0;i<n;i++) cin>>arr[i];
    quicksort(arr,0,n-1);
    for(int i=0;i<n;i++) cout<<arr[i]<<" ";
    return 0;
}

5、堆

 

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

int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int n;
    cin>>n;
    priority_queue<int,vector<int>,greater<int>> heap;
    while(n--){
        int op;
        cin>>op;
        if(op==1){
            int x;
            cin>>x;
            heap.push(x);
        }else if(op==2){
            cout<<heap.top()<<endl;
        }else{
            heap.pop();
        }
    }
    return 0;
}

6、线性筛质数

 

#include<iostream>
using namespace std;
const int N=1e8+10;
bool isPrime[N];
int prime[N],t;
int n,q;
int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n>>q;
    isPrime[0]=isPrime[1]=true;
    for(int i=2;i<=n;i++){
        if(!isPrime[i]) prime[t++]=i;
        for(int j=0;prime[j]*i<=n;j++){
            isPrime[prime[j]*i]=true;
            if(i%prime[j]==0) break;
        }
    }
    while(q--){
        int x;
        cin>>x;
        cout<<prime[x-1]<<endl;
    }
    return 0;
}

7、最小生成树

 

#include<iostream>
#include<algorithm>
using namespace std;
const int N=5e3+10,M=2e5+10;
int p[N];
struct Edge{
    int x,y,w;
}edges[M];
bool cmp(Edge A,Edge B){
    return A.w<B.w;
}
int find(int x){
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int n,m;
    cin>>n>>m;
    for(int i=0;i<m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        edges[i]={a,b,c};
    }
    for(int i=1;i<=n;i++) p[i]=i;
    sort(edges,edges+m,cmp);
    int res=0,cnt=0;
    for(int i=0;i<m;i++){
        int x=edges[i].x,y=edges[i].y,w=edges[i].w;
        x=find(x),y=find(y);
        if(x!=y){
            p[x]=y;
            res+=w;
            cnt++;
        }
    }
    if(cnt<n-1) puts("orz");
    else cout<<res;
    return 0;
}