Codeforces–Build a Contest
测试样例:
input
3 11
2 3 1 2 2 2 3 2 2 3 1
output
00100000001
inputCopy
4 8
4 1 3 3 2 3 3 3
output
00001000
这道题的题意就是说:从给定的m个题目中,每刚好够一轮的难度(也就是n)则对应的题目数输出1,其余的都输出0。
比如第一个样例,第三个题目与前面的两个题目刚好可以组成一轮,则在3处输出1,被拿走的题目不再参与下一轮的挑选。最后一个1与前面的又可以组成一轮,则在最后输出1,其余的位置皆输出0。
这道题我的想法是:声明一个b数组用来判断是否满足一轮,从1到m开始遍历,并且b[a[i]]++;再用一个for循环(j),从1到n遍历,用b数组来查看是否所有的b[j]都大于等于1,如果是,则满足一轮,则输出1,否则输出0。
但是这样做代码的时间复杂度是o(m*n)
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
int a[100005];
int b[100005];
int main()
{
memset(b,0,sizeof(b));
int n,m,i,j,k;
int x,num=0;
cin>>n>>m;
for(i=1;i<=m;i++)
cin>>a[i];
for(i=1;i<=m;i++)
{
int flag=1;
b[a[i]]++;
for(j=1;j<=n;j++)
{
if(b[j]<1)
flag=0;
}
if(flag==1)
{
cout<<1;
for(j=1;j<=n;j++)
b[j]--;
}
else
cout<<0;
}
cout<<endl;
return 0;
}
很明显这样是超时了的,代码的时间复杂度是1e10;
在学长的指导下,我对代码进行了优化
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
int a[100005];
int b[100005];
int main()
{
memset(b,0,sizeof(b));
int n,m,i,j,k;
int x,num=0;
cin>>n>>m;
for(i=1;i<=m;i++)
cin>>a[i];
for(i=1;i<=m;i++)
{
x=a[i];
if(b[x]==0)
num++;//如果这一轮中的数字没有出现过,则num++;
b[x]++;//数字每次出现,对应的b数组都会++;
if(num==n) //只有当num==n的时候才会进行下面的for循环
{
cout<<1;
num=0;
for(j=1;j<=n;j++)
{
b[j]--;
if(b[j]>0)
num++;
}
}
else
cout<<0;
}
cout<<endl;
return 0;
}
优化后的代码是以num来判断是否达到了一轮,这样代码的时间复杂度就缩减为o(n*m/n)了。
班门弄斧,见笑。
谢谢。