1. 程式人生 > >POJ 3067 Japan(歸併查逆序數)

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++; } }