HDU 4941 Magical Forest --STL Map应用

来源：互联网发布：mac apache forbidden 编辑：程序博客网时间：2024/04/29 07:04

题意：有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]里面，因为要存下，所以行需要离散，然后排个序，以后就可以二分地找了。

代码：

#include <iostream>#include <cstdio>#include <cstring>#include <cstdlib>#include <cmath>#include <algorithm>#include <vector>#include <map>using namespace std;#define N 100007int 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

0 0