赛码网——股神

题目

股神

题目描述

有股神吗？
有，小赛就是！

经过严密的计算，小赛买了一支股票，他知道从他买股票的那天开始，股票会有以下变化：第一天不变，以后涨一天，跌一天，涨两天，跌一天，涨三天，跌一天…依此类推。
为方便计算，假设每次涨和跌皆为1，股票初始单价也为1，请计算买股票的第n天每股股票值多少钱？

输入
输入包括多组数据；
每行输入一个n，1<=n<=10^9 。
样例输入
1
2
3
4
5

输出
请输出他每股股票多少钱，对于每组数据，输出一行。

样例输出
1
2
1
2
3

解题思路

列出前20天的股票价格，观察规律，可以发现第5天，第9天，第14天，股票价格与日期的差值分别为2， 4， 6【皆为2的倍数】，并且，第3-4天与第5天规律相同（天数和为3），第6-8天与第9天规律相同（天数和为4），第10-13天与第14天规律相同（天数和为5）。

故，只需判断某一天日期n，位于哪一个区间，即可求得股票价格，代码如下。

如有更优解，欢迎留言讨论。

代码

import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner scanner = new Scanner(System.in);
        while (scanner.hasNextInt()){
            int num = scanner.nextInt();
            int sum = 0;
            
            for (int i=2;;i++){
                if (num <= sum + i && num > sum){
                    System.out.println(num - (i-2) * 2);
                    break;
                }else{
                    sum = sum + i;
                }
            }
        }
        scanner.close();
    }
}