码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 4941 Magical Forest --STL Map应用

时间:2014-10-28 21:21:24      阅读:231      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   io   color   os   ar   for   sp   

题意: 有n*m个格子(n,m <= 2*10^9),有k(k<=10^5)个格子中有值,现在有三种操作,第一种为交换两行,第二种为交换两列,交换时只有两行或两列都有格子有值或都没有格子有值时才能交换,第三种操作是询问现在第A行第B列值为多少.

解法:格子太大,但是k比较小,所以考虑离散一下,把行离散出来,最多10^5行,列用map存下即可。

nowR[i] = j 时表示现在的第 i 行是原来的第 j 行, nowC表示列的

R[]表示该行有没有值, CntC[]表示列

然后交换时直交换上面几个东西即可。

查询时先找到nowR[mp[A]], nowC[B],在对应原来的那一行里面找B即可,普通找不好找,索性预先把每行有的元素存起来,存到一个vector V[10^5]里面,因为要存下,所以行需要离散,然后排个序,以后就可以二分地找了。

代码:

bubuko.com,布布扣
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
#define N 100007

int nowR[N],R[N],b[N];
map<int,int> mp,CntC,nowC;
vector<pair<int,int> > V[N];
struct Point
{
    int x,y,c;
}p[N];

int bsearch(int i,int low,int high,int x)
{
    while(low <= high)
    {
        int mid = (low+high)/2;
        if(V[i][mid].first == x)
            return mid;
        else if(V[i][mid].first > x)
            high = mid-1;
        else
            low = mid+1;
    }
    return low;
}

int main()
{
    int n,m,k,i,j,T;
    int t,cs = 1;
    int op,A,B;
    scanf("%d",&t);
    while(t--)
    {
        mp.clear(); nowC.clear(); CntC.clear();
        scanf("%d%d%d",&n,&m,&k);
        for(i=1;i<=k;i++)
        {
            V[i].clear();
            scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].c);
            p[i].x++, p[i].y++;
            nowC[p[i].y] = p[i].y;
            b[i] = p[i].x;
        }
        sort(b+1,b+k+1);
        int ind = unique(b+1,b+k+1)-b-1;
        for(i=1;i<=ind;i++)
            mp[b[i]] = i;
        memset(R,0,sizeof(R));
        for(i=1;i<=k;i++)
        {
            int toR = mp[p[i].x];
            R[toR]++;
            int toC = nowC[p[i].y];
            CntC[toC]++;
            V[toR].push_back(make_pair(p[i].y,p[i].c));
        }
        for(i=0;i<=ind;i++)
        {
            nowR[i] = i;
            sort(V[i].begin(),V[i].end());
        }
        scanf("%d",&T);
        printf("Case #%d:\n",cs++);
        while(T--)
        {
            scanf("%d%d%d",&op,&A,&B);
            A++,B++;
            if(op == 1)
            {
                int toA = mp[A];
                int toB = mp[B];
                if((R[toA]&&R[toB]) || (!R[toA]&&!R[toB]))
                {
                    swap(nowR[toA],nowR[toB]);
                    swap(R[toA],R[toB]);
                }
            }
            else if(op == 2)
            {
                if((CntC[A]&&CntC[B])||(!CntC[A]&&!CntC[B]))
                {
                    swap(CntC[A],CntC[B]);
                    swap(nowC[A],nowC[B]);
                }
            }
            else
            {
                int nA = nowR[mp[A]];
                int nB = nowC[B];
                int id = bsearch(nA,0,V[nA].size()-1,nB);
                if(id < 0 || id >= V[nA].size() || V[nA][id].first != nB)
                    puts("0");
                else
                    printf("%d\n",V[nA][id].second);
            }
        }
    }
    return 0;
}
View Code

 

HDU 4941 Magical Forest --STL Map应用

标签:style   blog   http   io   color   os   ar   for   sp   

原文地址:http://www.cnblogs.com/whatbeg/p/4057894.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!