Codeforces B. Jumps (Round 99 Rated for Div.2) (思维 & 数学)

传送门

题意： 初始值为0，第 i 次操作你可 +i 或 -1。试问要得到数x至少需要多少次操作。

思路：

• 对于第 i 次操作，若选择 +i 便可增加 i 个数，若选择 -1 便相当于减少了 i+1 个数。
• 对于 i 是从1开始取值，那么能减少的数值情况就是从2开始的。
• 于是先利用前缀和存在sum数组里面，找到以一个大于或等于x的前缀和sum[k]。若该sum[k]==x+1，那么答案就是k+1；若sum[k]==x，答案就刚好是k；其他情况都可通过中间减值得到。

代码实现：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <cstring>
#include <iostream>
#include <sstream>
#include <string>
#include <list>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <functional>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int t, x, sum[N], ans;

signed main()
{
    IOS;

    for(int i = 1; i <= 2000; i ++)
        sum[i] = sum[i-1]+i;

    cin >> t;
    while(t --){
        cin >> x;
        for(int i = 1; i <= 2000; i ++){
            if(sum[i]>=x){
                ans = i;
                break;
            }
        }
        if(sum[ans]==x+1) cout << ans+1 << endl;
        else cout << ans << endl;
    }

    return 0;
}

Py代码尝试：

t = int(input())

s = []
s.append(0)
for i in range(1, 2000):
    s.append(s[i-1]+i)

ans = 0
while t:
    t -= 1
    x = int(input())
    for i in range(1, 2000):
        if s[i] >= x:
            ans = i
            break
    if s[ans] == x+1:
        print(ans+1)
    else:
        print(ans)