Definition
An arrowhead matrix is a matrix that has all entries equal to 0, except the ones on the main diagonal, top row and leftmost column. In other words, the matrix should look like this:
* * * * * * * * 0 0 0 0 * 0 * 0 0 0 * 0 0 * 0 0 * 0 0 0 * 0 * 0 0 0 0 *
Where each * is any non-zero entry.
Task
Given a square matrix of non-negative integers, check whether it is arrowhead according to the definition above.
You may not take the size of the matrix as input, unless your language’s equivalent to an array is something like a pointer and a length (like C). It will always be at least 3 x 3.
The shortest code in bytes in each language wins.
Input and Output
You can pick among any of the following formats for receiving input:
- A matrix in the native matrix type (if your language has one)
- A 2D array1 (an array of 1D arrays, each corresponding to one row)
- A 1D array (since the matrix is always square)
- A string (you chose the spacing, but please do not abuse this in any way).
When it comes to providing output, you can either report a truthy / falsy value following the standard decision-problem definition, or choose any two distinct and consistent values.
Moreover, you can take input and give output through any standard method, in any programming language, while taking note that these loopholes are forbidden by default. If want to pick any other format or are unsure about something, please ask in the comments.
1: or your language's equivalent (list, vector, etc.)
Examples
Let's look at the following examples:
1 2 2 2 2 1 0 0 3 0 1 0 4 0 0 1
This is an arrowhead matrix (your programs should report a truthy value), because the elements on the main diagonal are 1 1 1 1, those on the top row are 1 2 2 2 and those on the leftmost column are 1 2 3 4. All other entries are 0, so this satisfies all the conditions.
3 5 6 7 1 0 8 0 0
This matrix is not arrowhead because there is a 0 on the main diagonal.
9 9 9 9 9 9 0 0 9 7 9 0 9 0 0 9
This one is not arrowhead either, because it contains a 7 in place of a 0.
More test cases
Truthy:
[[1, 1, 1], [1, 1, 0], [1, 0, 1]] [[1, 2, 3, 4], [1, 1, 0, 0], [1, 0, 1, 0], [1, 0, 0, 1]] [[1, 2, 2, 2], [2, 1, 0, 0], [3, 0, 1, 0], [4, 0, 0, 1]] [[34, 11, 35, 5], [56, 567, 0, 0], [58, 0, 679, 0], [40, 0, 0, 7]]
Falsy:
[[3, 5, 6], [7, 1, 0], [8, 0, 0]] [[9, 9, 9, 9], [9, 9, 0, 0], [9, 7, 9, 0], [9, 0, 0, 9]] [[1, 0, 3, 4], [1, 1, 0, 0], [1, 0, 1, 0], [1, 0, 0, 1]] [[1, 6, 3, 4], [13, 2, 0, 6], [29, 0, 1, 0], [2, 0, 0, 4]]
-
1\$\begingroup\$ Is it possible that the matrix can contain negative numbers \$\endgroup\$Adalynn– Adalynn2017年12月20日 19:14:26 +00:00Commented Dec 20, 2017 at 19:14
-
2\$\begingroup\$ @Zacharý No you may assume they are all non-negative. \$\endgroup\$Mr. Xcoder– Mr. Xcoder2017年12月20日 19:15:24 +00:00Commented Dec 20, 2017 at 19:15
-
\$\begingroup\$ Pedant: A two dimensional array and a matrix are not the same thing, nor is it the same as an array of arrays. Is input as a two dimensional array acceptable if your language of choice is civilised enough to support multidimensional arrays? \$\endgroup\$Ian Bush– Ian Bush2017年12月20日 20:07:27 +00:00Commented Dec 20, 2017 at 20:07
-
\$\begingroup\$ @IanBush Yes, a 2D array is totally fine. \$\endgroup\$Mr. Xcoder– Mr. Xcoder2017年12月20日 20:09:54 +00:00Commented Dec 20, 2017 at 20:09
-
10\$\begingroup\$ @Mr.Xcoder This would be a sufficiently different and interesting challenge if the arrowhead could point in any direction \$\endgroup\$dylnan– dylnan2017年12月21日 01:27:51 +00:00Commented Dec 21, 2017 at 1:27
41 Answers 41
Javascript (ES6), (削除) 48 (削除ここまで) 47 bytes
Saved 1 byte thanks to edc65
m=>m.some((r,y)=>r.some((c,x)=>(x*y&&x!=y)^!c))
Returns false for arrowhead matrices and true for non-arrowhead matrices (allowed since any two distinct values can be used to represent true and false)
Test cases:
f=m=>m.some((r,y)=>r.some((c,x)=>(x*y&&x!=y)^!c))
console.log(f([[1, 1, 1], [1, 1, 0], [1, 0, 1]]))
console.log(f([[1, 2, 3, 4], [1, 1, 0, 0], [1, 0, 1, 0], [1, 0, 0, 1]]))
console.log(f([[1, 2, 2, 2], [2, 1, 0, 0], [3, 0, 1, 0], [4, 0, 0, 1]]))
console.log(f([[34, 11, 35, 5], [56, 567, 0, 0], [58, 0, 679, 0], [40, 0, 0, 7]]))
console.log(f([[3, 5, 6], [7, 1, 0], [8, 0, 0]]))
console.log(f([[9, 9, 9, 9], [9, 9, 0, 0], [9, 7, 9, 0], [9, 0, 0, 9]]))
console.log(f([[1, 0, 3, 4], [1, 1, 0, 0], [1, 0, 1, 0], [1, 0, 0, 1]]))
console.log(f([[1, 6, 3, 4], [13, 2, 0, 6], [29, 0, 1, 0], [2, 0, 0, 4]]))
-
\$\begingroup\$ Now that's a truly clever approach! \$\endgroup\$Mr. Xcoder– Mr. Xcoder2017年12月20日 20:08:36 +00:00Commented Dec 20, 2017 at 20:08
-
1\$\begingroup\$ could this work ?
f=m=>m.some((r,y)=>r.some((c,x)=>(x*y&&x!=y)^!c))\$\endgroup\$edc65– edc652017年12月20日 22:21:36 +00:00Commented Dec 20, 2017 at 22:21 -
\$\begingroup\$ @edc65 Without the
f=of course;-)\$\endgroup\$Neil– Neil2017年12月21日 01:04:14 +00:00Commented Dec 21, 2017 at 1:04
J, (削除) 21 (削除ここまで) (削除) 20 (削除ここまで) (削除) 19 (削除ここまで) (削除) 17 (削除ここまで) 15 bytes
-4 bytes thanks to @GalenIvanov.
*-:1,1,.=&/:@}.
Takes input as a matrix (rank 2 array).
Explanation
Let the edit history be a lesson to you not to golf and write an explanation at the same time.
* -: 1, 1,. = & /: @ }. Let m be the input matrix.
= & /: @ }. Identity matrix 1 smaller than m.
}. Behead (m without its first row).
@ Composed with.
/: Grade up (get len(m) - 1 unique elements)
& Composed with.
= Self-classify (compare equality with
unique elements)
1,. Prepend a column of 1s
1, Prepend a row of 1s
* Signum (0 becomes 0, n > 0 becomes 1)
-: Does it match the generated arrowhead matrix?
Visual explanation
Note that this is done on the REPL (inputs are given starting with three spaces and output are given without any leading spaces). Because of that, I sometimes omit composition functions like @ and & since things on the REPL are evaluated right-to-left (functions are more complex).
Suppose you have the following sample matrix:
] m =. 4 4 $ 1 2 3 4 1 1 0 0 1 0 1 0 1 0 0 1
1 2 3 4
1 1 0 0
1 0 1 0
1 0 0 1
First, I'd like to explain (and give a shoutout to) @GalenIvanov's very clever way of generating the identity matrix, which is the following =&/:@}..
First, we behead the input matrix (}.).
}. m
1 1 0 0
1 0 1 0
1 0 0 1
Then we get the indices each row would be in were the rows sorted using /:-grade up.
/: }. m
2 1 0
Note that the resulting indices are unique: the list has no duplicate elements (and why would it? There's no way to place two elements in the same position in an array).
Finally, we use the niche but helpful =-self-classify. This monad compares each unique element to all of the other elements in an array. Remember how I mentioned it was important that the resulting indicies are unique? Since =-self-classify does the comparisons in the order that the unique elements appear in the list, the resulting output will be the identity matrix for a unique input (this is why =@i. is how you can make an identity matrix of a given length).
= /: }. m
1 0 0
0 1 0
0 0 1
NB. This is what is happening
(2 = 2 1 0) , (1 = 2 1 0) ,: (0 = 2 1 0)
1 0 0
0 1 0
0 0 1
Once we have the identity matrix, it's a matter of adding a row of ones and a column of ones, which is done very simply (if given an atom - i.e. a single element - the , family will repeat it to fill when it's being added):
1,. (=&/:@}. m)
1 1 0 0
1 0 1 0
1 0 0 1
1, (1,. =&/:@}. m)
1 1 1 1
1 1 0 0
1 0 1 0
1 0 0 1
Then we simply compare the generated arrowhead matrix to the signum of the input matrix.
* m
1 1 1 1
1 1 0 0
1 0 1 0
1 0 0 1
(* m) -: (1, 1,. =&/:@}. m)
1
-
2\$\begingroup\$ Isn't
*enough instead of0@<(for 17 bytes)? Try it \$\endgroup\$Galen Ivanov– Galen Ivanov2017年12月20日 19:57:44 +00:00Commented Dec 20, 2017 at 19:57 -
1\$\begingroup\$ @GalenIvanov good catch, I think so. Thanks! Time to re-edit the explanation lol. \$\endgroup\$cole– cole2017年12月20日 19:58:33 +00:00Commented Dec 20, 2017 at 19:58
-
1\$\begingroup\$ I think I found a new way to generate the identity matrix:
=&/:When I combined it with}., I got this*-:1,1,.=&/:@}.for 15 bytes Try it online! \$\endgroup\$Galen Ivanov– Galen Ivanov2017年12月20日 22:04:04 +00:00Commented Dec 20, 2017 at 22:04 -
1\$\begingroup\$ @GalenIvanov brilliant approach (both the use of
/:-grade and}.-behead), thank you again! I'll edit it in. \$\endgroup\$cole– cole2017年12月20日 22:10:05 +00:00Commented Dec 20, 2017 at 22:10 -
\$\begingroup\$ Hmm, in fact
*-:1,1,.=@}.works just fine - no need of fancy way to find the identity matrix. You can generate an identity matrix from the square matrix itself simply by=. So drop one row with}., make the identity matrix with=, add a row and a column with1and so on. \$\endgroup\$Galen Ivanov– Galen Ivanov2017年12月20日 22:17:35 +00:00Commented Dec 20, 2017 at 22:17
Wolfram Language (Mathematica), 47 bytes
Clip@#==Array[If[1<#!=#2>1,0,1]&,{1,1}Tr[1^#]]&
Explanation: Clip@# replaces all the non-zero numbers in the matrix with 1s, then we compare this to an array with dimensions {1,1}Tr[1^#] = {Length@#, Length@#} with 0 in position i,j when 1 < i != j > 1, and 1 otherwise.
(Roughly based on Uriel's answer.)
Here's another idea that's 16 bytes longer — feel free to steal it if you can golf it down:
Union@@Array[{1,#}~Tuples~2&,Length@#]==Most@Keys@ArrayRules@#&
APL (Dyalog Classic), (削除) 19 (削除ここまで) (削除) 16 (削除ここまで) (削除) 15 (削除ここまで) 13 bytes
-1 byte thanks to @ErikTheOutgolfer
(⎕IO←0)
×ばつ≡(∧=⌊)/ ̈∘⍳∘⍴
-2 bytes thanks to @ngn and @H.PWiz
How?
(2D input matrix S)
×ばつ≡Check whether S is positive only on ...(∧=⌊... the diagonals or the top row and left column ...)/ ̈∘⍳∘⍴... of S.
-
\$\begingroup\$ nice utilization of
⍳∘⍴for cartesian product. \$\endgroup\$Uriel– Uriel2017年12月20日 21:47:33 +00:00Commented Dec 20, 2017 at 21:47 -
\$\begingroup\$
×≡(=/∨1∊⊢)¨∘⍳∘⍴\$\endgroup\$Erik the Outgolfer– Erik the Outgolfer2017年12月21日 12:33:17 +00:00Commented Dec 21, 2017 at 12:33 -
1\$\begingroup\$
(=/∨1∊⊢)->(~≠⌊⌊)/\$\endgroup\$ngn– ngn2018年01月13日 12:58:12 +00:00Commented Jan 13, 2018 at 12:58 -
2\$\begingroup\$ @ngn Even better:
(∧=⌊)/, of course both require⎕IO←0\$\endgroup\$H.PWiz– H.PWiz2018年02月13日 03:23:43 +00:00Commented Feb 13, 2018 at 3:23
PowerShell, (削除) 112 (削除ここまで) 108 bytes
param($a)$o=+!(0-in$a[0]);1..($x=$a.count-1)|%{$i=$_;0..$x|%{$o*=(!($y=$a[$i][$_]),$y)[!$_-or$_-eq$i]}};!!$o
Takes input and manipulates as an array-of-arrays, since PowerShell doesn't support matrices (outside of the .NET Direct3D transform matrices support, which is something entirely different).
The whole algorithm is based on the fact that non-zero numbers are truthy and zero is falsey in PowerShell, and using multiplication to determine those truthy/falsey values.
We first take the first row, $a[0], and check whether 0 is -in that array, store that into our $output variable. If anything in that row is zero, then the $o is also zero, otherwise it's one, done by a quick cast-to-int with +.
Next we loop from 1 up to $a.count-1, setting $x along the way -- we're going to be looping through each row one at a time.
Each iteration we set helper variable $i to keep track of what row we're on, then loop from 0 to $x to iterate each element in this row. Inside the inner loop, we're again multiplying $o, this time by selecting from a tuple setup as a pseudo-ternary operator.
The tuple's conditional, !$_-or$_-eq$i, says "when we're on the 0th column, or the column matches the row (i.e., the main diagonal)" to select the second half of the tuple when truthy or the first half when falsey. The tuple is composed of !($y=$a[$i][$_]), $y. The first half sets $y for golfing the second half, but either way we're selecting the current element. The first half does Boolean negation on it, while the second half just takes the element as-is. Thus, if we're not on the 0th column nor the main diagonal, we ensure that the element is zero by taking the Boolean-not of it. Similarly, we ensure the 0th column or main diagonal is non-zero by simply taking it.
So now that we've iterated through every element in the matrix, $o is either going to be 0 if some element was incorrect, or some non-zero integer if it is an arrowhead matrix. We double-Boolean-not that to get either False or True respectively, to make our output consistent, and that's left on the pipeline where printing is implicit.
-
\$\begingroup\$
+=[int]? That is nice. \$\endgroup\$root– root2017年12月21日 20:34:33 +00:00Commented Dec 21, 2017 at 20:34 -
\$\begingroup\$ @root One of the PowerShell tips. \$\endgroup\$AdmBorkBork– AdmBorkBork2017年12月21日 21:26:33 +00:00Commented Dec 21, 2017 at 21:26
Jelly, (削除) 14 (削除ここまで) 12 bytes
ŒDμḢ;Ḣ€Ȧ>FẸ$
-2 bytes from Pietu1998
Explanation
[[9,7,1],
[7,1,0],
[7,0,1]]
Use the above matrix as an example input.
ŒDμḢ;Ḣ€Ȧ>FẸ$
ŒD Diagonals → [[9, 1, 1], [7, 0], [1], [7], [7, 0]]
μ New monadic link
Ḣ Head → [9, 1, 1]. Alters diagonals list.
;Ḣ€ Append with the head of each of the other diagonals → [9, 1, 1, 7, 1, 7, 7]
Ȧ Logical all → 1
FẸ$ Flatten what's left in diagonals then take logical any → [[0],[],[],[0]] → [0,0] → 0
> Matrix is an arrowhead iff result of Ȧ > result of Ẹ
-
\$\begingroup\$ @wizzwizz4 I'm not sure what you mean \$\endgroup\$dylnan– dylnan2017年12月22日 21:09:12 +00:00Commented Dec 22, 2017 at 21:09
-
-
\$\begingroup\$ I meant the actual visual representation of the code that you provided in your explanation. I was trying to be funny but it obviously didn't work. I'll clean up these comments. \$\endgroup\$wizzwizz4– wizzwizz42017年12月23日 16:38:49 +00:00Commented Dec 23, 2017 at 16:38
APL (Dyalog), (削除) 21 (削除ここまで) (削除) 18 (削除ここまで) 17 bytes
×ばつ≡1⍪1,(=/ ̈∘⍳1-⍨⍴)
How?
This one goes the other way -
=/ ̈∘⍳ - creates the identity matrix
1-⍨⍴ - for n - 1
1⍪1, - prepends a column and a row of 1s
≡ - compares with
×ばつ - the original matrix, after it has gone an element-wise signum-ing
MATL, 15 bytes
gtZyXy,!llY(]X=
Input is a matrix (using ; as row separator). Output is 1 for arrowhead, 0 otherwise.
Try it online! Or verify all test cases.
Explanation
g % Implicit input. Convert to logical values (nonzero becomes true and
% zero becomes false)
t % Duplicate
Zy % Size
Xy % Identity matrix of that size
, % Do twice
! % Transpose
ll % Push 1 twice
Y( % Write 1 at all entries of row 1
] % End
X= % Are the two matrices (input and constructed) equal? Implicit display
-
1\$\begingroup\$ What exactly is Indeity matrix? \$\endgroup\$Erik the Outgolfer– Erik the Outgolfer2017年12月20日 20:10:25 +00:00Commented Dec 20, 2017 at 20:10
-
13\$\begingroup\$ @EriktheOutgolfer obviously a matrix containing a deity. \$\endgroup\$cole– cole2017年12月20日 20:11:52 +00:00Commented Dec 20, 2017 at 20:11
-
5\$\begingroup\$ @cole perhaps related to a matrix over the Elysian field \$\endgroup\$jld– jld2017年12月21日 01:51:45 +00:00Commented Dec 21, 2017 at 1:51
Python 2, 75 bytes
lambda m,E=enumerate:all((x[j]>0)-(i>0<j!=i)for i,x in E(m)for j,y in E(m))
Python 2, 85 bytes
Taking the array as a 1D matrix:
def f(m):s=len(m)**.5;print all((v<1)^(0in(p>s,p%s,p//s-p%s))for p,v in enumerate(m))
R, (削除) 78 (削除ここまで) (削除) 70 (削除ここまで) (削除) 69 (削除ここまで) (削除) 68 (削除ここまで) (削除) 54 (削除ここまで) 53 bytes
function(m){d=diag(nrow(m))
d[1,]=d[,1]=1
all(d!=!m)}
Porting Luis Mendo's answer is much shorter than my former approach.
Thanks to rturnbull for pointing out a bug, and golfing down a byte!
old answer, 68 bytes:
function(m,i=which(!m,T))all(i[,1]-i[,2],i!=1,sum(m>0)==3*nrow(m)-2)
duckmayr's answer tests that all entries on the main diagonal and first row/column (m[i]) are nonzero and the rest (m[-i]) are zero, using some nice arithmetic to get the diagonal and the first row.
This answer, however, tests to make sure that (1) zero entries are not on the main diagonal or the first row/column, and (2) that there are, given an n x n matrix, 3*n-2 nonzero entries.
which returns the indices where its input is TRUE, and with the optional arr.ind=T, returns an array of indices for each array dimension, in this case, two.
Hence when any(i[,1]==i[,2]), there exists a zero on the diagonal, and when any(i==1), there exists a zero in the first row or the first column.
Finally, a little arithmetic shows that the number of nonzero entries must be 3*n-2, n from the first column, n-1 from the diagonal, and n-1 from the first row.
-
\$\begingroup\$ This doesn't seem to work for arrow matrices where the values are not 1. Did you mean
all(!m==!d)in the last line? \$\endgroup\$rturnbull– rturnbull2017年12月22日 07:17:13 +00:00Commented Dec 22, 2017 at 7:17 -
\$\begingroup\$ @rturnbull ah! Thank you. R operator syntax is so weird. I really meant
(!!m)==dbut!has lower precedence than==. I thinkd==!!mshould do the trick, though. \$\endgroup\$Giuseppe– Giuseppe2017年12月22日 11:53:20 +00:00Commented Dec 22, 2017 at 11:53 -
\$\begingroup\$ It looks like
d!=!mdoes the same, for one byte less. You can save another byte by usingpryr::fsyntax rather thanfunctiontoo. \$\endgroup\$rturnbull– rturnbull2017年12月22日 20:27:02 +00:00Commented Dec 22, 2017 at 20:27 -
\$\begingroup\$ I tried to golf it but the best i can do is still 53. \$\endgroup\$JayCe– JayCe2018年06月05日 13:39:44 +00:00Commented Jun 5, 2018 at 13:39
-
\$\begingroup\$ @JayCe nah both your answer and mine can be golfed to 52, and I'm not sure why it didn't occur to me before... I'd post yours as separate; the one-line approach is quite nice and I suspect there may be some more room for improvement in yours \$\endgroup\$Giuseppe– Giuseppe2018年06月05日 14:07:18 +00:00Commented Jun 5, 2018 at 14:07
C (gcc), (削除) 80 (削除ここまで) (削除) 75 (削除ここまで) 69 bytes
i;f(A,n)int*A;{for(i=0;i<n*n;)n*=A[i]<1^(i<n|i%n<1|i/n==i++%n);n=!n;}
Saved 5 bytes thanks to scottinet! And 6 more thanks to ceilingcat!
Reused the test code from this answer.
Linearly scans the array for any incorrect values, returning 0 for an arrowhead matrix and 1 otherwise. We check by computing the exclusive or of whether the item at a given position is zero and whether that position is on the arrow.
Encoding the information of the 2D array into one dimension leads to a fairly simple set of conditions. If we let i be our 0-based index into the n dimensional array, then i<n describes the first row. Similarly, i%n==0 describes the first column and i/n==i%n describes the diagonal.
The best trick I found for handling the return is to set the dimension to zero when encountering an error. This causes the loop to terminate immediately, then returning the logical negation of the dimension will give us one of two distinct values. scottinet found the way to make GCC return it more nicely.
-
-
-
\$\begingroup\$ @scottinet Thanks! I was having trouble figuring out which value I should set to use that trick. \$\endgroup\$FryAmTheEggman– FryAmTheEggman2017年12月21日 15:23:32 +00:00Commented Dec 21, 2017 at 15:23
-
\$\begingroup\$ Actually, I don't believe your first golf works. It passed the test cases because there was never a zero in the first position. Added a case and reverted that change. \$\endgroup\$FryAmTheEggman– FryAmTheEggman2017年12月21日 15:32:53 +00:00Commented Dec 21, 2017 at 15:32
-
\$\begingroup\$ int test0[] = {0, 1, 1, 1, 1, 0, 1, 0, 1}; printf("%d\n", f(test0, 3)); Has to return 0 not 1 (if is the 3x3 matrx 011 110 101) because a[0,0] is 0 \$\endgroup\$user58988– user589882018年01月13日 10:08:44 +00:00Commented Jan 13, 2018 at 10:08
Python 2, (削除) 92 (削除ここまで) 90 bytes
def f(m):l=len(m);return all((m[a%l][a/l]<1)^any([a<l,a%l<1,a/l==a%l])for a in range(l*l))
Credits
- Reduced from 92 bytes to 90 by Mr. Xcoder
Pip, (削除) 31 (削除ここまで) (削除) 23 (削除ここまで) 22 bytes
{0<_!=B>0MC#a==0=_MMa}
This is a function that takes a 2D nested list of numbers. Try it online!
Explanation
A whole lot of comparisons going on here. The first thing to know is that comparison operators in Pip can be chained together, like in Python: 5>4>3 is 5>4 and 4>3 (true), not (5>4)>3 (false). The second is that this doesn't apply to ==, the "exactly equals" operator. Another difference: regular comparisons have higher precedence than the mapping operators MC and MM and can be used in lambda expressions, while == has lower precedence and can't.
{ } Define a function with argument a:
0<_!=B>0MC#a Generate a matrix (as nested lists) that has 0 on the first row,
first column, and main diagonal, and 1 elsewhere (see below for
details)
0=_MMa Map the function 0=_ to the elements of the elements of a,
generating a matrix that is 0 where a is nonzero and vice versa
== Test if the two matrices are equal, returning 0 or 1 accordingly
To generate the first matrix, we use MC, "map-coords." This operator takes a number, generates a square coordinate grid of that size, and maps a function to each (x,y) coordinate pair, returning a list of lists of the results. For example, {a+b} MC 3 would give the result [[0; 1; 2]; [1; 2; 3]; [2; 3; 4]].
Here, the size of the grid is #a, the size of our original argument. The function is 0<_!=B>0, which is a shorter way of writing {0 < a != b > 0}:
{ } Function; a and b are the arguments (in our case, row and column)
0<a Return 1 (truthy) if a is greater than 0
!=b and a is not equal to b
>0 and b is greater than 0
This returns 0 for the first row/column and the main diagonal, and 1 elsewhere.
Haskell, 62 bytes
-3 bytes thanks to Mr. Xcoder. -13 bytes thanks to user28667. -5 bytes thanks to Zgarb.
z=zip[0..]
f m=and[(i==j||i*j<1)==(a>0)|(i,r)<-z m,(j,a)<-z r]
-
1\$\begingroup\$ 80 bytes.... You almost always forget about
<1and such tricks? :P \$\endgroup\$Mr. Xcoder– Mr. Xcoder2017年12月20日 20:31:58 +00:00Commented Dec 20, 2017 at 20:31 -
1\$\begingroup\$
(x==y||x==0||y==0)==(m!!y!!x/=0)should be shorter \$\endgroup\$user28667– user286672017年12月20日 21:44:36 +00:00Commented Dec 20, 2017 at 21:44 -
1
Pyth, (削除) 22 (削除ここまで) 21 bytes
This is definitely not the language for matrix manipulation.
.As.e+!MWk.Db,0k,@bkh
For each row b and its index k in the matrix (.e), grabs the first and kth entries (left side and diagonal) with ,@bkh and (+) all the other entries with .Db,0k. If k isn't 0 to correspond to the first row (Wk), then !not M all of those entries. Once all of those have been selected, make sure all of them are true. (.As) If there's a 0 where there shouldn't be, then the corresponding location will be grabbed as is and mess up the and, and if there's a nonzero where there shouldn't be, it'll be ! notted to 0, which is also false.
-1 bytes for swapping the orders around.
-
1\$\begingroup\$ Wow this solution is really nice given that Pyth is quite parallel with matrix manipulation. Probably up for another Pyth duel tomorrow :P \$\endgroup\$Mr. Xcoder– Mr. Xcoder2017年12月20日 21:24:36 +00:00Commented Dec 20, 2017 at 21:24
-
\$\begingroup\$ You might be able to shorten this using either
@VQUQor.DVQUQFor diagonals / deleting diagonals. But that would require a completely different approach. Not sure though... (BTW forgot to update link?) \$\endgroup\$Mr. Xcoder– Mr. Xcoder2017年12月20日 21:42:56 +00:00Commented Dec 20, 2017 at 21:42 -
\$\begingroup\$ @Mr.Xcoder Fixed link, I'll try to mess around with other strategies tomorrow. \$\endgroup\$Steven H.– Steven H.2017年12月20日 22:24:28 +00:00Commented Dec 20, 2017 at 22:24
-
\$\begingroup\$ I arrived at an alternative 21-byter using my
VQUQidea:>.A++hCQhQ.(VQUQsstCt. This seems highly redundant, though. You might be able to tweak it in order to save a few bytes. \$\endgroup\$Mr. Xcoder– Mr. Xcoder2017年12月21日 14:37:53 +00:00Commented Dec 21, 2017 at 14:37
APL+WIN, (削除) 36 (削除ここまで) 33 bytes
(↑⍴m)=+/(+⌿m)=×ばつn×ばつn←⍳↑⍴m←⎕
Prompts for screen input of an APL 2d matrix.
Clojure, (削除) 128 (削除ここまで) (削除) 95 (削除ここまで) (削除) 92 (削除ここまで) 85 bytes
#(every? neg?(for[R[(range(count %))]i R j R]((if((set[i(- i j)j])0)- dec)((% i)j))))
It is always exciting to see two consecutive opening brackets.
Original version:
#(and(apply =(map assoc(for[i(rest %)](subvec i 1))(range)(repeat 0)))(every? pos?(concat(map nth %(range))(% 0)(map first %))))
The first part works by associng diagonal elements of the sub-matrix to zero, and checking that all rows are equal :) I used a similar trick at Jacobian method.
Latter part concatenates the diagonal + first row and column and checks that they are positive.
Javascript (ES6), 58 bytes
My solution for Javascript:
m=>m.some((r,i)=>m[0][i]*r[0]*r[i]==0|r.filter(c=>i*c)[2])
Not as clever as Herman's answer, but I just felt like I should post it here too.
isArrowHead=m=>m.some((r,i)=>m[0][i]*r[0]*r[i]==0|r.filter(c=>i*c)[2])
console.log("-----should be false-----")
console.log(isArrowHead([[1, 1, 1], [1, 1, 0], [1, 0, 1]]))
console.log(isArrowHead([[1, 2, 3, 4], [1, 1, 0, 0], [1, 0, 1, 0], [1, 0, 0, 1]]))
console.log(isArrowHead([[1, 2, 2, 2], [2, 1, 0, 0], [3, 0, 1, 0], [4, 0, 0, 1]]))
console.log(isArrowHead([[34, 11, 35, 5], [56, 567, 0, 0], [58, 0, 679, 0], [40, 0, 0, 7]]))
console.log("-----should be true-----")
console.log(isArrowHead([[3, 5, 6], [7, 1, 0], [8, 0, 0]]))
console.log(isArrowHead([[9, 9, 9, 9], [9, 9, 0, 0], [9, 7, 9, 0], [9, 0, 0, 9]]))
console.log(isArrowHead([[1, 0, 3, 4], [1, 1, 0, 0], [1, 0, 1, 0], [1, 0, 0, 1]]))
console.log(isArrowHead([[1, 6, 3, 4], [13, 2, 0, 6], [29, 0, 1, 0], [2, 0, 0, 4]]))
console.log("-----end-----")
-
3\$\begingroup\$ Welcome to PPCG! \$\endgroup\$Steadybox– Steadybox2017年12月22日 16:20:20 +00:00Commented Dec 22, 2017 at 16:20
Python 3, (削除) 72 (削除ここまで) 71 bytes
lambda x,e=enumerate:any(0**n^(0<i!=j>0)for i,r in e(x)for j,n in e(r))
Thanks to @xnor for golfing off 1 byte!
-
\$\begingroup\$ I think
0<i!=j>0saves a byte, \$\endgroup\$xnor– xnor2017年12月23日 00:30:01 +00:00Commented Dec 23, 2017 at 0:30 -
\$\begingroup\$ @xnor Thanks! I don't think I've ever re-used a number in a comparison chain... \$\endgroup\$Dennis– Dennis2017年12月23日 00:37:12 +00:00Commented Dec 23, 2017 at 0:37
Clojure, (削除) 212 (削除ここまで) (削除) 206 (削除ここまで) 188 bytes
-6 bytes by removing some missed spaces, and shortcutting range. I might have to let this sit so I can think of a better way.
-18 bytes thanks to @NikoNyrh, and creating shortcuts for map.
(fn[m](let[r range a map z zero?](and(every? #(not(z %))(concat(m 0)(rest(a #(% 0)m))(a get m(r))))(every? z(apply concat(into(a #(take(dec %)%2)(r)(a rest m))(a take-last(r)(reverse(rest m)))))))))
Awful, just awful. I don't know why I can't wrap my head around a reasonable solution.
Takes a nested vector as input.
(defn arrowhead? [matrix]
(let [; Get the 0th cell of the 0th row, then the 1st cell of the 1st row...
main-diagonal (map get matrix (range))
; Get the 0th cell of each row
first-col (rest (map #(% 0) matrix))
arrowhead (concat (matrix 0) first-col main-diagonal)
;
right-rest (map take-last (range) (reverse (rest matrix)))
left-rest (map #(take (dec %) %2) (range) (map rest matrix))
rest-matrix (apply concat (into left-rest right-rest))]
; And check them
(and (every? pos? %) arrowhead
(every? zero? rest-matrix))))
I tried rewriting this from scratch using a different method, and it ended up longer. Instead of manually carving out the "rest" sections of the matrix, I instead decided to try generating all the coordinates in the matrix, generating the coordinates of the arrowhead, then use clojure.set/difference to get the non-arrowhead cells. Unfortunately, the call to that built-in is costly:
223 bytes
(fn[m](let[l(range(count m))g #(get-in m(reverse %))e every? a(for[y l x l][x y])k #(map % l)r(concat(k #(do[% %]))(k #(do[0%]))(k #(do[% 0])))](and(e #(zero?(g %))(clojure.set/difference(set a)(set r)))(e #(pos?(g %)))r)))
(defn arrowhead? [matrix]
(let [length-range (range (count matrix))
get-cell #(get-in matrix (reverse %))
all-coords (for [y length-range
x length-range]
[x y])
k #(map % length-range)
diag (k #(do[% %]))
top-side (k #(do [0 %]))
left-side (k #(do [% 0]))
arrowhead (concat diag top-side left-side)
; 22 bytes! Ouch
rest-cells (clojure.set/difference (set all-coords) (set arrowhead))]
(and (every? #(zero? (get-cell %)) rest-cells)
(every? #(pos? (get-cell %)) arrowhead))))
-
\$\begingroup\$ There is quite lot of room for improvement, for example
#(drop 1 %)is same asrestand#(not(zero? %))is same aspos?(as we have non-negative numbers). You might want to have a look at my 128-byte answer, which has similar approacha s this one. After implementing that I realized that it is a lot shorted to deal with index-based access in a for-loop. \$\endgroup\$NikoNyrh– NikoNyrh2017年12月22日 15:28:36 +00:00Commented Dec 22, 2017 at 15:28 -
\$\begingroup\$ @NikoNyrh Ya, I wasn't in a very good groove that day. I don't know how I forgot about
rest. I should probably just scrap this attempt and try again. \$\endgroup\$Carcigenicate– Carcigenicate2017年12月22日 15:35:04 +00:00Commented Dec 22, 2017 at 15:35
Stax, 11 bytes CP437
ä¢⌠┐xntH↔BU
Unpacked version with 13 bytes:
B|AsF:10i^\=*
Finally tied Husk and beaten by Jelly by just one byte ...
Explanation
B Push tail (all except 1st row) of the input array, then push the head (1st row)
|A All elements in the head are truthy
This will be used as an accumulator
sF For each element in the tail, execute the rest of the program
:1 All truthy indices
0i^\ Expected truthy indices (0 and the current row number)
= The truthy indices are as expected
* Perform logical "and" with the accumulator
Implicit output of the final accumulator
Husk, (削除) 12 11 (削除ここまで) 10 bytes
≡1 ́Ṫ§^*=l·L
Explanation
≡1 ́Ṫ§^*=l·L Input is a k×ばつk array A.
L The length, k.
l· The range [0,1..k-1].
́Ṫ Outer product with itself by this function:
Arguments are two numbers x and y.
= Equality of x and y
§^ to the power of
* x times y.
≡1 Does the result have the same shape and distribution of truthy values as A?
The idea is that Husk defines 0 to the power of 0 as 1, so the outer product has 1s on the first row and column.
Also, 1 to the power of any number is 1, so the outer product has 1s on the diagonal.
Other entries are 0 to the power of some positive number, which is 0.
This gives a binary arrowhead matrix, which we compare to the input with ≡.
R, (削除) 50 (削除ここまで) 49 bytes
- -1 byte thanks to pajonk
function(x,n=1:dim(x)-1)all((!n%o%n*!diag(n))-!x)
This solution is similar to the other existing 2 answers ([1], [2]) but smaller: thanks to outer and using 0-indexing, the upper row as well as the leftmost column are easily populated with 0.
-
1
05AB1E, (削除) 42 (削除ここまで) (削除) 29 (削除ここまで) (削除) 28 (削除ここまで) (削除) 27 (削除ここまで) 21 bytes
Å\IнIøн««PĀi¦€¦D€àsOQ
I've only just started with 05AB1E; matrices aren't my strong suit; nor are they 05AB1E's strong suit.. All and all not a great combination.. (削除) So can without any doubt be golfed quite a bit.. (削除ここまで) EDIT: It indeed can, see this 23 bytes answer of @Mr.Xcoder. EDIT 2: After coming back to this challenge six years later, I was indeed able to golf a few bytes without changing the general approach. ;)
Try it online or verify all test cases.
Explanation:
# (example input: [[1,2,3,4],[5,6,0,0],[7,0,8,0],[9,0,0,10]])
Å\ # Take the main diagonal of the (implicit) input-matrix
# → [1,6,8,10]
Iн # Take the first row of the input-matrix
# → [1,2,3,4]
Iøн # Take the first column of the input-matrix
# → [1,5,7,9]
«« # Merge the three lists together:
# → [1,6,8,10,1,2,3,4,1,5,7,9]
P # Take the product of all three results
# → 3628800
Āi # If this product is not 0
# (which means the diagonal, first row, and first column contained no zeros)
¦ # Use the (implicit) input-matrix, and remove its first row
# → [[5,6,0,0],[7,0,8,0],[9,0,0,10]]
€¦ # And also remove the first item of each row
# → [[6,0,0],[0,8,0],[0,0,10]]
D # Duplicate that
ۈ # Take the maximum of each
# → [6,8,10]
sO # As well as the sum of each
# → [6,8,10]
Q # And check if all maximums and sums are equal
# (this means each column and row only contains two numbers (row/column
# and diagonal); everything else is a zero)
-
1\$\begingroup\$
ćsøćsε¾èˆ)¼} ̃OН)ćs ̃ĀP‹should save you 6 bytes. Try it online! This is most certainly golfable though, but I don't have time to shorten it further now. But nice approach! \$\endgroup\$Mr. Xcoder– Mr. Xcoder2018年05月26日 11:37:07 +00:00Commented May 26, 2018 at 11:37 -
\$\begingroup\$ @Mr.Xcoder Nice approach as well! Hmm, but it's so different from my approach that I have the feeling it would be better as a separated answer.. If you want you can post it yourself, or I will add it to my answer, but also leave the 29 byte answer (or 28, since
Āinstead of0›would save a byte - was unable to find the one-char command for>0..) \$\endgroup\$Kevin Cruijssen– Kevin Cruijssen2018年05月26日 11:55:48 +00:00Commented May 26, 2018 at 11:55 -
\$\begingroup\$ I'll leave that up to you; if you don't include the 23-byter into your answer, I'll post it as a separate one when I get the chance to golf it further. I'm fine either way :) \$\endgroup\$Mr. Xcoder– Mr. Xcoder2018年05月26日 11:58:38 +00:00Commented May 26, 2018 at 11:58
-
\$\begingroup\$ @Mr.Xcoder Both are also fine by me, so you can post it yourself, since you are the one that came up with it. Let me know when you have, then I will add a link to your shorter 05AB1E answer in mine (and upvote it of course). Normally I wouldn't mind replacing my code with yours, but since the approaches are so different this time you can post it as a separated answer. :) (Have a nice weekend!) \$\endgroup\$Kevin Cruijssen– Kevin Cruijssen2018年05月26日 12:04:14 +00:00Commented May 26, 2018 at 12:04
-
\$\begingroup\$ Sure, that works. Have a nice weekend too! \$\endgroup\$Mr. Xcoder– Mr. Xcoder2018年05月26日 12:05:06 +00:00Commented May 26, 2018 at 12:05
R, (削除) 81 (削除ここまで) 79 bytes
function(x){n=nrow(x);i=c(seq(1,n^2,n+1),1:n,seq(1,n^2,n));all(x[i]>0,x[-i]<1)}
-2 bytes thanks to Mr. Xcoder
-
\$\begingroup\$ 79 bytes. \$\endgroup\$Mr. Xcoder– Mr. Xcoder2017年12月20日 19:50:43 +00:00Commented Dec 20, 2017 at 19:50
-
-
C, 117 bytes
i,j,r;f(A,n)int*A;{for(i=r=0;i<n;++i)for(j=-1;++j<n;(!i||!j||i==j)&&!A[i*n+j]&&++r)i*j&&i-j&&A[i*n+j]&&++r;return!r;}
PowerShell, 186 bytes
$a=$($args);if($a[0]-contains0){0;exit};0..($a.Length-1)|%{if($a[$_][0]-eq0-or$a[$_][$_]-eq0){0;exit};$r=$a[$_];$d=$_;if($_-ne0){1..($r.Length-1)|%{if($r[$_]-ne0-and$_-ne$d){0;exit}}}};1
-
2\$\begingroup\$ Some golfs -- use
param($a)to take input, the-containscan be swapped for an-inand all of the-eq0can be swapped for!. Finally, you can loop from1up to$a.lengthand get rid of theif($_-ne0)in the loop body. \$\endgroup\$AdmBorkBork– AdmBorkBork2017年12月20日 21:24:41 +00:00Commented Dec 20, 2017 at 21:24
Explore related questions
See similar questions with these tags.