A - Tit for Tat CodeForces - 1516A
题意:
一个长度为 n 的数组,现在希望你对这个数组进行至多k次操作:在数组中选择 2 个不同的元素,将第一个元素加 1 ,然后从第二个元素减 1 。不过他有一个要求,进行操作后数组中的所有元素都必须保持非负数。在进行至多 k 次操作之后,你能否获得一个数组在字典序上最小呢?
题解:
要是的字典序最小,就要让高位置尽可能减,低位置尽可能加,注意跳过0的情况
代码:
#include<bits/stdc++.h>
#define debug(a,b) printf("%s = %d\n",a,b);
typedef long long ll;
using namespace std;inline int read(){int s=0,w=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();//s=(s<<3)+(s<<1)+(ch^48);return s*w;
}
const int maxn=400;
int a[maxn];
int main()
{int t;cin>>t;while(t--){memset(a,0,sizeof(a));int n,k;cin>>n>>k;for(int i=1;i<=n;i++){cin>>a[i];}int l=1,r=n;while(k&&l<r){while(a[l]==0)l++;if(l>=n)break;if(a[l]>0)a[l]--;a[r]++;k--; }for(int i=1;i<=n;i++)cout<<a[i]<<" ";cout<<endl;}return 0;
}