1. 程式人生 > >線段樹、KMP、HASH模板

線段樹、KMP、HASH模板

ray down dto ret esp mar string int poi

線段樹

#include<cstdio>
using namespace std;
int n,p,a,b,m,x,y,ans;
struct node
{
    int l,r,w,f;
}tree[400001];
inline void build(int k,int ll,int rr)//建樹 
{
    tree[k].l=ll,tree[k].r=rr;
    if(tree[k].l==tree[k].r)
    {
        scanf("%d",&tree[k].w);
        return;
    }
    int m=(ll+rr)/2
; build(k*2,ll,m); build(k*2+1,m+1,rr); tree[k].w=tree[k*2].w+tree[k*2+1].w; } inline void down(int k)//標記下傳 { tree[k*2].f+=tree[k].f; tree[k*2+1].f+=tree[k].f; tree[k*2].w+=tree[k].f*(tree[k*2].r-tree[k*2].l+1); tree[k*2+1].w+=tree[k].f*(tree[k*2+1].r-tree[k*2+1].l+1); tree[k].f
=0; } inline void ask_point(int k)//單點查詢 { if(tree[k].l==tree[k].r) { ans=tree[k].w; return ; } if(tree[k].f) down(k); int m=(tree[k].l+tree[k].r)/2; if(x<=m) ask_point(k*2); else ask_point(k*2+1); } inline void change_point(int k)//單點修改 { if(tree[k].l==tree[k].r) { tree[k].w
+=y; return; } int m=(tree[k].l+tree[k].r)/2; if(x<=m) change_point(k*2); else change_point(k*2+1); tree[k].w=tree[k*2].w+tree[k*2+1].w; } inline void ask_interval(int k)//區間查詢 { if(tree[k].l>=a&&tree[k].r<=b) { ans+=tree[k].w; return; } if(tree[k].f) down(k); int m=(tree[k].l+tree[k].r)/2; if(a<=m) ask_interval(k*2); if(b>m) ask_interval(k*2+1); } inline void change_interval(int k)//區間修改 { if(tree[k].l>=a&&tree[k].r<=b) { tree[k].w+=(tree[k].r-tree[k].l+1)*y; tree[k].f+=y; return; } if(tree[k].f) down(k); int m=(tree[k].l+tree[k].r)/2; if(a<=m) change_interval(k*2); if(b>m) change_interval(k*2+1); tree[k].w=tree[k*2].w+tree[k*2+1].w; } int main() { scanf("%d",&n);//n個節點 build(1,1,n);//建樹 scanf("%d",&m);//m種操作 for(int i=1;i<=m;i++) { scanf("%d",&p); ans=0; if(p==1) { scanf("%d",&x); ask_point(x);//單點查詢,輸出第x個數 printf("%d",ans); } else if(p==2) { scanf("%d%d",&x,&y); change_point(1);//單點修改 } else if(p==3) { scanf("%d%d",&a,&b);//區間查詢 ask_interval(1); printf("%d\n",ans); } else { scanf("%d%d%d",&a,&b,&y);//區間修改 change_interval(1); } } }

KMP

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
using namespace std;
char a[1000005];
char b[10005];
int nxt[1000005];
int la,lb;
void makenext()
{
    int k=-1,j=0;
    nxt[j]=k;
    while(j<lb)
    {
        if(k==-1||b[j]==b[k])
        {
            k++,j++;
            nxt[j]=k;
        }
        else
            k=nxt[k];
    }
}
void kmp()
{
    makenext();
    int i=0,j=0;
    while(i<la)
    {
        if(j==-1||a[i]==b[j])
            j++,i++;
        else
            j=nxt[j];
        if(j==lb)
        {
            cout<<i-lb+1<<endl;
            j=0,i--;
        }
    }
}
int main()
{
    cin>>a;
    cin>>b;
    la=strlen(a);
    lb=strlen(b);
    kmp();
    for(int i=1;i<=lb;i++)
        cout<<nxt[i]<<" ";
    return 0;
}

HASH

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#define num 500003
using namespace std;
int hash[3*num];
int star[500000];
bool vis[500000];
int main()
{
    memset(hash,-128,sizeof(hash));
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        int x;
        bool mark=0;
        cin>>x;
        int y=abs(x)%num;
        while(hash[y]>=-2000000000)
        {
            if(hash[y]==x)
            {
                mark=1;
                break; 
            }
            y++;
        }
        hash[y]=x;
        if(mark==1)
            cout<<1;
        else
            cout<<0;
    }
    return 0;
}

模板來源

http://www.cnblogs.com/IUUUUUUUskyyy/

http://www.cnblogs.com/TheRoadToTheGold/

線段樹、KMP、HASH模板