BZOJ 1483 [HNOI 2009] 드 림 푸딩 (계발 식 합병)

3898 단어 데이터 구조
제목 의 중국어.
방법: 같은 색깔 의 색깔 이 바 뀌 기 때문에 합병 만 있 을 뿐 분리 가 되 지 않 습 니 다. 우 리 는 계발 식 합병 을 이용 하여 2 더 미 를 합병 하려 면 작은 폭력 더 미 를 큰 더미 에 합병 할 수 있 습 니 다. 한 푸딩 은 최대 로 logn 회 만 합병 할 수 있 습 니 다. 폭력 합병 후 크기 는 최소 2 곱 하기 때 문 입 니 다.
저 는 쓰기 가 폭력 적 입 니 다. vector 로 서로 다른 색 의 푸딩 을 저장 하고 맵 으로 어떤 색 에 대응 하 는 vector 를 저장 합 니 다.
AC 코드:
#pragma comment(linker, "/STACK:102400000,102400000")
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define ll __int64
#define ull unsigned __int64
#define eps 1e-8
#define NMAX 1000000010
#define MOD 1000003
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI acos(-1)
template
inline void scan_d(T &ret)
{
    char c;
    int flag = 0;
    ret=0;
    while(((c=getchar())'9')&&c!='-');
    if(c == '-')
    {
        flag = 1;
        c = getchar();
    }
    while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
    if(flag) ret = -ret;
}
const int maxn = 200000+10;
vectorv[maxn];
mapmp;
int bel[maxn],nct,data[maxn];

int vec(int x)
{
    if(mp[x] == 0)
    {
        v[nct].clear();
        mp[x] = nct++;
    }
    return mp[x];
}

int main()
{
#ifdef GLQ
    freopen("input.txt","r",stdin);
//    freopen("o1.txt","w",stdout);
#endif
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        mp.clear();
        nct = 1;
        int ans = 0;
        data[0] = -NMAX;
        for(int i = 1; i <= n; i++)
        {
            scanf("%d",&data[i]);
            if(data[i] != data[i-1]) ans++;
            int pos = vec(data[i]);
            v[pos].push_back(i);
            bel[i] = pos;
        }
        while(m--)
        {
            int op;
            scanf("%d",&op);
            if(op == 1)
            {
                int x,y;
                scanf("%d%d",&x,&y);
                if(x == y) continue;
                int p1 = vec(x), p2 = vec(y);
                int sz1 = v[p1].size(), sz2 = v[p2].size();
                if(sz1 < sz2)
                {
                    for(int i = 0; i < sz1; i++)
                    {
                        int tmp = v[p1][i];
                        if(tmp != 1 && bel[tmp-1] == p2) ans--;
                        if(tmp != n && bel[tmp+1] == p2) ans--;
                    }
                    for(int i = 0; i < sz1; i++)
                    {
                        int tmp = v[p1][i];
                        v[p2].push_back(tmp);
                        bel[tmp] = p2;
                    }
                    v[p1].clear();
                }
                else
                {
                    for(int i = 0; i < sz2; i++)
                    {
                        int tmp = v[p2][i];
                        if(tmp != 1 && bel[tmp-1] == p1) ans--;
                        if(tmp != n && bel[tmp+1] == p1) ans--;
                    }
                    for(int i = 0; i < sz2; i++)
                    {
                        int tmp = v[p2][i];
                        v[p1].push_back(tmp);
                        bel[tmp] = p1;
                    }
                    v[p2].clear();
                    mp[y] = p1;
                    mp[x] = 0;
                }
            }
            else printf("%d
",ans); } } return 0; }

좋은 웹페이지 즐겨찾기