CodeForces1320A. Journey Planning

CodeForces1320A

题目描述

# A. Journey Planning

难度 ：1400

标签 ：data structres、DP、greedy、math、sortings

思路

将题目中所给条件$c_{i+1}-c_i=b_{c_{i+1}}-b_{c_i}$转化为$b_{c_i}-c_i=b_{c_{i+1}}-c_{i+1}$
令$s_i=b_{c_i}-c_i$
即将问题转换为求$s_i$相同的位置上的$b_i$的最大值。
我们用一个循环去遍历数组，用$map$维护每一种$s_i$上的$b_i$的和，并用$res$记录其最大值
最后得到的$res$即为结果

时间复杂度

O(n)

C++ 代码

#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
int b[maxn],s[maxn];
map<int,ll> mp;
int main(){
    ll res=0;
    int n;cin>>n;
    for(int i=1;i<=n;i++){
        cin>>b[i];
        s[i]=b[i]-i;
        if(!mp[s[i]]) mp[s[i]]=0;
        mp[s[i]]+=b[i];
        res=max(res,mp[s[i]]);
    }
    cout<<res<<endl;
    return 0;
}