Codeforces 1321B Journey Planning

题目链接

题意是给你一个数组\(b\), \(c\)是从\(1到n\),找到一个子集\([c_1, c_2,···, c_k]\)，满足\(c_i>c_{i-1}, c_i-c_{i-1} = b_{c_i}-b_{c_{i-1}}\), 使得\(b_{c_i}\)的和最大
那我们可以转换一下等式, \(c_i-b_{c_i} = c_{i-1}-b_{c_{i-1}}\), 也就是说根据\(c与b\)分组，求和最大的一组

#include<bits/stdc++.h>
using namespace std;
#define ms(x,y) memset(x, y, sizeof(x))
#define lowbit(x) ((x)&(-x))
typedef long long LL;
typedef pair<int,int> pii;

const int maxn = 6e6+7;

int buf[maxn];
LL ans[maxn];

void run_case() {
    int n; cin >> n;
    for(int i = 1; i <= n; ++i) {
        int t; cin >> t;
        ans[t - i + 200000] += t;
    }
    LL ret = 0;
    for(int i = 0; i < maxn; ++i)
        ret = max(ret, ans[i]);
    cout << ret;
}

int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    cout.flags(ios::fixed);cout.precision(2);
    //int t; cin >> t;
    //while(t--)
    run_case();
    cout.flush();
    return 0;
}