QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#490688#7613. Inverse Problemucup-team052WA 9207ms810392kbC++233.9kb2024-07-25 16:06:232024-07-25 16:06:29

Judging History

你现在查看的是最新测评结果

  • [2024-07-25 16:06:29]
  • 评测
  • 测评结果:WA
  • 用时:9207ms
  • 内存:810392kb
  • [2024-07-25 16:06:23]
  • 提交

answer

#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const int md = 1e9 + 7;
const int mod = 1e9 + 7;

inline int add(int x, int y) {
	if (x + y >= md) return x + y - md;
	return x + y;
}

inline int sub(int x, int y) {
	if (x < y) return x - y + md;
	return x - y;
}

inline int mul(int x, int y) {
	return 1ull * x * y % md;
}
inline int mul(int x,int y,int z) {return mul(mul(x,y),z);}

inline int qpow(int x, int y) {
	int ans = 1;
	while (y) {
		if (y & 1) ans = mul(ans, x);
		y >>= 1; x = mul(x, x);
	}
	return ans;
}
int Inv(int x) {return qpow(x,mod-2);}
const int L=130;
const int LIM=6;
int n,dv[L+5][L+5],idv[L+5][L+5];
int Q,q[15],ok[15],sok;
vector<int> ans[15];
int ansn[15];
vector<pair<int,__int128>> small[L],large[L];
bitset<mod> vis;
void dfs_small(int sum,int cur,int prod,__int128 hsh)
{
	small[sum].emplace_back(prod,hsh);
	for(int i=cur;i<=LIM&&sum+i<=n-2;i++)
	{
		dfs_small(sum+i,i,mul(prod,dv[n-2][i]),((hsh<<1)|1)<<(i-1));
	}
}
void dfs_large(int sum,int cur,int prod,__int128 hsh)
{
	large[sum].emplace_back(prod,hsh);
	for(int i=cur;sum+i<=n-2;i++)
	{
		dfs_large(sum+i,i,mul(prod,dv[n-2][i]),((hsh<<1)|1)<<(i-1));
	}
}
void init()
{
	for(int i=0;i<=n;i++) small[i].clear(),large[i].clear();
	dfs_small(0,1,1,0);
	dfs_large(0,LIM+1,1,0);
}
vector<int> getvec(int L,__int128 hl,int R,__int128 hr)
{
	vector<int> ans;
	int cur=0;
	while(hl)
	{
		cur++;
		if(hl&1) ans.push_back(cur),cur=0;
		hl>>=1;
	}
	cur=0;
	while(hr)
	{
		cur++;
		if(hl&1) ans.push_back(cur),cur=0;
		hr>>=1;
	}
	return ans;
}
int main() {
 #ifdef xay5421
	freopen("a.in","r",stdin);
 #endif
	for(int i=1;i<=L;i++)
	{
		dv[i][0]=1;
		for(int j=1;j<=L;j++) dv[i][j]=mul(dv[i][j-1],i-j+1),idv[i][j]=Inv(dv[i][j]);
	}
	cin>>Q;
	for(int i=1;i<=Q;i++)
	{
		cin>>q[i];
		if(q[i]==1) ok[i]=1,sok++;
		if(q[i]==2)
		{
			ok[i]=1,sok++;
		}
	}
	for(n=3;n<=125&&sok<Q;n++)
	{
		fprintf(stderr,"n = %d\n",n);
		init();
		int adw=mul(n,n-1);
		for(int L=0;L<=n-2;L++)
		{
			int R=n-2-L;
			// printf("%d %d\n",(int)small[L].size(),(int)large[R].size());
			if((int)small[L].size()<(int)large[R].size())
			{
				for(auto [p,hsh]:large[R]) vis.set(p);
				for(auto [p,hsh]:small[L])
				{
					int cv=mul(adw,p),icv=Inv(cv);
					for(int qid=1;qid<=Q;qid++)
					{
						if(ok[qid]) continue;
						int ned=mul(q[qid],icv);
						if(vis[ned])
						{
							for(auto [p_,hl]:large[R]) if(p_==ned)
							{
								// printf("%d : %d %d : %d %d\n",qid,p,p_,(int)hl,(int)hsh);
								ans[qid]=getvec(L,hl,R,hsh);
								ansn[qid]=n;
								ok[qid]=1,sok++;
								break;
							}
						}
					}
				}
				for(auto [p,hsh]:large[R]) vis.reset(p);
			}
			else
			{
				for(auto [p,hsh]:small[L]) vis.set(p);
				for(auto [p,hsh]:large[R])
				{
					int cv=mul(adw,p),icv=Inv(cv);
					for(int qid=1;qid<=Q;qid++)
					{
						if(ok[qid]) continue;
						int ned=mul(q[qid],icv);
						if(vis[ned])
						{
							for(auto [p_,hl]:small[L]) if(p_==ned)
							{
								// printf("%d : %d %d : %d %d\n",qid,p,p_,(int)hl,(int)hsh);
								ans[qid]=getvec(L,hl,R,hsh);
								ansn[qid]=n;
								ok[qid]=1,sok++;
								break;
							}
						}
					}
				}
				for(auto [p,hsh]:small[L]) vis.reset(p);
			}
		}
	}
	for(int i=1;i<=Q;i++)
	{
		if(q[i]==1) printf("1\n");
		else if(q[i]==2) printf("2\n1 2\n");
		else
		{
			printf("%d\n",ansn[i]);
			while((int)ans[i].size()<ansn[i]) ans[i].push_back(0);
	//		for(int j=0;j<(int)ans[i].size();j++) ans[i][j]++;
			for(int j=0;j<ansn[i]-1;j++)
			{
				for(int c=0;c<(int)ans[i].size();c++) for(int d=0;d<(int)ans[i].size();d++)
				{
					if(ans[i][c]==0&&ans[i][d]>0)
					{
						ans[i][c]--,ans[i][d]--;
						printf("%d %d\n",c+1,d+1);
					}
				}
			}
			for(int c=0;c<(int)ans[i].size();c++) for(int d=c+1;d<(int)ans[i].size();d++)
			{
				if(ans[i][c]==0&&ans[i][d]==0)
				{
					ans[i][c]--,ans[i][d]--;
					printf("%d %d\n",c+1,d+1);
				}
			}
		}
	}
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 3ms
memory: 5944kb

input:

4
2
360
1
509949433

output:

2
1 2
5
3 1
4 1
5 2
1 2
1
10
9 1
10 2
1 3
2 4
3 5
4 6
5 7
6 8
7 8

result:

ok OK (4 test cases)

Test #2:

score: -100
Wrong Answer
time: 9207ms
memory: 810392kb

input:

9
185396120
468170792
837583517
696626231
338497514
762842660
800028852
928391161
733524004

output:

14
11 1
12 1
13 2
14 2
1 3
2 4
3 5
4 6
5 7
6 8
7 9
8 10
9 10
122
7 1
8 1
9 1
10 1
11 1
12 1
13 1
14 1
15 1
16 1
17 1
18 1
19 1
20 1
21 1
22 1
23 1
24 1
25 1
26 1
27 1
28 1
29 1
30 1
31 1
32 1
33 1
34 1
35 1
36 2
37 2
38 2
39 2
40 2
41 2
42 2
43 2
44 2
45 2
46 2
47 2
48 2
49 2
50 3
51 3
52 3
53 3
54 ...

result:

wrong answer Graf nie jest drzewem!