11
\$\begingroup\$

Explanation of the challenge

Given an input, return what shapes it could be. There are only a couple options.

  1. Square
  2. Triangle

Square

A square is in the following format:

XXXX
XXXX
XXXX
XXXX

So, in this example, the input would be:

16

And the output would be:

1 or "Square"

Triangle

We are only checking for equilateral triangles:

   X
  X X
 X X X
X X X X

So, in this example, the input would be:

10

And the output would be:

2 or "Triangle"

Rules for input and output

Input must be a number.

Output can be an integer list or string list as long as you specify how your output works in your post. The output will be either a triangle, square, or both; never neither.


Test Cases

36 ->  1,2 or 3 or ["Square","Triangle"] or "Both"

15 ->  2 or "Triangle"

4  ->  1 or "Square"

Leaderboard

var QUESTION_ID=118960,OVERRIDE_USER=20260;function answersUrl(e){return"https://api.stackexchange.com/2.2/questions/118980/answers?page="+e+"&pagesize=100&order=desc&sort=creation&site=codegolf&filter="+ANSWER_FILTER}function commentUrl(e,s){return"https://api.stackexchange.com/2.2/answers/"+s.join(";")+"/comments?page="+e+"&pagesize=100&order=desc&sort=creation&site=codegolf&filter="+COMMENT_FILTER}function getAnswers(){jQuery.ajax({url:answersUrl(answer_page++),method:"get",dataType:"jsonp",crossDomain:!0,success:function(e){answers.push.apply(answers,e.items),answers_hash=[],answer_ids=[],e.items.forEach(function(e){e.comments=[];var s=+e.share_link.match(/\d+/);answer_ids.push(s),answers_hash[s]=e}),e.has_more||(more_answers=!1),comment_page=1,getComments()}})}function getComments(){jQuery.ajax({url:commentUrl(comment_page++,answer_ids),method:"get",dataType:"jsonp",crossDomain:!0,success:function(e){e.items.forEach(function(e){e.owner.user_id===OVERRIDE_USER&&answers_hash[e.post_id].comments.push(e)}),e.has_more?getComments():more_answers?getAnswers():process()}})}function getAuthorName(e){return e.owner.display_name}function process(){var e=[];answers.forEach(function(s){var r=s.body;s.comments.forEach(function(e){OVERRIDE_REG.test(e.body)&&(r="<h1>"+e.body.replace(OVERRIDE_REG,"")+"</h1>")});var a=r.match(SCORE_REG);a&&e.push({user:getAuthorName(s),size:+a[2],language:a[1],link:s.share_link})}),e.sort(function(e,s){var r=e.size,a=s.size;return r-a});var s={},r=1,a=null,n=1;e.forEach(function(e){e.size!=a&&(n=r),a=e.size,++r;var t=jQuery("#answer-template").html();t=t.replace("{{PLACE}}",n+".").replace("{{NAME}}",e.user).replace("{{LANGUAGE}}",e.language).replace("{{SIZE}}",e.size).replace("{{LINK}}",e.link),t=jQuery(t),jQuery("#answers").append(t);var o=e.language;/<a/.test(o)&&(o=jQuery(o).text()),s[o]=s[o]||{lang:e.language,user:e.user,size:e.size,link:e.link}});var t=[];for(var o in s)s.hasOwnProperty(o)&&t.push(s[o]);t.sort(function(e,s){return e.lang>s.lang?1:e.lang<s.lang?-1:0});for(var c=0;c<t.length;++c){var i=jQuery("#language-template").html(),o=t[c];i=i.replace("{{LANGUAGE}}",o.lang).replace("{{NAME}}",o.user).replace("{{SIZE}}",o.size).replace("{{LINK}}",o.link),i=jQuery(i),jQuery("#languages").append(i)}}var ANSWER_FILTER="!t)IWYnsLAZle2tQ3KqrVveCRJfxcRLe",COMMENT_FILTER="!)Q2B_A2kjfAiU78X(md6BoYk",answers=[],answers_hash,answer_ids,answer_page=1,more_answers=!0,comment_page;getAnswers();var SCORE_REG=/<h\d>\s*([^\n,]*[^\s,]),.*?(\d+)(?=[^\n\d<>]*(?:<(?:s>[^\n<>]*<\/s>|[^\n<>]+>)[^\n\d<>]*)*<\/h\d>)/,OVERRIDE_REG=/^Override\s*header:\s*/i;
body{text-align:left!important}#answer-list,#language-list{padding:10px;width:290px;float:left}table thead{font-weight:700}table td{padding:5px}
<script src="https://ajax.googleapis.com/ajax/libs/jquery/2.1.1/jquery.min.js"></script> <link rel="stylesheet" type="text/css" href="//cdn.sstatic.net/codegolf/all.css?v=83c949450c8b"> <div id="answer-list"> <h2>Leaderboard</h2> <table class="answer-list"> <thead> <tr><td></td><td>Author</td><td>Language</td><td>Size</td></tr></thead> <tbody id="answers"> </tbody> </table> </div><div id="language-list"> <h2>Winners by Language</h2> <table class="language-list"> <thead> <tr><td>Language</td><td>User</td><td>Score</td></tr></thead> <tbody id="languages"> </tbody> </table> </div><table style="display: none"> <tbody id="answer-template"> <tr><td>{{PLACE}}</td><td>{{NAME}}</td><td>{{LANGUAGE}}</td><td>{{SIZE}}</td><td><a href="{{LINK}}">Link</a></td></tr></tbody> </table> <table style="display: none"> <tbody id="language-template"> <tr><td>{{LANGUAGE}}</td><td>{{NAME}}</td><td>{{SIZE}}</td><td><a href="{{LINK}}">Link</a></td></tr></tbody> </table>

\$\endgroup\$
8
  • \$\begingroup\$ @ScottMilner In the post I specified you could use your own output method. \$\endgroup\$
    – Neil
    May 4 '17 at 21:45
  • \$\begingroup\$ The output needs to be ["Square"], as in a singleton list of the string Square? Or the string ["Square"]? Even just the string Square seems needlessly specific. \$\endgroup\$
    – xnor
    May 4 '17 at 21:45
  • 5
    \$\begingroup\$ Possibly a dupe of Polygonal Numbers!, which asks to list all k for a which a number is k-gonal. This restricts to k=3 and k=4. \$\endgroup\$
    – xnor
    May 4 '17 at 21:56
  • 1
    \$\begingroup\$ Is it acceptable to output a 1 or 0 to indicate whether it's a square, and another 1 or 0 to indicate if it's a triangle? \$\endgroup\$
    – Maria
    May 4 '17 at 22:38
  • 4
    \$\begingroup\$ Is [True, False] an OK output format (first one for triangle, second for square)? \$\endgroup\$
    – xnor
    May 4 '17 at 23:10

17 Answers 17

5
\$\begingroup\$

Japt, 18 15 11 bytes

[U8*UÄ]®¬v1

Try it online!

\$\endgroup\$
1
  • \$\begingroup\$ You can save a few bytes with shortcuts: [U8*UÄ]®¬u1 ¥0 Also, u1 ¥0 can be reduced to v1 to save a couple bytes :-) \$\endgroup\$ May 4 '17 at 23:18
4
\$\begingroup\$

Jelly, 9 7 bytes

×8‘,µÆ²

Try it online!

Outputs [1,1] for both, [0,1] for square, [1,0] for triangle, [0,0] for neither.

Explanation

×8‘,µÆ² - main link, input a
   ,    - a list of the following two:
×8‘     - a×8+1 (which is square when a is triangle)
        - (implicit) a
    µÆ² - On the previous, check if each is a square
\$\endgroup\$
4
\$\begingroup\$

JavaScript (ES7), 40 32 bytes

Returns 1 for square, 2 for triangle, 3 for both or 0 for neither.

n=>!(n**.5%1)+2*!((8*n+1)**.5%1)

Try It

Displays "Square/Triangle/Both/Neither" for clarity

f=
n=>!(n**.5%1)+2*!((8*n+1)**.5%1)
i.addEventListener("input",_=>o.innerText=["Neither","Square","Triangle","Both"][f(+i.value)])
<input id=i type=number><pre id=o>

\$\endgroup\$
2
  • 1
    \$\begingroup\$ Nice one! I think you can remove both pairs of parentheses around exponentiations (that is, n=>(n**.5%1?0:1)+((8*n+1)**.5%1?0:2). You can save another couple bytes with !(n**.5%1) \$\endgroup\$ May 4 '17 at 23:20
  • 1
    \$\begingroup\$ In that case, 2*!((8*n+1)**.5%1) might save a byte too. \$\endgroup\$
    – Neil
    May 4 '17 at 23:52
3
\$\begingroup\$

Haskell, 49 48 37 bytes

EDIT:

  • -1 byte: @xnor suggested parametrizing on k.
  • -11 bytes: And then turned it all inside out...

f takes an integer and returns a list: [3] for square, [2] for triangle, and [2,3] for both (or [] for neither.)

f n=[k|k<-[2,3],0<-scanl(-)n[1,k..n]]

Try it online!

This uses that triangular numbers are sums 1+2+3+4+...+i while square numbers are sums 1+3+5+7+...+i, thus by starting with n and using scanl(-) to subtract either the range [1,2..n] or the range [1,3..n] consecutively, each of those cases will eventually hit 0.

\$\endgroup\$
4
  • 1
    \$\begingroup\$ Nice idea with the scan. I think it would help to write the two options as [1,k..n] for k<-[2,3]. \$\endgroup\$
    – xnor
    May 4 '17 at 23:30
  • 1
    \$\begingroup\$ One weird possibility is f n=[k|k<-[2,3],0<-scanl(-)n[1,k..n]]. \$\endgroup\$
    – xnor
    May 4 '17 at 23:40
  • \$\begingroup\$ @xnor That's starting to get pretty different, sure you don't want to post it yourself? \$\endgroup\$ May 4 '17 at 23:46
  • \$\begingroup\$ @xnor Someone should do it, so I'm taking silence for a no ;) \$\endgroup\$ May 5 '17 at 12:01
2
\$\begingroup\$

PHP, 67 bytes

Returns 2 for square, 1 for triangle, 3 for both and 0 for none

<?=(($q=sqrt($i=$argn))==($q^0))*2+(2*$i==($x=sqrt(2*$i)^0)*$x+$x);

Try it online!

\$\endgroup\$
2
\$\begingroup\$

Ohm, 15 7 bytes

D8*≥«Æ²

Try it online!

EDIT: Saved 8 bytes thanks to this Jelly answer.

Outputs [true, true] if square and triangle, [true, false] for square, and [false, true] for triangle.

Explanation:

D8*≥«Æ²  Main wire, input integer a

D        Duplicate input
 8*≥     Multiply duplicate by 8, increment
    «    Pair
     Ʋ  Check if perfect square
\$\endgroup\$
2
\$\begingroup\$

TI-BASIC, 30 23 bytes

-7 for more flexible output requirements.

:Prompt X         //Get input
:{√(X),.5+√(2X+.5 //Compute the square root of X, and the inverse of the triangular number function: X(X-1)/2
:Ans=int(Ans      //Create a list of booleans for whether the values are whole or not, then print

Takes input as an iteger.

This returns a list of integers, either 0 or 1, first for being square, then for being triangular.

\$\endgroup\$
2
\$\begingroup\$

Pyth - 16 bytes (possibly 13)

+*2sI@Q2sI@h*8Q2

Prints 2 if square, 1 if triangle, 3 if both, 0 if neither

Try it

Also, 19 18 17

sI@Q2sI@h*8Q2

The first line is 1 if the number is square, 0 if not. The 2nd line is 1 if number is triangle, 0 if not.

Try it

\$\endgroup\$
1
\$\begingroup\$

JavaScript (ES7), 52 bytes

Returns 1 for square, 2 for triangle, 3 for both (and 0 for none, although irrelevant).

f=(n,t=0,s=(n**.5|0)**2==n)=>t<n?f(n-++t,t,s):2*!n|s

Demo

f=(n,t=0,s=(n**.5|0)**2==n)=>t<n?f(n-++t,t,s):2*!n|s

console.log(f(7))
console.log(f(15))
console.log(f(25))
console.log(f(36))

\$\endgroup\$
2
  • \$\begingroup\$ Scoring lower than you makes me feel I've missed something! \$\endgroup\$
    – Shaggy
    May 4 '17 at 22:55
  • \$\begingroup\$ @Shaggy I don't think so. It seems like I didn't choose the right recipe, here. Well done. :-) \$\endgroup\$
    – Arnauld
    May 4 '17 at 23:34
1
\$\begingroup\$

Python 2, 39 bytes

lambda n:[`x**.5%1==0`for x in 8*n+1,n]

Try it online!

Outputs like ["True", "False"] for triangle then square. Probably could be shorter if allowed output format is clarified.

\$\endgroup\$
3
  • \$\begingroup\$ Bools don't seem to be covered by the output methods allowed. \$\endgroup\$ May 4 '17 at 23:05
  • \$\begingroup\$ @ØrjanJohansen I think it might be covered by string list? I'll ask. \$\endgroup\$
    – xnor
    May 4 '17 at 23:09
  • \$\begingroup\$ I don't think this is one of the allowed outputs. \$\endgroup\$ May 4 '17 at 23:34
1
\$\begingroup\$

Batch, 120 bytes

@set/ar=i=j=k=0
:g
@if %j%==%1 set/ar+=1
@if %k%==%1 set/ar+=2
@set/aj+=i+=1,k=i*i
@if %j% leq %1 goto g
@echo %r%

Outputs 1 for triangular numbers, 2 for squares, 3 for both, 0 for neither.

\$\endgroup\$
1
\$\begingroup\$

Python 3, 49 42 bytes

lambda i:(i**.5%1==0)+2*((1+8*i)**.5%1==0)

Returns:

  • 1 for Triangle
  • 2 for Square
  • 3 for Both
\$\endgroup\$
2
  • \$\begingroup\$ You don't need the isdigit check, input is guaranteed to be a number. \$\endgroup\$ May 4 '17 at 22:43
  • \$\begingroup\$ Didnt notice the change in the rules. Thanks \$\endgroup\$ May 4 '17 at 22:52
1
\$\begingroup\$

C#, 61 bytes

s=n=>System.Math.Sqrt(n)%1==0;n=>(s(n)?1:0)+(s((8*n)+1)?2:0);

Could be a lot shorter if there was (or I knew) an alternative to System.Math.Sqrt.

Formatted version with test cases:

Func<int, bool> s = n => System.Math.Sqrt(n) % 1 == 0;

Func<int, int> f = n => (s(n) ? 1 : 0) + (s((8 * n) + 1) ? 2 : 0);

Console.WriteLine(f(36));
Console.WriteLine(f(15));
Console.WriteLine(f(4));
Console.WriteLine(f(2));
\$\endgroup\$
1
\$\begingroup\$

Retina, 25 bytes

*M`(^1|1\1)+$
(^1|11\1)+$

Takes input in unary, outputs t and s on separate lines, where t and s are 0 or 1 indicating whether the shape is a triangle and/or square, respectively.

Try it online! (Contains an additional \ to put both outputs on the same line to make the test cases more easily distinguishable.)

Explanation

Triangular numbers are the sums of consecutive integers starting from one. Square numbers are the ums of consecutive odd integers starting from one. Both of those are fairly easy to match with forward references.

*M`(^1|1\1)+$

This prints 0 or 1 depending on whether the given regex matches, but it doesn't actually modify the string (the * indicates a dry run). The regex either matches the initial one with ^1 or it matches exactly one more 1 than in the previous iteration with 1\1 (the \1 refers to what (^1|1\1) matched last time).

(^1|11\1)+$

This is basically the same, except that we don't need a dry run and M is implicit. To go up by odd integers instead of all integers, we simply add two 1s on each iteration with 11\1.

\$\endgroup\$
1
\$\begingroup\$

MATL, 11 10 bytes

t:UmGt:Ysm

Outputs two values 0 or 1 separated by newline. The first indicates if the input is square, and the second indicates if it is triangular.

Try it online!

Explanation

Consider input 4 as an example.

t     % Implicit input. Duplicate
      % STACK: 4, 4
:     % Range
      % STACK: 4, [1 2 3 4]
U     % Square, elementwise. This gives square numbers
      % STACK: 4, [1 4 9 16]
m     % Ismember
      % STACK: 1
Gt    % Push input again. Duplicate
      % STACK: 1, 4, 4
:     % Range
      % STACK: 1, 4, [1 2 3 4]
Ys    % Cumulative sum. This gives triangular numbers
      % STACK: 1, 4, [1 3 6 10]
m     % Ismember. Implicitly display
      % STACK 1, 0
\$\endgroup\$
0
\$\begingroup\$

Python 3, 72 67 bytes

x=input()
k=int((x*2)**.5)
print((x==int(x**.5)**2)+2*(2*x==k*k+k))

A bit longer than wanted. This is a full program that takes input from STDIN and outputs to STDOUT. STDERR should be empty as long as there isn't an I/O Exception and the input isn't invalid.

\$\endgroup\$
5
  • \$\begingroup\$ Why len(input)? \$\endgroup\$
    – Maria
    May 4 '17 at 22:24
  • \$\begingroup\$ You can change this to Python 2 and remove the len (which I think should be int). \$\endgroup\$ May 4 '17 at 22:40
  • 2
    \$\begingroup\$ @Svetlana The specs were changed after I answered. Thanks for notifying me. \$\endgroup\$
    – hyper-neutrino
    May 4 '17 at 23:38
  • 1
    \$\begingroup\$ @pycoder ^ Thanks as well \$\endgroup\$
    – hyper-neutrino
    May 4 '17 at 23:38
  • \$\begingroup\$ Why is this getting downvotes? \$\endgroup\$
    – hyper-neutrino
    May 5 '17 at 1:09
0
\$\begingroup\$

Octave, 38 bytes

@(x)~mod(x^.5,1)+2*any(x==cumsum(1:x))

Outputs 1 for square, 2 for triangular, 3 for both, 0 for neither.

Try it online!

\$\endgroup\$

Not the answer you're looking for? Browse other questions tagged or ask your own question.