AtCoder Beginner Contest 324

C - Error Correction

大意是：给定一个字符串a，以及一组字符串，如果字符串与a满足以下之一即可

我写的有点麻烦。。

#include<bits/stdc++.h>
using namespace std;
void solve(){
	int n;
	cin>>n;
	string s;
	cin>>s;
	vector<int>ans;
	for(int i=1;i<=n;i++){
		string t;
		cin>>t;
		int a=0,b=0,c=0,d=0;
		if(t.size()==s.size()){
			int cnt=0;
			for(int i=0;i<s.size();i++){
				if(s[i]!=t[i])cnt++;
				if(cnt>1){
					break;
				}
			}
			if(cnt==0)b=1;
			if(cnt==1)a=1;
		}else if(t.size()==s.size()+1){
			int cnt=0;
			for(int i=0,j=0;i<s.size();i++,j++){
				if(s[i]!=t[j]){
					j++;
					if(s[i]!=t[j]){
						cnt++;
						break;
					}
				}
			}
			if(!cnt)c=1;
		}else if(s.size()==t.size()+1){
			int cnt=0;
			for(int i=0,j=0;i<t.size();i++,j++){
				if(t[i]!=s[j]){
					j++;
					if(t[i]!=s[j]){
						cnt++;
						break;
					}
				}
			}
			if(!cnt)d=1;
		}
		if((a&&!b&&!c&&!d)||(!a&&b&&!c&&!d)||(!a&&!b&&c&&!d)||(!a&&!b&&!c&&d)){
			ans.push_back(i);
		}
	}
	cout<<ans.size()<<"\n";
	for(auto c:ans){
		cout<<c<<" ";
	}
}
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int t=1;
	//cin>>t;
	for(int i=1;i<=t;i++)solve();
	return 0;
}