POJ 3067 Japan(歸併查逆序數)
阿新 • • 發佈:2018-11-11
題目大意就是有東西兩片區域,每片區域從上到下分別有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++;
}
}