POJ 3067 Japan(归并查逆序数)

题目大意就是有东西两片区域,每片区域从上到下分别有m和n个城市,编号为1,2,3……
有k条高速公路连接着两片区域的城市,比如输入1 2就是东边的一号城市连接着西边的2号城市。然后要我们求的是这些公路的交叉点有多少个。
一开始我看就1000个城市嘛,暴力应该ok啊,然后写了一发暴力,果断T了。然后想想1000*1000,百分百T啊。
于是开始想着用其他的方法。才发现原来这道题本质上就是求逆序对数的问题,将左边的道路按降序排列,遇到相同的数字,就按照右边的数字降序排列(必须降序,不然会多一个逆序对),然后求一下右边的逆序对数就是交叉点的数量了。
因为之前刚用归并写了一发查逆序数,所以这道题也用归并来查。写完一共有三个错误……
1、 不能用cin和cout,会T的
2、 没有注意到题目要求的输出格式,直接输出了,WA
3、 这个卡了我好一会儿,就是相同的数字不能算逆序数,我不小心算了。就WA了。
其次还要注意,我写完看了一下别人的discuss发现有很多人错在了sum上面,因为sum需要longlong才能存下,我一开始就下意识的设成了longlong,所以没卡在这上面。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=2000000;
vector<pair<int,int > >vis;
int c[maxn]={0};
int tmp[maxn]={0};
int m,n,k;
int t;
long long sum=0;
void mergy(int left,int mid,int right)
{
    int i=left;
    int j=mid+1;
    int kkk=0;
    while((i<=mid)&&(j<=right))
    {
        if(c[i]<=c[j])
        {
            tmp[kkk++]=c[i++];
        }
        else
        {
            tmp[kkk++]=c[j++];
            sum+=(mid-i+1);
        }
    }
    while(i<=mid)
    {
        tmp[kkk++]=c[i++];
    }
    while(j<=right)
    {
        tmp[kkk++]=c[j++];
    }
    for(int o=0;o<kkk;o++)
    {
        c[o+left]=tmp[o];
    }
}
void kk(int begin,int end)
{
    int bit=(begin+end)/2;
    if(begin==end)return ;
    kk(begin,bit);
    kk(bit+1,end);
    mergy(begin,bit,end);
}
bool cmp(pair<int ,int > a,pair<int ,int > b)
{
    if(a.first==b.first)return a.second<b.second;
    else return a.first<b.first;
}
int main()
{
    cin>>t;
    int ij=1;
    while(t--)
    {
        vis.clear();
        sum=0;
        memset(tmp,0,sizeof tmp);
        memset(c,0,sizeof c);
        scanf("%d%d%d",&m,&n,&k);
        for(int i=0;i<k;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            vis.push_back(make_pair(a,b));
        }
        sort(vis.begin(),vis.end(),cmp);
        for(int i=0;i<vis.size();i++)
        {
            c[i]=vis[i].second;
        }
        kk(0,k-1);
        printf("Test case %d: %lld\n",ij,sum);
        ij++;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_41564817/article/details/82219568