Codeforces 1201 C.Maximum Median

題目鏈接
題意：找最大的中位數，可以操作ｋ次，每次可以給某個數＋１；
思路：從中位數到最後一次遍歷，相等就下一個，比前面的大就對前面的數（從當前到中位數）＋１，知道數組最後或者ｋ<=0；
個人思考：本人思路沒錯，但在實施的時候不知道出了什麼問題一直ＷＡ　６，有點鬱悶，最後還是沒忍住去偷看了大佬的代碼；
根據大佬的思路寫的代碼（基本就是抄的）：

#include <iostream>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <set>
#include <list>
#include <queue>
#include <stack>
#include <vector>
#include <algorithm>
#define ULL unsigned long long
using namespace std;
const int maxn=2e5+10;
long long int a[maxn];
int main()
{
    long long int i,n,m,j,k,l,p,q,s;
    scanf("%lld%lld",&n,&k);
        for(i=0; i<=n-1; i++)
        {
            scanf("%lld",&a[i]);
        }
        sort(a,a+n);
        long long int cnt=1;
        long long int ans=a[n/2];
        for(i=n/2;k>0&&i+1<n;i++)
        {
            long long int v=min(a[i+1]-ans,k/cnt);//選擇其中較小的，如果後者較小，ｋ＝０，跳出循環
            ans=ans+v;
            k=k-v*cnt;//餘額減少
            cnt++;//區間＋１
        }
        if(k>0)
        {
            ans=ans+(k/cnt);
        }
        printf("%lld\n",ans);
        return 0;
}

自己寫的代碼WA 6

#include <iostream>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <set>
#include <list>
#include <queue>
#include <stack>
#include <vector>
#include <algorithm>
#define ULL unsigned long long
using namespace std;
const int maxn=2e5+10;
long long int a[maxn];
int main()
{
    long long int i,n,m,j,k,l,p,q,s;
    while(~scanf("%lld%lld",&n,&k))
    {
        for(i=0; i<=n-1; i++)
        {
            scanf("%lld",&a[i]);
        }
        sort(a,a+n);
        while(k>0)
        {
            s=1;
            for(i=n/2; i<=n-1; i++)
            {
                //printf("1\n");
                if(a[i]==a[i+1])
                {
                    s++;
                }
                else
                {
                    p=a[i+1]-a[i];
                    break;
                }
            }
            //printf("%d\n",s);
            if(s==n-n/2)
            {
                for(p=1;p*s<=k;p++)
                {
                    //printf("yes\n");
                }
            }
            if(p*s<=k)
            {
                k=k-p*s;
                for(i=n/2; i<=n/2+s; i++)
                {
                    a[i]=a[i]+p;
                }
                //printf("%lld\n",p);
            }
            else
            {
                //printf("9\n");
                q=p-1;
                for(; q>0; q--)
                {
                    if(q*s<=k)
                    {
                        k=k-q*s;
                        for(i=n/2; i<=n/2+s; i++)
                        {
                            a[i]=a[i]+q;
                        }
                        break;
                    }
                }
                if(q<=0)
                {k=0;}
            }
        }
        printf("%lld",a[n/2]);
        return 0;
    }
}