标签：Johnny int res ll Grandmaster ans sum mod

题意：

给定 \(n\) 个数 \(P={p^{k_i}}\)，将这 \(n\) 个数划分到两个集合中，使得这两个集合各自和之间的差值最小，求最小的差值对 \(1e9+7\) 取模。
\(1≤n,p≤10^6,0≤k_i≤10^6)\)

分析：

\(p\) 进制，数学归纳法。
首先，确定最优的策略：
  将 \(k_i\) 从小到大进行排序，从大的开始取。求前缀和（在和大的集合中为正，另一个为负），当前缀和正时，下一个数取负，当前缀和为负时，下一个数取正。这样可以保证最终的差值最小。
证明：
（以下从大到小）
  假设有 \(n\) 个数的时候这种策略是正确的。当有 \(n+1\) 个数的时候，使用这种策略给 \(n+1\) 个数指定正负，得出前 \(n\)个数的和为 \(sum_n\)。记第 \(n+1\) 个数为 \(a_{n+1}\)，如果\(sum_n=0\)，则按照这种策略求得和为 \(sum_{n+1}=a_{n+1}\)，如果 \(sum_n>0\)，则按照这种策略求得和为 \(sum_{n+1}=sum_n−a_n\)。
​  假设存在一种最优策略求得和为 \(best_{n+1}<sum_{n+1}\)，记这种更优策略求得前 \(n\) 个数的和为 \(best_n\)(如果 \(best_n<0\),那么可以把最优策略规定的正负性取反得到等价策略，使得\(best_n≥0\))，根据数学归纳法的假设可得：\(sum_n≤best_n\)，即应对前 \(n\) 个数的时候贪心策略与"最优策略"差不多。所以最优策略在面对第 \(n+1\) 个数的时候会做和贪心策略等价的选择，说明最优策略不会得到优于贪心策略的答案，即最优策略就是贪心策略。

实现：

按照策略，从大到小处理。
代码中 \(while\) 中的 \(ans\) 是当前数 \(p^{k_{cnt}}\) 的倍数。如果直接求数，肯定会爆 \(long long\) ，因此要借助取模。但每次都取模会影响正负的判断，因此只有当大于 \(mod\) 的时候，才取模。又因为，如果数 \(p^{k_{cnt}}\) 的倍数都大于 \(mod\) 。那么，后面所有的数都可以直接取负（因为最多 \(10^6\) 个数），不用再判断了。

代码：

#include <bits/stdc++.h>
using namespace std;
const int mod=1000000007;
typedef long long ll;
const int N=1e6+6;
int k[N];
ll power(ll a,int b)
{
    ll res=1;
    while(b)
    {
        if(b&1) res=res*a%mod;
        b>>=1;
        a=a*a%mod;
    }
    return res%mod;
}
bool check(ll a,int b)
{
    ll res=1;
    while(b)
    {
        if(b&1) res=res*a;
        if(res>mod) return 1;
        a=a*a;
        if(a>mod) return 1;
        b>>=1;
    }
    return 0;
}
int main()
{
    int t,n,p;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&p);
        for(int i=1;i<=n;i++)
            scanf("%d",&k[i]);
        sort(k+1,k+1+n);
        int cnt=n-1,f=0,lp=k[n];
        ll ans=1;//表示倍数
        while(cnt)
        {
            int t=k[cnt];
            cnt--;
            if(!f)//没有取模过
            {
                if(check(1LL*p,lp-t)&&ans!=0||ans*power(1LL*p,lp-t)>mod)//当前数的倍数>mod
                    f=1;
                ans=ans*power(1LL*p,lp-t)%mod;
                if(ans>0) ans-=1;
                else ans=1;
            }
            else
            {
                ans=ans*power(1LL*p,lp-t)%mod;
                ans=(ans-1+mod)%mod;
            }
            lp=t;
        }
        ans=ans*power(1LL*p,lp)%mod;
        printf("%lld\n",(ans+mod)%mod);
    }
    return 0;
}

参考博客

标签：Johnny,int,res,ll,Grandmaster,ans,sum,mod
来源： https://www.cnblogs.com/1024-xzx/p/13055557.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。