Atcoder 281 解题报告
A题
循环输出即可
#include <bits/stdc++.h>
using namespace std;
int n;
int main(){
cin >> n;
for(int i=n;i>=0;--i){
cout << i << endl;
}
return 0;
}
B题
按照要求判断
#include <bits/stdc++.h>
using namespace std;
string s;
int main(){
cin >> s;
int len=s.size();
if(len!=8){
cout << "No" << endl;
return 0;
}
if(s[0]<'A' || s[0]>'Z'){
cout << "No" << endl;
return 0;
}
if(s[1]<='0' || s[1]>'9'){
cout << "No" << endl;
return 0;
}
for(int i=2;i<len-1;++i){
if(s[i]<'0' || s[i]>'9'){
cout << "No" << endl;
return 0;
}
}
if(s[len-1]<'A' || s[len-1]>'Z'){
cout << "No" << endl;
return 0;
}
cout << "Yes" << endl;
return 0;
}
C题
打一个前缀和数组,去掉时间内所有的循环,判断在哪一个区间即可
#include <bits/stdc++.h>
using namespace std;
#define ll long long
int n,a[100010];
ll t,tim[100010],sum=0;
int main(){
cin >> n >> t;
for(int i=1;i<=n;++i){
cin >> a[i];
sum+=a[i];
}
tim[1]=0;
for(int i=2;i<=n;++i)
tim[i]=tim[i-1]+a[i-1];
ll l=t/sum;
t-=sum*l;
int op=0;
for(int i=1;i<=n;++i){
if(tim[i]>t){
op=i-1;
break;
}
}
if(op==0)
op=n;
cout << op << " ";
cout << t-tim[op] << endl;
return 0;
}
D题
动规
dp(i,j,k)表示在前 i 个数中选择 j 个数的最大的和,使这个和 %d 等于 k。
显然,对于第i个数来说有两种情况,选还是不选。
如果选,dp(i,j,k)=dp(i-1,j-1,( (k-a[i])%d+d)%d )+a[i]
如果不选,dp(i,j,k)=dp(i-1,j,k)。
而解就是dp(n,k,0)。
#include <bits/stdc++.h>
using namespace std;
int n,kk,d,a[110];
long long dp[110][110][110];
int main(){
memset(dp,-1,sizeof(dp));
dp[0][0][0]=0;
cin >> n >> kk >> d;
for(int i=1;i<=n;++i)
cin >> a[i];
for(int i=1;i<=n;++i)
for(int j=0;j<=i;++j)
for(int k=0;k<d;++k){
dp[i][j][k]=dp[i-1][j][k];
if(dp[i-1][j-1][((k-a[i])%d+d)%d]!=-1 && j>0)
dp[i][j][k]=max(dp[i][j][k],dp[i-1][j-1][(((k-a[i])%d)+d)%d]+a[i]);
}
cout << dp[n][kk][0] << endl;
return 0;
}
E题
优先队列
定义两个优先队列,L是一个大根堆,用来存需要相加的K个数,R是一个小根堆,用来存K个数之外的在当前区间里的数。
每当这个区间变化时,就需要判断所抛弃的第i-m个数在不在L里,如果在,则从R和a[i]中选出一个较小数存进L中,另一个数存进R中。如果不在,则判断a[i]和L中的最大数的大小,如果a[i]较小,则把a[i]加入L中,L被替换出来的数加入R中。否则把a[i]加入R中。
#include <bits/stdc++.h>
using namespace std;
#define ll long long
int n,m,k,a[200010];
bool vh[200010];
struct nod{
ll w;
ll id;
nod(ll _w=0,ll _id=0):w(_w),id(_id){};
};
nod tem[200010];
bool cmp(nod a,nod b){
return a.w<b.w || (a.w==b.w && a.id<b.id);
}
struct cmpl{
bool operator()(nod p,nod q){
return p.w<q.w || (p.w==q.w && p.id>q.id);
}
};
struct cmpr{
bool operator()(nod p,nod q){
return p.w>q.w || (p.w==q.w && p.id>q.id);
}
};
int main(){
//freopen("data.in","r",stdin);
//freopen("zhengjie.out","w",stdout);
cin >> n >> m >> k;
for(int i=1;i<=n;++i)
cin >> a[i];
priority_queue<nod,vector<nod>,cmpr> r;
priority_queue<nod,vector<nod>,cmpl> l;
for(int i=1;i<=m;++i)
tem[i].w=a[i],tem[i].id=i;;
sort(tem+1,tem+m+1,cmp);
ll ans=0;
ll maxx=0,maxid=0;
for(int i=1;i<=m;++i){
if(i<=k){
l.push(nod(tem[i].w,tem[i].id));
if(maxx<=tem[i].w)
maxx=tem[i].w,maxid=tem[i].id;
vh[tem[i].id]=1;
ans+=tem[i].w;
}
else
r.push(nod(tem[i].w,tem[i].id));
}
cout << ans << " ";
for(int i=m+1;i<=n;++i){
if(vh[i-m]==1){
vh[i-m]=0;
ans-=a[i-m];
nod x;
while(!r.empty()){
x=r.top();
if(x.id<=i-m)
r.pop();
else
break;
}
if(x.w<=a[i] && x.w!=0){
r.push(nod(a[i],i));
r.pop();
l.push(x);
vh[x.id]=1;
if(x.w>maxx)
maxx=x.w,maxid=x.id;
ans+=x.w;
}
else{
l.push(nod(a[i],i));
vh[i]=1;
if(a[i]>maxx || maxid==i-m){
nod t;
while(!l.empty()){
t=l.top();
if(vh[t.id]==0)
l.pop();
else
break;
}
maxx=t.w,maxid=t.id;
}
ans+=a[i];
}
}
else{
nod x;
if(a[i]<maxx){
ans-=maxx;
ans+=a[i];
vh[maxid]=0;
l.push(nod(a[i],i));
vh[i]=1;
while(!l.empty()){
x=l.top();
if(vh[x.id]==0)
l.pop();
else
break;
}
r.push(nod(maxx,maxid));
maxx=x.w;
maxid=x.id;
}
else
r.push(nod(a[i],i));
}
cout << ans << " ";
}
return 0;
}