Submission #856484

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
856484	2023-10-03T16:27:01 Z	MilosMilutinovic	PIN (CEOI10_pin)	C++14	60 / 100	298 ms	14796 KB

pin

#include <bits/stdc++.h>

#define pb push_back
#define fi first
#define se second
#define mp make_pair

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;

int readint(){
	int x=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}

int n,d;
char s[50005][4];

int main(){
	scanf("%d%d",&n,&d); d=4-d;
	for(int i=1;i<=n;i++) scanf("%s",s+i);
	map<string,int> cnt;
	long long ans=0;
	for(int i=1;i<=n;i++){
		for(int mask=(1<<4)-1;mask>=0;mask--){
			int pc=__builtin_popcount(mask);
			if(pc<d) continue;
			string str="";
			for(int j=0;j<4;j++) if(mask>>j&1) str+=s[i][j]; else str+='*';
			if (d==4){
				ans+=cnt[str];
				continue;
			}
			if (d==3){
				if(pc==4) ans-=4*cnt[str];
				if(pc==3) ans+=cnt[str];
				continue;
			}
			if(d==2){
				if(pc==3) ans-=3*cnt[str];
				if(pc==2) ans+=cnt[str];
				continue;
			}
			if(d==1){
				if(pc==2) ans-=2*cnt[str];
				if(pc==1) ans+=cnt[str];
				continue;
			}
			if(d==0){
				assert(false);
			}
			ans+=cnt[str]*(d%2==pc%2?1:-1);
		}
		for(int mask=0;mask<(1<<4);mask++){
			string str="";
			for(int j=0;j<4;j++) if(mask>>j&1) str+=s[i][j]; else str+='*';
			cnt[str]+=1;
		}
	}
	printf("%lld\n",ans);
	return 0;
}

/*
4 1
0000
a010
0202
a0e2

4 2
0000
a010
0202
a0e2
*/

Compilation message

pin.cpp: In function 'int main()':
pin.cpp:25:32: warning: format '%s' expects argument of type 'char*', but argument 2 has type 'char (*)[4]' [-Wformat=]
   25 |  for(int i=1;i<=n;i++) scanf("%s",s+i);
      |                               ~^  ~~~
      |                                |   |
      |                                |   char (*)[4]
      |                                char*
pin.cpp:24:7: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   24 |  scanf("%d%d",&n,&d); d=4-d;
      |  ~~~~~^~~~~~~~~~~~~~
pin.cpp:25:29: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   25 |  for(int i=1;i<=n;i++) scanf("%s",s+i);
      |                        ~~~~~^~~~~~~~~~

Subtask #1

60.0 / 100.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	7 ms	1112 KB	Output is correct
2	Correct	7 ms	856 KB	Output is correct
3	Runtime error	1 ms	604 KB	Execution killed with signal 6
4	Correct	76 ms	3428 KB	Output is correct
5	Correct	95 ms	3664 KB	Output is correct
6	Correct	113 ms	3652 KB	Output is correct
7	Correct	90 ms	3412 KB	Output is correct
8	Correct	111 ms	4104 KB	Output is correct
9	Correct	173 ms	5460 KB	Output is correct
10	Correct	230 ms	5720 KB	Output is correct
11	Correct	127 ms	3920 KB	Output is correct
12	Incorrect	215 ms	5456 KB	Output isn't correct
13	Incorrect	132 ms	4260 KB	Output isn't correct
14	Runtime error	2 ms	604 KB	Execution killed with signal 6
15	Runtime error	3 ms	764 KB	Execution killed with signal 6
16	Correct	169 ms	11240 KB	Output is correct
17	Correct	298 ms	14796 KB	Output is correct
18	Runtime error	2 ms	860 KB	Execution killed with signal 6
19	Runtime error	3 ms	860 KB	Execution killed with signal 6
20	Runtime error	3 ms	860 KB	Execution killed with signal 6