SOJ1802 Atomic Nucleus Investigation

原创于 2012-03-14 20:29:40 发布 · 561 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

SOJ

本文介绍了一种利用线段树解决集合维护问题的方法，包括插入、删除元素及求两数差最小值等操作，并提供了两种实现思路：一是直接通过线段树节点更新维护集合信息；二是借助STL中的set和multiset来高效处理集合操作。

http://www.soj.me/show_problem.php?pid=1802

题意：维护一个整数的集合，有以下操作：

1.插入一个数，若已存在则不操作。

2.删除一个数，若不存在则不操作。

3.输出当前集合中两数差的最小值。

简单线段树

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;

#define N 131072
#define M 1000000

int n1,n;

struct point
{
    int max;
    int min;
    int dis;
}p[2*N-1];

void init()
{
     int i,j;
     
     for (i=0;i<=2*N-1;i++)
     {
         p[i].max=-M;
         p[i].min=M;
         p[i].dis=M;
     }
}

int getparent(int m)
{
    return m/2;
}

int getlchild(int m)
{
    return m*2;
}

int getrchild(int m)
{
    return m*2+1;
}

void update(int m)//线段树适宜采用这种更新节点的方式
{
     int lc=getlchild(m),rc=getrchild(m);
     
     p[m].min=min(p[lc].min,p[rc].min);
     p[m].max=max(p[lc].max,p[rc].max);
     p[m].dis=min(p[lc].dis,p[rc].dis);
     if ( (p[rc].min!=M) && ((p[rc].min-p[lc].max)<p[m].dis) ) p[m].dis=p[rc].min-p[lc].max;
}

void insert(int m)
{
     int m1=m+N-1;//order
     if (p[m1].min==M)
     {
        p[m1].min=m;
        p[m1].max=m;
        
        int t=getparent(m1);
        while (t!=0)
        {
              update(t);
              t=getparent(t);
        }
     }
}

void remove(int m)
{
     int m1=m+N-1;//order
     if (p[m1].min!=M)
     {
        p[m1].min=M;
        p[m1].max=-M;
        
        int t=getparent(m1);
        while (t!=0)
        {
              update(t);
              t=getparent(t);
        }
     }
}

int main()
{
    //freopen("a.txt","r",stdin);
    //freopen("b.txt","w",stdout);
    
    int i,j,k,l,t,kk,m;
    cin>>t;
    char s[15];
    
    for (kk=1;kk<=t;kk++)
    {
        scanf("%d",&n);
        init();
        
        for (i=1;i<=n;i++)
        {
            scanf("%s",s);
            if (s[0]=='g')
            {
               scanf("%d",&m);
               insert(m);
            }
            else if (s[0]=='r')
            {
                scanf("%d",&m);
                remove(m);
            }
            else 
            {
                 if (p[1].dis==M) printf("-1\n");
                 else printf("%d\n",p[1].dis);
            }
        }
        
        if (kk<t) cout<<endl;
    }
    
    return 0;
}

//线段树更新节点的操作不宜使用下面的通过被修改的子节点去影响父节点的方式
/*void insert(int m)
{
     int m1=m+n1;//order
     if (p[m1].min==M)
     {
        p[m1].min=m;
        p[m1].max=m;
        p[m1].dis=M;
        
        int t=getparent(m1);
        while (t!=0)
        {
              if (p[m1].min<p[t].min) 
              p[t].min=p[m1].min;
              if (p[m1].max>p[t].max) 
              p[t].max=p[m1].max;
              if (p[m1].dis<p[t].dis) 
              p[t].dis=p[m1].dis;
              int lc=getlchild(t),rc=getrchild(t);
              int ss=p[rc].min,sss=p[lc].max;
              if ( (p[rc].min!=M) && (p[lc].max!=-M) && ((p[rc].min-p[lc].max)<p[t].dis) )
              p[t].dis=p[rc].min-p[lc].max;
              
              m1=t;
              t=getparent(t);
        }
     }
}

void remove(int m)
{
     int m1=m+n1;//order
     if (p[m1].min!=M)
     {
        p[m1].min=M;
        p[m1].max=-M;
        p[m1].dis=M;
        
        int t=getparent(m1);
        while (t!=0)
        {
              p[t].min=M;
              p[t].max=-M;
              p[t].dis=M;
              
              if (p[m1].min<p[t].min) 
              p[t].min=p[m1].min;
              if (p[m1].max>p[t].max) 
              p[t].max=p[m1].max;
              if (p[m1].dis<p[t].dis) 
              p[t].dis=p[m1].dis;
              int lc=getlchild(t),rc=getrchild(t);
              if ( (p[rc].min!=M) && (p[lc].max!=-M) && ((p[rc].min-p[lc].max)<p[t].dis) )
              p[t].dis=p[rc].min-p[lc].max;
              
              m1=t;
              t=getparent(t);
        }
     }
}*/

STL维护俩树，一个记录点，一个记录当前的各点距离，只需记录大小小邻的点的距离即可。

SOJ老是Restricted Function，蛋疼无比。。。。。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<vector>
#include<algorithm>
#include<string>
#include<iostream>
#include<stack>
#include<set>
#include<map>
using namespace std;

set<int> a;
multiset<int> b;

void remove(int m)
{
    multiset<int>::iterator mit;
    mit=b.find(m);
    b.erase(mit);//multiset中erase迭代器和erase值的效果是不同的，前者只删除一个，后者将重复值均删除。
}

int main()
{
    //freopen("a","r",stdin);

    int i,t,kk,m,n;
    cin>>t;
    char s[15];

    for (kk=1;kk<=t;kk++)
    {
        scanf("%d",&n);
        a.clear();
        b.clear();
        set<int>::iterator it;

        for (i=1;i<=n;i++)
        {
            scanf("%s",s);
            if (s[0]=='g')
            {
               scanf("%d",&m);

               it=a.find(m);
               if (it==a.end())
               {
                   a.insert(m);

                   if (a.size()!=1)
                   {
                        set<int>::iterator it1;
                        it1=a.find(m);
                        it1++;
                        it=a.find(m);

                        if (it==a.begin())
                        {
                            int dis=(*it1)-(*it);
                            b.insert(dis);
                        }
                        else if (it1==a.end())
                        {
                            it1--;//迭代器也可以--操作
                            it--;
                            int dis=(*it1)-(*it);
                            b.insert(dis);
                        }
                        else
                        {
                            set<int>::iterator it2;
                            it2=a.find(m);
                            it2--;
                            int dis=(*it1)-(*it);
                            b.insert(dis);
                            dis=(*it)-(*it2);
                            b.insert(dis);
                            dis=(*it1)-(*it2);
                            remove(dis);
                        }
                   }
               }
            }
            else if (s[0]=='r')
            {
                scanf("%d",&m);

                it=a.find(m);
                if (it!=a.end())
                {
                    if (a.size()!=1)
                    {
                        set<int>::iterator it1;
                        it1=a.find(m);
                        it1++;
                        it=a.find(m);

                        if (it==a.begin())
                        {
                            int dis=(*it1)-(*it);
                            remove(dis);
                        }
                        else if (it1==a.end())
                        {
                            it1--;
                            it--;
                            int dis=(*it1)-(*it);
                            remove(dis);
                        }
                        else
                        {
                            set<int>::iterator it2;
                            it2=a.find(m);
                            it2--;
                            int dis=(*it1)-(*it);
                            remove(dis);
                            dis=(*it)-(*it2);
                            remove(dis);
                            dis=(*it1)-(*it2);
                            b.insert(dis);
                        }
                    }

                    a.erase(it);
                }
            }
            else
            {
                 if (b.size()==0) printf("-1\n");
                 else printf("%d\n",*b.begin());
            }
        }

        if (kk<t) printf("\n");
    }

    return 0;
}

标签

#iterator #insert #function