Birthday Gift
题目链接:https://ac.nowcoder.com/acm/contest/5795/B
题意:给定两个数组
例如
a=[1 3 5 4 2]
b=[2 4 3 5 3]
从 A,B中选出两个索引
如选x,y(1<=x,y<=5(五个数))
那么他的得分就是min(a[x]+b[y],b[y]+a[x])
我们的目的就是选出两个索引是的其得分最大
观察他的数据范围是2e5
因此我们要考虑时间复杂度O(N2)以下
思路:
我最先想的是二分答案,但是二分答案检查答案的过程和找出最大值的暴力方法是一样的,都是O(N2),那么多加一个二分,复杂度反而加了log(N)
我自己思考了一下,没有想到优化的算法,突然灵光一闪
我们可以优化数据范围
在求质数的时候我们就学过一个办法,可以做到接近O(N)删除
那么在深度学习上有一个算法叫NMS
我们可以参考上面的方法
首先建立一个结构体
typedef struct p{ ll x,y; }t; t m[200005],n[200005];
输入数据后对结构体排序
typedef struct p{ ll x,y; }t; t m[200005],n[200005];
再通过我们筛选的方法
for(ll i=a;i>=1;i--) { ll flag=0; if(m[i].y==-1) continue; for(ll j=i-1;j>=1;j--) { if(m[j].y==-1) continue; if(m[i].y>=m[j].y) { if(flag) m[j].y=-1; else flag=1; } } }
主要思想就是:
假如一个索引,他比另一个数,上下都大,那么另一个数就没有存在的意义了,但是注意并不是所有的都比他小的数都删去,第二小以后的才删去(为啥?自己思考一下)
完整代码
#include <bits/stdc++.h> using namespace std; typedef long long ll; typedef struct p{ ll x,y; }t; t m[200005],n[200005]; bool cmp(p a,p b) { return a.x<b.x; } int main() { ll a,b,c,d,e,f,g; cin>>a; for(ll i=1;i<=a;i++) cin>>m[i].x; for(ll i=1;i<=a;i++) cin>>m[i].y; sort(m+1,m+1+a,cmp); for(ll i=a;i>=1;i--) { ll flag=0; if(m[i].y==-1) continue; for(ll j=i-1;j>=1;j--) { if(m[j].y==-1) continue; if(m[i].y>=m[j].y) { if(flag) m[j].y=-1; else flag=1; } } } ll k=0; for(ll i=1;i<=a;i++) { if(m[i].y!=-1) { k++; n[k].x=m[i].x; n[k].y=m[i].y; } } ll ans=-99999999; for(ll i=1;i<=k;i++) { for(ll j=i+1;j<=k;j++) { ans=max(ans,min(n[i].x+n[j].y,n[j].x+n[i].y)); } } cout<<ans; return 0; }