多次求和构造目标数组

给你一个整数数组 target 。一开始,你有一个数组 A ,它的所有元素均为 1 ,你可以执行以下操作:

    1、令 x 为你数组里所有元素的和
    2、 选择满足 0 <= i < target.size 的任意下标 i ,并让 A 数组里下标为 i 处的值为 x 。
    你可以重复该过程任意次

如果能从 A 开始构造出目标数组 target ,请你返回 True ,否则返回 False 。

  • N == target.length
  • 1 <= target.length <= 5 * 10^4
  • 1 <= target[i] <= 10^9

这里数组中元素不超int的最大值,但是他们的和可能超过了,一定要小心

解题思路是逆向法,数组中最大值一定是上一次数组的累计和,从而可以推出上一次在最大值位置的值,以此类推。

在这里我想说一下我遇到的坑,坑主要在测试用例上,代码如下:

bool isPossible(vector<int>& target) {
        int n = target.size();
        if(n==1)
        {
            if(target[0]==1) return true;
            else return false;
        }
        priority_queue<int> q;
        long long sum = 0;
        for(int i=0;i<target.size();i++) {
            q.push(target[i]);
            sum += target[i];
        }
        while(1){
            if(q.top()==1) return true;
            int x = q.top();
            q.pop();
            if(q.top()==1){
                int res = (x-1)%(sum-x);
                cout<<x<<' '<<sum<<' '<<res<<endl;
                if(res!=0) return false;
                else return true;
            }
            
            int num = (x - q.top())/(sum-x) + 1;
            cout<<x<<' '<<sum<<' '<<num<<endl;
            
            long long int y = x-num*(sum-x);//这里一定要有long long int
            sum = x-(num-1)*(sum-x);
            cout<<y<<endl;
            if(y<=0) return false;
            
            q.push(y);
        } 
    }

代码中sum一定要long long int ,另外坑在y,也一定long long int,如果测试用例是true的话,y用int是可以的,因为数组中元素的值是不会超过int 最大值,但问题出来测试用例是false的情况下,如果测试用例是[10^9]*10,即数组有10个1000000000,求出y是负的,而且int肯定溢出,所以一定要用long long int

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值