很简单看是否能被4整除。
如果前面都是0,那么直接这段0是不可能消除的,接着遇到第一个1我们就开始从后面往前面找一个到最后一个0,因为不管是101010还是11110000最后都可以消除的只剩下一个0.
代码(比较丑):
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define mod 11092019
#define ff(a,b,c) for(int a=b;i<=c;i++)
typedef long long ll;
double pi=acos(-1);
int a[1000010];
ll d[50010],n;
ll v=0,ans=1;
void gg()
{
char ch[100010],c2[100010];
int k,p=0,l,r;
int ff=0;
bool kk=0;
cin>>k;
getchar();
for(int i=1; i<=k; i++)
scanf("%c",&ch[i]);
ch[0]='0';
if(ch[1]=='1'&&ch[k]=='0')
{
cout<<'0'<<'\n';
return;
}
for(int i=1; i<=k; i++)
{
if(ch[i]=='0') c2[p++]='0';
if(ch[i]=='1')
{
if(kk==0)
{
for(int j=k; j>i; j--)
if(ch[j]=='0')
{
c2[p++]='0';
i=j;
kk=1;
break;
}
if(kk==1)
continue;
}
c2[p++]='1';
}
}
for(int i=0; i<p; i++) cout<<c2[i];
cout<<'\n';
}
int main()
{
int n;
cin>>n;
//prime();
while(n--)
gg();
}
一开始想错了,后面才反应过来,首先我们先排序,然后把最大的k个数都给分配了,接着我们把剩下的数从剩下的数中依次从最大的开始分配,这样才能得到最优解。
代码:
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define sc(a) scanf("%lld",&a)
#define scc(a,b) scanf("%lld%lld",&a,&b)
#define sccc(a,b,c) scanf("%lld%lld%lld",&a,&b,&c)
#define scccc(a,b,c,d) scanf("%lld%lld%lld%lld",&a,&b,&c,&d)
#define sccccc(a,b,c,d,e) scanf("%lld%lld%lld%lld%lld",&a,&b,&c,&d,&e)
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define mod 11092019
#define ff(a,b,c) for(int a=b;i<=c;i++)
typedef long long ll;
double pi=acos(-1);
ll d[200010],n;
int fp[200010];
ll v=0;
void gg()
{
ll n,k,ans=0,l,r,t;
cin>>n>>k;
l=n-k+1,r=n;
for(int i=1;i<=n;i++) cin>>d[i];
for(int i=1;i<=k;i++) cin>>fp[i];
sort(d+1,d+n+1);
sort(fp+1, fp+k+1);
for(int i=n-k+1;i<=n;i++) ans+=d[i];
for(int i=1;i<=k;i++)
{
if(fp[i]==1)
{
ans+=d[r];
r--;
}
else
{
l=l-fp[i]+1;
ans+=d[l];
r--;
}
}
cout<<ans<<'\n';
}
int main()
{
int n;
cin>>n;
//prime();
while(n--)
gg();
}