北京师范大学程序设计总决赛-I. 如何办好比赛

题目链接

http://www.bnuoj.com/v3/contest_show.php?cid=9358#problem/I

题目

又到了一年一度的程序设计大赛了~

现在参赛选手在机房前排起了一列长队，这里面有萌新也有大佬，萌新都很仰慕大佬，由于大佬们的参赛，萌新们对这次比赛的精彩程度格外期待。对于每个萌新来说，他/她/它对本次的比赛的期待度为排在他/她/它前面的大佬的数量，而这次比赛的总期待度等于每个萌新的期待度之和。

SK同学作为本次比赛的组织者，希望比赛的期待度能够刚刚好，太低的话会让大家兴致不高，太高的话会被喷不满足预期。

现在SK同学可以交换任意相邻的两名参赛选手，但是比赛马上就要开始了，SK同学想知道最少要进行多少次交换才能使得这次比赛的总期待度刚好为k，你能帮帮他吗？

Input

第一行是一个正整数T(<= 10)，表示测试数据的组数，

对于每组测试数据，

第一行是两个正整数n(<= 1000000)和k(<= 1000000000)，分别表示队列长度和最终的比赛总期待度，

接下来一行包含n个字符，表示这个队列，第i个字符表示队列里的第i个人，’D’表示大佬，’M’表示萌新，保证不会出现其它字符。

Output

对于每组测试数据，输出一行，包含一个整数，表示最少的交换次数，无解输出-1。

Sample Input

2
3 1
DMM
3 3
DMM

Sample Output

1
-1

分析

把D看作0，M看作1。原字符队列变成01序列，对于每个1求出其左边所有的0的个数，累加起来就是当前的期望度。
如果想要使期望度变小，将0和右边的1交换即可。遍历一遍队列求出每个0右边的1的数量，累加起来就是期望能减小的极限。

同理，累加每个1右边的0的数量就是期望能增加的极限。
然后看k是否在当前期望能变化的范围内即可。注意要用long long.

时间复杂度O(N).

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        int n,k;
        cin>>n>>k;
        string s;
        cin>>s;
        ll sum=0,ans=0;
        for(int i=0; i<n; i++)
        {
            if(s[i]=='D')
                sum++;
            else
                ans+=sum;
        }
        if(ans>=k)
        {
            sum=0;
            ll nen=0;
            for(int i=n-1; i>=0; i--)
            {
                if(s[i]=='M') sum++;
                else nen+=sum;
            }
            if(ans-nen<=k) cout<<ans-k<<endl;
            else cout<<-1<<endl;
        }
        else
        {
            sum=0;
            ll nen=0;
            for(int i=0;i<=n-1;i++)
            {
                if(s[i]=='M') sum++;
                else nen+=sum;
            }
            if(ans+nen>=k) cout<<k-ans<<endl;
            else cout<<-1<<endl;
        }
    }
    return 0;
}