題目描述
http://cxsjsxmooc.openjudge.cn/2018t1springall/066/
題目思路
- 這道題看着簡單,但是有很多坑
- 要考慮實力差相等的時候取id比較小的對手
- 要考慮時間複雜度,下面我的第一種解法就TLE了
解法一
二維數組存id和實力值,基本上是O(n*n)的時間複雜度,但是測試用例中的100000組輸入,代碼超時
#include <iostream>
#include<math.h>
#include <fstream>
using namespace std;
#define N 100001
int a[N][2],fight[N];
int main() {
freopen("in.txt","r",stdin);
int n;
cin>>n;
a[0][0]=1,a[0][1]=1000000000;
for(int i=1;i<=n;i++){
cin>>a[i][0]>>a[i][1];
}
for(int i=1;i<=n;i++){
int min=a[0][1];
for(int j=i-1;j>=0;j--){
if(abs(a[i][1]-a[j][1])<=min){//不能忽略實力值相等
if(abs(a[i][1]-a[j][1])==min){
if(a[j][0]<fight[i]){//實力差相等,取id小的
fight[i]=a[j][0];
}
}else{
min=abs(a[i][1]-a[j][1]);
fight[i]=a[j][0];
}
}
}
}
for(int i=1;i<=n;i++){
cout<<a[i][0]<<" "<<fight[i]<<endl;
}
fclose(stdin);
return 0;
}
解法二
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<vector>
using namespace std;
map<int,int>a; //實力值,ID
map<int,int>::iterator it;
int n,x,pw;
int main()
{
freopen("in.txt","r",stdin);
a[1000000000]=1;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&x,&pw);
it=a.lower_bound(pw);
if(it==a.end()) it--;
int idx=it->second;
int pw1=abs(it->first-pw);
if(it!=a.begin())
{
it--;
if(abs(it->first-pw)<pw1||(pw1==abs(it->first-pw)&&it->second<idx))
idx=it->second;
}
printf("%d %d\n",x,idx);
it=a.find(pw);
if(it==a.end()||it->second>x)
a[pw]=x;
}
fclose(stdin);
return 0;
}