Codeforces Round #817 (Div. 4) D. Line

Problem - 1722D - Codeforces

题意;

在一条水平线上有 n 个人，每个人都向左或向右看。 每个人在他们所看的方向计算人数。 该行的值是每个人计数的总和。 例如排列LRRLL，其中L代表左看的人，R代表右看的人，每个人的计数为[0,3,2,3,4]，值为0+3 +2+3+4=12。 您将获得排队人员的初步安排。 对于从 1 到 n 的每个 k，如果最多可以改变 k 个人的方向，则确定该线的最大值。

思路：

一眼贪，考虑贡献：越在两边的人越有机会得到更大的贡献

极端地考虑就是从两边到中间改变方向使得贡献更大

而我们只需考虑贡献的大小，因此只需把每个位置换个方向产生的贡献记录在数组中，从大到小排序，依次加上即可

Code：

#include <bits/stdc++.h>
using namespace std;
const int mxn=2e5+10;
#define int long long
string s;
int n,sum=0;
int a[mxn];
void init(){
    sum=0;
    s.clear();
}
bool cmp(int x,int y){
    return x>y;
}
void solve(){
    init();
    cin>>n>>s;
    s=" "+s;
    //cout<<s.size()<<endl;
    for(int i=1;i<=s.size()-1;i++){
        if(s[i]=='L') sum+=i-1;
        else sum+=n-i;
    }
    for(int i=1;i<=s.size()-1;i++){
        if(s[i]=='L'&&n-i>i-1) a[i]=n-i-i+1;
        else if(s[i]=='R'&&n-i<i-1) a[i]=i-1-n+i;
        else a[i]=0;
    }
    sort(a+1,a+1+s.size()-1,cmp);
    //cout<<sum<<endl;
    for(int i=1;i<n;i++){
        sum+=a[i];
        cout<<sum<<" ";
    }
    sum+=a[n];
    cout<<sum<<endl;
}
signed main(){
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int T;
    cin>>T;
    while(T--) solve();
}