Fork me on GitHub

【UR #4】元旦激光炮

uoj 传送门
第一次做交互题,大概明白了是啥意思,就是写一个函数,帮助实现一个操作。
题目中有要求,调用题目中给出的可以用的函数次数越少越好,这就是优化所在。
这道题就是在三个升序数组中找所有数的第K大,我们要尽量少用题目中已给出的函数。
我们每次现在三个数组中找出第k/3大,k-=k/3,那把找出的数的所在数组的前面的数删掉,这样下去,直到k为0就好啦。这样调用的次数肯定很少,就可以拿到100分啦。

#include "kth.h"
#include<iostream>
int minn(int x,int y){return x<y?x:y;}; 
int query_kth(int n_a, int n_b, int n_c, int k)
{
    int pa=0,pb=0,pc=0,tmin;
    while(k)
    {
        int l=(k-2)/3;
        int t1=get_a(pa+l);
        int t2=get_b(pb+l);
        int t3=get_c(pc+l);
        tmin=minn(minn(t1,t2),t3);
        l++;
        if(tmin==t1) pa+=l;
        else if(tmin==t2) pb+=l;
        else if(tmin==t3) pc+=l;
        k-=l;
    }
    return tmin;
}
posted @ 2017-09-24 17:48  primes  阅读(171)  评论(0编辑  收藏  举报