A. Tit for Tat
题意:给定一个长度为n的数组a和一个可以操作的次数k,每次操作可以在数组a中选择一个数使其-1并选择另一个数使其+1,问经过k次操作后可以使得字典序最小的序列是什么。
关于字典的比较是这样的:从左边第一位开始,如果两个串这一位上的字符不一样,则结束比较,这一位上字符小的串字典序更小,否则继续比较下一位。特殊的,如果比较至某一位时某个串比另一个串长度短提前结束,则长度短的串字典序更小。
思路:
本题用的是数字来比较,其实思想是一样的,每次操作可以选择一个一个数+1,选择一个数-1。那么可以发现,我们-1的操作对越往前的数进行操作,操作后的字典序的减小就越大,而+1的操作对越往后的数进行操作,会使整个串的字典序增幅越小。那么我们用一个指针i
指向串的第一位,每次操作使a[i]
减1(如果a[i]
为0则将i
向后移一位),使a[n]
加1(a[n]
为最后一位元素),就可以使操作后的字典序最小
代码:
#include <cstdio>
#include <iostream>
using namespace std;
int a[110];
int main(){
int t;
cin >> t;
while(t--){
int n, k;
cin >> n >> k;
for(int i = 1; i <= n; i++) cin >> a[i];
for(int i = 1; i <= n && k;){
if(a[i] == 0) i++;
else a[i]--, a[n]++, k--;
}
for