北京师范大学第十六届程序设计竞赛决赛-重现赛 I-如何办好比赛

北京师范大学第十六届程序设计竞赛决赛-重现赛 I-如何办好比赛

链接：https://www.nowcoder.com/acm/contest/117/I

来源：牛客网

题目描述

又到了一年一度的程序设计大赛了~
 
现在参赛选手在机房前排起了一列长队，这里面有萌新也有大佬，萌新都很仰慕大佬，由于大佬们的参赛，萌新们对这次比赛的精彩程度格外期待。对于每个萌新来说，他/她/它对本次的比赛的期待度为排在他/她/它前面的大佬的数量，而这次比赛的总期待度等于每个萌新的期待度之和。
 
SK同学作为本次比赛的组织者，希望比赛的期待度能够刚刚好，太低的话会让大家兴致不高，太高的话会被喷不满足预期。
 
现在SK同学可以交换任意相邻的两名参赛选手，但是比赛马上就要开始了，SK同学想知道最少要进行多少次交换才能使得这次比赛的总期待度刚好为k，你能帮帮他吗？

输入描述:

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

对于每组测试数据，

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

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

输出描述:

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

示例1

输入

2
3 1
DMM
3 3
DMM

输出

1
-1

---

思路

先计算一下当前的期待度是多少，最少交换次数即是它们的差值（的绝对值）。

如果所给的k值过大，那么即是无解。

AC代码

#include <iostream>
#include <cstring>
typedef long long ll;
using namespace std;

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