Update index.ts

masx200 · masx200 · commit 1c7b0c34f3ce · 2022-09-27T20:46:48.000+08:00
diff --git a/tree-of-coprimes/index.ts b/tree-of-coprimes/index.ts
@@ -21,7 +21,7 @@ function getCoprimes(nums: number[], edges: number[][]): number[] {
             }
         }
     }
-
+    const set = new Set(nums);
     const visited = new Set<number>();
     const currents: [number, number][][] = Array(51)
         .fill(0)
@@ -34,7 +34,7 @@ function getCoprimes(nums: number[], edges: number[][]): number[] {
         const value = nums[node];
 
         let ans = [-1, -1];
-        for (const i of Array(51).keys()) {
+        for (const i of set) {
             if (currents[i].length && prime[value][i]) {
                 const get = currents[i].at(-1) as [number, number];
 

Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,7 @@ function getCoprimes(nums: number[], edges: number[][]): number[] {`
`21`	`21`	`}`
`22`	`22`	`}`
`23`	`23`	`}`
`24`		`-`
	`24`	`+ const set = new Set(nums);`
`25`	`25`	`const visited = new Set<number>();`
`26`	`26`	`const currents: [number, number][][] = Array(51)`
`27`	`27`	`.fill(0)`
`@@ -34,7 +34,7 @@ function getCoprimes(nums: number[], edges: number[][]): number[] {`
`34`	`34`	`const value = nums[node];`
`35`	`35`
`36`	`36`	`let ans = [-1, -1];`
`37`		`- for (const i of Array(51).keys()) {`
	`37`	`+ for (const i of set) {`
`38`	`38`	`if (currents[i].length && prime[value][i]) {`
`39`	`39`	`const get = currents[i].at(-1) as [number, number];`
`40`	`40`