import random #--------粒 子 群 最 適 化------------------------------ #評価関数: def criterion(x, y): z=-abs(k-(x*(x+1))/2) return z #粒子の位置の更新を行う関数 def update_position(x, y, vx, vy): new_x = x + vx new_y = y + vy return new_x, new_y #粒子の速度の更新を行う関数 def update_velocity(x, y, vx, vy, p, g, w=0.5, ro_max=1): #パラメーターroはランダムに与える ro1 = random.uniform(0, ro_max) ro2 = random.uniform(0, ro_max) #粒子速度の更新を行う new_vx = w * vx + ro1 * (p[0] - x) + ro2 * (g[0] - x) new_vy = w * vy + ro1 * (p[1] - y) + ro2 * (g[1] - y) return new_vx, new_vy def main(): N = 600 #粒子の数 x_min, x_max = 0, k y_min, y_max = 0, 0 #粒子位置, 速度, パーソナルベスト, グローバルベストの初期化を行う ps = [[random.uniform(x_min, x_max), random.uniform(y_min, y_max)] for i in range(N)] vs = [[0.0, 0.0] for i in range(N)] personal_best_positions = list(ps) personal_best_scores = [criterion(p[0], p[1]) for p in ps] best_particle = personal_best_scores.index(max(personal_best_scores)) global_best_position = personal_best_positions[best_particle] T = 300 #世代数(ループの回数) for t in range(T): for n in range(N): x, y = ps[n][0], ps[n][1] vx, vy = vs[n][0], vs[n][1] p = personal_best_positions[n] #粒子の位置の更新を行う new_x, new_y = update_position(x, y, vx, vy) ps[n] = [new_x, new_y] #粒子の速度の更新を行う new_vx, new_vy = update_velocity(new_x, new_y, vx, vy, p, global_best_position) vs[n] = [new_vx, new_vy] #評価値を求め, パーソナルベストの更新を行う score = criterion(new_x, new_y) if score > personal_best_scores[n]: personal_best_scores[n] = score personal_best_positions[n] = [new_x, new_y] #グローバルベストの更新を行う best_particle = personal_best_scores.index(max(personal_best_scores)) global_best_position = personal_best_positions[best_particle] #最適解 # print(max(personal_best_scores), global_best_position) return max(personal_best_scores), global_best_position[0], global_best_position[1] #-------------------------------------------------------------- k=int(input()) best=100000000000000 xx=0 yy=0 for i in range(3): kouho_best, kouho_x, kouho_y=main() if best>kouho_best: best=kouho_best xx=kouho_x yy=kouho_y #print(best) if best>-0.000000000001: print("YES") print(int(xx)) else: print("NO")