30pts求调
查看原帖
30pts求调
405894
233L楼主2021/10/6 12:05

提交记录

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,d;
ll ans;
string s;
vector<int>a[5];
int c[4][4];
map<string,int>mp;

int bits(int x){
	int re=0;
	for(int i=8;i;i>>=1)
		if(x&i)re++;
	return re;
}
void init(){
	scanf("%d%d",&n,&d);
	for(int i=0;i<16;i++)
		a[bits(i)].push_back(i);
	c[0][0]=1;
	for(int i=1;i<4;i++){
		c[i][0]=1;
		for(int j=1;j<=i;j++)
			c[i][j]=c[i-1][j]+c[i-1][j-1];
	}
}
void aa(){
	string t;
	int re=0;
	int tru[4]={};
	for(int i=1;i<=d;i++){
		for(int j=0;j<a[i].size();j++){
			t=s;
			for(int k=0;k<4;k++)
				if(a[i][j]&(1<<k))t[k]='#';
			re+=mp[t];
		}
		for(int j=1;j<i;j++)
			re-=tru[j]*c[i][j];
		tru[i]=re;
	}
	ans+=tru[d];
	//printf("%d\n",tru[d]);
}
void add(){
	string t;
	for(int i=1;i<=d;i++)
		for(int j=0;j<a[i].size();j++){
			t=s;
			for(int k=0;k<4;k++)
				if(a[i][j]&(1<<k))t[k]='#';
			mp[t]++;
		}
}
int main(){
	init();
	
	for(int i=1;i<=n;i++){
		cin>>s;
		aa();
		add();
	}
	printf("%lld",ans);
} 

不太清楚哪里错了,看的第一篇题解,那篇题解是反着写的(即把不同改为相同),我写的是直接求不同的个数。

2021/10/6 12:05
加载中...