PROBLEM D. Alarm Clock
题
OvO http://codeforces.com/contest/898/problem/D
codeforces 898d
解
从前往后枚举，放进去后不合法就拿出来，记录拿出来的次数
中途每放进去一个数，会影响到一个区间，标记这个区间的首位（做差分，首+1，尾-1），同时维护这些标记的前缀和
#include <iostream>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdio> using namespace std; const int M=1e6+44; int n,m,k;
int s[M];
int sum[M],num[M]; int main()
{
int lst,ans=0;
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)
scanf("%d",&s[i]);
sort(s+1,s+n+1);
memset(sum,0,sizeof(sum));
lst=0;
for(int i=1;i<=n;i++)
{
for(int j=lst+1;j<=s[i];j++)
sum[j]+=sum[j-1];
lst=s[i];
if(sum[s[i]]-sum[max(0,s[i]-m)]+1>=k)
ans++;
else
sum[s[i]]++;
}
printf("%d\n",ans);
return 0;
}
PROBLEM E. Squares and not squares
题
OvO http://codeforces.com/contest/898/problem/E
codeforces 898e
解
预处理出10000个左右的完全平方数，然后对于每个输入的数归类他们是否是完全平方数，并且算出各自的最小代价（二分之类的）
然后根据代价排序，从小到大补足
#include <iostream>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdio>
#include <set> using namespace std; typedef long long ll; const ll M=2e5+44; struct Node
{
ll num,val;
} g[M],ng[M]; ll lg,lng;
ll n,m;
ll s[M];
ll p[M];
ll lp;
set<ll> se; bool cmp(Node a,Node b)
{
return a.val<b.val;
} void init()
{
se.clear();
lp=100000;
for(ll i=0;i<=lp;i++)
p[i]=1ll*i*i,se.insert(p[i]);
} int main()
{
ll tmp,ans;
init();
scanf("%I64d",&n);
lg=lng=0;
for(ll i=1;i<=n;i++)
{
scanf("%I64d",&s[i]);
if(se.find(s[i])==se.end())
{
ng[++lng].num=s[i];
tmp=upper_bound(p,p+lp+1,s[i])-p;
ng[lng].val=min(p[tmp]-s[i],s[i]-p[tmp-1]);
}
else
{
g[++lg].num=s[i];
if(s[i]==0) g[lg].val=2;
else g[lg].val=1;
}
}
sort(ng+1,ng+lng+1,cmp);
sort(g+1,g+lg+1,cmp);
m=n/2;
ans=0;
if(lng>m)
{
for(ll i=1;i<=lng-m;i++)
ans+=ng[i].val;
}
else
{
for(ll i=1;i<=lg-m;i++)
ans+=g[i].val;
}
printf("%I64d\n",ans);
return 0;
}
PROBLEM F. Restoring the Expression
题
OvO http://codeforces.com/contest/898/problem/F
codeforces 898f
解
记三段为A,B,C,枚举C段的长度lc，分别判断la=lc-1,la=lc,lb=lc-1,lb=lc，这4种情况是否合法。
判断的话，做个哈希差不多的东西用来判断加法结果是否正确（哈希sed=10），截断的部分会涉及到除法，用逆元处理。哈希判断正确之后真实值再判断一次
#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <algorithm> using namespace std; const int M=1e6+44;
const int sed=10;
const int mod=1e9+7; int n;
char str[M];
int s[M];
int suf[M];
int bas[M],inv[M]; long long pr(int a, int b)
{
long long r=1,base=a;
while(b!=0)
{
if(b&1)
r=(r*base)%mod;
base=(base*base)%mod;
b>>=1;
}
return r;
} bool check(int la,int lb,int lc)
{
int sava=la,savb=lb,savc=lc;
if(la>lc || lb>lc || la<1 || lb<1 || lc<1) return false;
int lia=1,lib=la+1,lic=la+lb+1;
if((la>1 && s[lia]==0) || (lb>1 && s[lib]==0) || (lc>1 && s[lic]==0)) return false;
int ria=la,rib=la+lb,ric=la+lb+lc;
int vala,valb,valc,up=0,tmp;
while(lc>0)
{
if(la>0) vala=s[ria];
else vala=0;
if(lb>0) valb=s[rib];
else valb=0;
tmp=up+vala+valb;
up=tmp/10; tmp%=10;
if(tmp!=s[ric])
return false;
la--; lb--; lc--;
ria--; rib--; ric--;
}
if(up) return false;
for(int i=1;i<=n;i++)
{
printf("%d",s[i]);
if(i==sava) printf("+");
if(i==sava+savb) printf("=");
}
return true;
} int getval(int li,int ri)
{
int ret,tmp;
tmp=(0ll+suf[li]-suf[ri+1])%mod;
tmp=(1ll*tmp*inv[n-ri])%mod;
ret=(tmp%mod+mod)%mod;
return ret;
} bool solve(int la,int lb,int lc)
{
int sava=la,savb=lb,savc=lc;
if(la>lc || lb>lc || la<1 || lb<1 || lc<1) return false;
int lia=1,lib=la+1,lic=la+lb+1;
if((la>1 && s[lia]==0) || (lb>1 && s[lib]==0) || (lc>1 && s[lic]==0)) return false;
int ria=la,rib=la+lb,ric=la+lb+lc;
int vala,valb,valc;
vala=getval(lia,ria);
valb=getval(lib,rib);
valc=getval(lic,ric);
// cout<<la<<' '<<lb<<' '<<lc<<endl;
// cout<<vala<<' '<<valb<<' '<<valc%mod<<endl;
if((0ll+vala+valb)%mod==valc%mod && check(la,lb,lc))
return true;
else return false;
} int main()
{
scanf("%s",str+1);
n=strlen(str+1);
for(int i=1;i<=n;i++)
s[i]=str[i]-'0';
bas[0]=1;
for(int i=1;i<=n;i++)
bas[i]=1ll*bas[i-1]*sed%mod;
for(int i=0;i<=n;i++)
inv[i]=pr(bas[i],mod-2);
suf[n+1]=0;
for(int i=n;i>=1;i--)
suf[i]=(suf[i+1]+1ll*bas[n-i]*s[i]%mod)%mod;
for(int i=1;i<=n;i++)
{
if(solve(i-1,n-i-(i-1),i) || solve(i,n-i-i,i) || solve(n-i-(i-1),i-1,i) || solve(n-i-i,i,i))
return 0;
}
return 0;
}