Codeforces Round 647 B题 Johnny and Grandmaster

Johnny and Grandmaster

       本题是Codeforces Round 647 B题，也是Div2的E题。当时Div2大概只有两百人过，但其实后来发现也不是很难，并没有什么特别高深的思想方法。思路来源借鉴了heyuhhh的题解，贴个链接heyuhhh的blog。
       原题就不贴了。题目的大意是给$n$个数和底数$p$，要求把这$n$个数分成两个集合$A$和$B$，设$S_A=\sum _{i=1}^{|A|}{p}^{a_i}$，$S_B=\sum _{i=1}^{|B|}{p}^{b_i}$，求$min|S_A-S_B|$。其中$1\le n,p\le 10^6$。由于最后的结果可能很大，只需返回%1000000007的结果。另外，注意是要总的差最小的情况，而非取模之后的值最小。比如说两种情况分别得到的差是2和1000000008，也应该取2的情况。
       比赛的时候满脑子都是dp，因为想到要让两个集合的差最小，自然就是要最接近所有数和的一半，是一个典型的背包问题。但这里指数$n$的值太大了，算出具体数值似乎比较棘手，顿时就没思路了。
       其实仔细用数学直觉想一想这个问题。我们知道指数是上升很快的，如果将最大的元素放在$B$里，为了差最小，自然就需要很多个小的元素放在$A$里去抵消，然而这个抵消就需要分几种情况讨论。由于排序不影响结果，我们将数组$a$排序后从大到小处理，假设首先处理$a_i$，我们将$a_i$放入了$B$，然后把接下来的元素放进$A$里去抵消，可能出现如下情况：
       第一种，放进$A$一个元素之后，$S_A<S_B$。这个时候直接继续从$A$里取元素就可以了，争取能够凑的更大。
       第二种，放进$A$一个元素之后，$S_A=S_B$。这是很舒适的情况，把大的给抵消掉了。这个时候就应该回到最开始，再把剩余最大的放入$B$，相似处理。为什么可以这么贪心处理呢？直觉上是没问题的，这里给出一个还凑合的证明。可以假设存在一个更优解，里面$a_i$是被其他一些元素抵消掉的。具体的，我们把$a$分成四部分。集合$S_1$是两种抵消$a_i$的策略都用到的元素，$S_2$是仅贪心用到的，$S_3$是仅假设的更优解用到的元素，$S_4$是两种抵消策略都没用到的元素。贪心处理会剩余$S_3,S_4$，更优解剩下$S_2,S_4$，其中$S_2$求和等于$S_3$求和，而且对$S_2$每个元素，可以对$S_3$的一个或多个元素，构成单射，使得两两的和相等(注意到这里只含有$p$的次方形式，$S_2$里面一个更大的元素一定可以替换成若干个$p$的更小次方之和)。也即，更优解剩下来的每一种划分都一定可以用$S_3,S_4$表示出来，所以贪心可以得到最优解。
       还有第三种情况，就是$S_A<S_B$。但其实这是不可能的，因为加到等于$S_B$的情况时，就已经进入情况二处理了。这也是比较容易忽略的一个地方。
       所以算法就出来了。然后有几个比较trick的地方，一个是关于幂之和$S$和这种抵消关系的表示上，直接表示和比较困难，我们可以对大元素进行拆分，即$a_i$放到集合$B$里之后，对当前遍历到的元素$a_{now}$，把$a_i$拆分成若干个$a_{now}$，然后消去一个。然后如果发现拆的个数已经大于$n$了，就不用再拆了，因为我们知道剩下所有的加在一起也打不过$a_i$了，直接continue就可以了。情况二等于的条件，就等价于拆完、抵消之后，$a_i$等价的$a_{n}ow$个数等于0，就可以进入下一次循环，放入新的$a_i$。还有在最后不可避免的要算幂，需要用一下快速幂，最后写下来代码并不复杂，代码如下：

#include <iostream>
#include <stdio.h>
#include <string>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <vector>
using namespace std;
int n,p,t;
int mod=1e9+7;
typedef long long ll;
ll quickpow(ll a,ll b){
    ll res=1;
    while(b){
        if(b&1) res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}
int main() {
    cin>>t;
    while(t--){
        cin>>n>>p;
        vector<int> a(n);
        for(int i=0;i<n;i++) cin>>a[i];
        if(p==1){
            if(n&1) cout<<1<<endl;
            else cout<<0<<endl;
            continue;
        }
        sort(a.begin(),a.end());
        vector<int> l,r;
        while(a.size()>0){
            r.push_back(a.back());
            a.pop_back();
            ll nownum=1;
            int last=r.back();
            for(int i=a.size()-1;i>=0;i--){
                l.push_back(a[i]);
                int d=last-a[i];
                last=a[i];
                a.pop_back();
                while(d&&nownum<n){
                    nownum*=p;
                    d--;
                }
                nownum--;
                if(d==0&&nownum==0) break;
            }
        }
        int res=0;
        for(auto x:l){
            res=(res+mod-quickpow(p,x))%mod;
        }
        for(auto x:r){
            res=(res+quickpow(p,x))%mod;
        }
        cout<<res<<endl;
    }
    return 0;
}