#919993 (C++17) No.2505 matriX cOnstRuction

提出ソース

結果

問題	No.2505 matriX cOnstRuction
ユーザー	沙耶花
提出日時	2023-10-13 23:05:47
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	1,061 ms / 2,500 ms
コード長	3,440 bytes
コンパイル時間	4,886 ms
コンパイル使用メモリ	271,680 KB
実行使用メモリ	137,488 KB
最終ジャッジ日時	2024-09-15 18:55:40
合計ジャッジ時間	18,173 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	117 ms 5,248 KB
testcase_02	AC	92 ms 5,376 KB
testcase_03	AC	91 ms 5,376 KB
testcase_04	AC	90 ms 5,376 KB
testcase_05	AC	90 ms 5,376 KB
testcase_06	AC	124 ms 5,376 KB
testcase_07	AC	43 ms 5,376 KB
testcase_08	AC	43 ms 5,376 KB
testcase_09	AC	42 ms 5,376 KB
testcase_10	AC	42 ms 5,376 KB
testcase_11	AC	43 ms 5,376 KB
testcase_12	AC	43 ms 5,376 KB
testcase_13	AC	43 ms 5,376 KB
testcase_14	AC	42 ms 5,376 KB
testcase_15	AC	43 ms 5,376 KB
testcase_16	AC	43 ms 5,376 KB
testcase_17	AC	42 ms 5,376 KB
testcase_18	AC	43 ms 5,376 KB
testcase_19	AC	43 ms 5,376 KB
testcase_20	AC	42 ms 5,376 KB
testcase_21	AC	43 ms 5,376 KB
testcase_22	AC	43 ms 5,376 KB
testcase_23	AC	42 ms 5,376 KB
testcase_24	AC	43 ms 5,376 KB
testcase_25	AC	44 ms 5,376 KB
testcase_26	AC	44 ms 5,376 KB
testcase_27	AC	43 ms 5,376 KB
testcase_28	AC	43 ms 5,376 KB
testcase_29	AC	44 ms 5,376 KB
testcase_30	AC	28 ms 5,376 KB
testcase_31	AC	27 ms 5,376 KB
testcase_32	AC	27 ms 5,376 KB
testcase_33	AC	27 ms 5,376 KB
testcase_34	AC	37 ms 6,612 KB
testcase_35	AC	24 ms 5,520 KB
testcase_36	AC	39 ms 8,520 KB
testcase_37	AC	34 ms 6,444 KB
testcase_38	AC	49 ms 11,816 KB
testcase_39	AC	117 ms 36,312 KB
testcase_40	AC	48 ms 11,688 KB
testcase_41	AC	688 ms 134,940 KB
testcase_42	AC	51 ms 12,072 KB
testcase_43	AC	650 ms 137,488 KB
testcase_44	AC	646 ms 134,792 KB
testcase_45	AC	644 ms 137,460 KB
testcase_46	AC	687 ms 134,940 KB
testcase_47	AC	692 ms 137,476 KB
testcase_48	AC	623 ms 134,936 KB
testcase_49	AC	703 ms 134,944 KB
testcase_50	AC	713 ms 134,940 KB
testcase_51	AC	1,061 ms 5,376 KB
testcase_52	AC	379 ms 71,936 KB
testcase_53	AC	372 ms 69,404 KB
testcase_54	AC	367 ms 71,936 KB
testcase_55	AC	45 ms 5,376 KB
testcase_56	AC	43 ms 5,376 KB
testcase_57	AC	42 ms 5,376 KB
testcase_58	AC	44 ms 6,644 KB
testcase_59	AC	51 ms 8,344 KB
testcase_60	AC	56 ms 8,816 KB
testcase_61	AC	44 ms 5,376 KB
testcase_62	AC	43 ms 5,376 KB
testcase_63	AC	29 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <stdio.h>
#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
using mint = static_modint<943718401>;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 1000000000000000001
template <typename T>
struct trie{
	T init_value;
	struct node{
		vector<int> next;
		T v;
		T sum;
		int depth;
		node(int wordSize,T iv,int d){
			next.resize(wordSize,-1);
			v = iv;
			sum = iv;
			depth = d;
		}
		
		int link=-1;
	};
	vector<node> nodes;

	int wordSize;
	trie(int sz,T iv){
		init_value = iv;
		wordSize = sz;
		nodes.push_back(node(wordSize,init_value,0));
	}
	
	void add(string &S,T x,int cPos=0,int cNode=0){
		if(cPos==S.size()){
			nodes[cNode].v = func(nodes[cNode].v,x);
			return;
		}
		int c = encode(S[cPos]);
		if(nodes[cNode].next[c]==-1){
			nodes[cNode].next[c] = nodes.size();
			nodes.push_back(node(wordSize,init_value,nodes[cNode].depth+1));
		}
		
		int nextNode = nodes[cNode].next[c];
		add(S,x,cPos+1,nextNode);
	}
	
	long long dfs(int cNode=0){
		long long ret = Inf32;
		rep(i,2){
			if(nodes[cNode].next[i]!=-1)ret = min(ret,dfs(nodes[cNode].next[i])); 
			else ret = min(ret,0LL);
		}
		ret += nodes[cNode].v;
		return ret;
	}
	void set_link(){
		nodes[0].link = 0;
		queue<int> Q;
		Q.push(0);
		while(Q.size()!=0){
			int now = Q.front();
			Q.pop();
			nodes[now].v = func(nodes[now].v,nodes[nodes[now].link].v);
			nodes[now].sum = func(nodes[now].sum,nodes[now].v);
			for(int i=0;i<wordSize;i++){
				int to = nodes[now].next[i];
				if(to==-1)continue;
				int x = now;
				while(x!=0){
					x = nodes[x].link;
					if(nodes[x].next[i]!=-1){
						x = nodes[x].next[i];
						break;
					}
				}
				nodes[to].link = x;
				nodes[to].sum = func(nodes[to].sum,nodes[now].sum);
				Q.push(to);
			}
		}
	}
	
	T query(string &S,int cPos = 0,int cNode=0){
		T ret = init_value;
		if(cPos==S.size())return ret;
		int c = encode(S[cPos]);
		
		int nextNode = nodes[cNode].next[c];
		if(nextNode==-1){
			if(nodes[cNode].link!=-1){
				if(cNode!=0)ret = func(ret,query(S,cPos,nodes[cNode].link));
				else ret = func(ret,query(S,cPos+1,cNode));
			}
		}
		else{
			ret = func(ret,nodes[nextNode].v);
			ret = func(ret,query(S,cPos+1,nextNode));
		}
		
		return ret;
		
	}
	
	int encode(char c){
		return c-'0';
	}
	
	T func(T a,T b){
		return a+b;
	}
	
};

void solve(){
	int n,m;
	cin>>n>>m;
	vector<int> r(n),c(m);
	rep(i,n)cin>>r[i];
	rep(i,m)cin>>c[i];
	vector a(n,vector<int>(m));
	rep(i,n){
		rep(j,m){
			cin>>a[i][j];
		}
	}
	vector<int> rx(n),cx(m);
	rep(i,n)rx[i] = a[i][0];
	for(int j=1;j<m;j++){
		cx[j] = a[0][j] ^ rx[0];
	}
	rep(i,n){
		rep(j,m){
			if((rx[i] ^ cx[j])!=a[i][j]){
				cout<<-1<<endl;
				return;
			}
		}
	}
	trie<long long> T(2,0);
	int ans = Inf32;
	rep(_,2){
		rep(i,n){
			{
				string s = "";
				rep(j,30){
					s += '0' + ((rx[i]>>j)&1);
				}
				reverse(s.begin(),s.end());
				T.add(s,-1);
			}
			string s = "";
			bool f = false;
			for(int j=29;j>=0;j--){
				if(((r[i]>>j)&1)==0){
					if((rx[i]>>j)&1)s += '0';
					else s += '1';
					if(f)T.add(s,1);
					else T.add(s,1000000000LL);
					s.pop_back();
				}
				else f = true;
				s += (((r[i]>>j)&1) ^ ((rx[i]>>j)&1)) + '0';
			}
		}
		swap(n,m);
		swap(rx,cx);
		swap(r,c);
	}
	long long ret = T.dfs() + n + m;
	if(ret>100000000)ret = -1;
	cout<<ret<<endl;
}

int main(){
	
	int _t;
	cin>>_t;
	rep(_,_t){
		solve();
	}
	
	return 0;
}

yukicoder

結果

テストケース

ソースコード