JavaScript挑战 - Sherlock和阵列

时间:2016-04-29 11:49:20

标签: javascript unit-testing

我在采访中遇到了以下挑战,但存在一些限制。

Watson为Sherlock提供了长度为A的数组N。然后他要求他确定数组中是否存在一个元素,使其左边的元素总和等于其右边元素的总和。如果左/右没有元素,则总和被认为是零。正式地,找到i,这样, A1+A2...A(i−1)=A(i+1)+A(i+2)...AN

输入格式

第一行包含T,即测试用例的数量。对于每个测试用例,第一行包含N,即数组A中的元素数。每个测试用例的第二行包含N以空格分隔的整数,表示数组A

输出格式

对于每个测试用例,如果数组中存在元素,则打印YES,使得其左侧元素的总和等于其右侧元素的总和;否则打印NO

约束

1≤T≤10

1≤N≤10^5

1≤Ai≤2×10^4

1≤i≤N

我已经解决了它但是在某些测试用例中失败了,我想知道编码的缺陷。我花了将近4-5个小时但无法解决它。

我的解决方案是 -

function processData(input) {
    input = input.split('\n');
    var counter=0;
    var sum = function(n){
        var r=[];
        for(var k=0;k<n.length;k++){
            if(!isNaN(n[k])) {
                if(n[k] >= 1 && n[k] <= (2 * Math.pow(10,4))){
                    r.push(n[k].trim());
                }
            } 
        }
        return r.reduce(function(a, b) { return Number(a) + Number(b); }, 0);
    }
    for(var i=2;i<=input.length;i+=2){
        var ret='NO';
        if(counter<=10){
            input[i] = input[i].split(' ');
            if(input[i].length <= Math.pow(10,5) &&   input[i-1] <= input[i].length && input[i-1] >= 1){
                for(var j=0;j<input[i].length;j++){
                    if(sum(input[i].slice(0,j)) ===  sum(input[i].slice(j+1,input[i].length))){
                        ret = 'YES';
                        break;
                    }
                }
            }
        }
        counter++;
        console.log(ret);
    };

} 

process.stdin.resume();
process.stdin.setEncoding("ascii");
_input = "";
process.stdin.on("data", function (input) {
    _input += input;
});

process.stdin.on("end", function () {
   processData(_input);
});

挑战链接 - https://www.hackerrank.com/challenges/sherlock-and-array

2 个答案:

答案 0 :(得分:1)

我无法在手机上轻松编写代码,但这是我的解决方案的想法。我会在键盘上做一次正确的编辑。

让我们承认输入文件的解析是微不足道的。然后你只需编写一个函数,为单个数组返回yes或no。

步骤1:减少数组以获得其元素的总和:TotalSim

步骤2:循环数组并保持元素的部分和:LeftSum

步骤3:如果LeftSum === TotalSum - LeftSum返回是

第4步:数组结束:返回false

请注意,javascript中的整数是精确的,直到2 ^ 53 -1,这意味着您的约束保证不会发生溢出

答案 1 :(得分:0)

遍历案例,然后在每种情况下查找第一个元素是否符合 - 在元素的每一侧运行reduce并比较结果直到找到匹配或用完元素。

这应该有效:

let test = input => input
     .map(line => line
         .some((el, index) => line.slice(0, index).reduce((p, c) => p + c, 0)
           === line.slice(index - line.length + 1).reduce((p, c) => p + c, 0)))
     .map(result => result ? 'YES' : 'NO');

test(cases); // outputs array of YESs and NOs