QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#632014#9457. Prime Setucup-team3555#AC ✓41ms34972kbC++172.7kb2024-10-12 11:29:372024-10-12 11:29:37

Judging History

你现在查看的是测评时间为 2024-10-12 11:29:37 的历史记录

  • [2024-10-13 19:27:34]
  • 自动重测本题所有获得100分的提交记录
  • 测评结果:AC
  • 用时:66ms
  • 内存:33724kb
  • [2024-10-13 19:27:24]
  • hack成功,自动添加数据
  • (/hack/967)
  • [2024-10-13 18:37:57]
  • 管理员手动重测本题所有得分≥97分的提交记录
  • 测评结果:100
  • 用时:44ms
  • 内存:32880kb
  • [2024-10-13 18:24:51]
  • hack成功,自动添加数据
  • (/hack/965)
  • [2024-10-13 18:17:54]
  • hack成功,自动添加数据
  • (/hack/963)
  • [2024-10-12 11:29:37]
  • 评测
  • 测评结果:100
  • 用时:41ms
  • 内存:34972kb
  • [2024-10-12 11:29:37]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
#define pb push_back
const ll N=1e4+3,INF=(ll)1e18;
struct Flow
{
	ll tot=-1;
	vector<ll>ve[N],to,lim,val; 
	int Add(int x,int y,ll z,ll w)
	{
		ve[x].pb(++tot);to.pb(y);lim.pb(z);val.pb(w);
		ve[y].pb(++tot);to.pb(x);lim.pb(0);val.pb(-w);
		return tot-1;
	}
	ll S,T,cost,dis[N],cur[N];
	bool vis[N];
	bool Spfa()
	{
		memset(cur,0,sizeof(cur));
		memset(dis,0x3f,sizeof(dis));
		memset(vis,0,sizeof(vis));
		queue<int>q;dis[S]=0;q.push(S);
		while(!q.empty())
		{
			int x=q.front();q.pop();vis[x]=0;
			for(int id:ve[x])if(dis[to[id]]>dis[x]+val[id]&&lim[id])
			{
				dis[to[id]]=dis[x]+val[id];
				if(!vis[to[id]])q.push(to[id]),vis[to[id]]=1;
			}
		}
		return dis[T]<INF;
	}
	ll Dfs(ll x,ll res)
	{
		if(x==T)return res;
		ll flow=0;vis[x]=1;
		for(int i=cur[x];i<(int)ve[x].size()&&res;i++)
		{
			cur[x]=i;ll id=ve[x][i],y=to[id],w=min(res,lim[id]),z;
			if(!vis[y]&&dis[x]+val[id]==dis[y]&&w)
			    z=Dfs(y,w),flow+=z,res-=z,lim[id]-=z,lim[id^1]+=z,cost+=z*val[id];
		}
		if(!flow)dis[x]=INF;
		vis[x]=0;return flow;
	}
	ll Maxflow(int s,int t)
	{
		S=s;T=t;ll flow=0;cost=0;
		while(Spfa())flow+=Dfs(S,INF);
		return flow;
    }
    void Clear(int sum)
    {
    	for(int i=0;i<=sum;i++)ve[i].clear();
    	to.clear();lim.clear();val.clear();tot=-1;
	}
}G;
const int K=2e6+3;
int n,KK,S,T,cnt,a[N],b[N],c[N],pri[K];
bool vis[K];
void Solve()
{
	cin>>n>>KK;int S=n+1,T=S+1;int m=0;
	for(int i=1;i<=n;i++)cin>>a[i],b[i]=c[i]=0;
	sort(a+1,a+n+1);G.Clear(T);
	for(int i=1;i<=n;i++)if(a[i]==1)m=i;
	for(int i=m+1;i<=n;i++)if(a[i]%2==0)b[i]=G.Add(S,i,1,0);else b[i]=G.Add(i,T,1,0);
	for(int i=m+1;i<=n;i++)for(int j=i+1;j<=n;j++)if(!vis[a[i]+a[j]])
	{
		int x=i,y=j;
		if(a[x]%2==1)swap(x,y);
		G.Add(x,y,1,1+(vis[a[x]+1]==0));c[x]=c[y]=1;
	}
	int flow=G.Maxflow(S,T);vector<int>cur;
	int nb[2]={0},kk=0,mk=0;
	if(!m)
	{
		if(flow>=KK){cout<<KK*2<<endl;return;}
		for(int i=m+1;i<=n;i++)if(G.lim[b[i]]&&c[i])kk++;
		cout<<flow*2+min(KK-flow,kk)<<endl;return;
	}
	for(int i=1;i<=n;i++)if(a[i]%2==0&&G.lim[b[i]])cur.push_back(i);
	sort(cur.begin(),cur.end(),[&](int X,int Y){return c[X]<c[Y];});
	for(int x:cur)if(!vis[a[x]+1])
	{
		if(m)m--,nb[c[x]]++,c[x]=0;
		else kk++,c[x]=0; 
	}
	for(int i=m+1;i<=n;i++)if(G.lim[b[i]]&&c[i])kk++;
	flow+=m/2+nb[0]+nb[1];
	if(m%2==1)for(int i=2;i<=n;i++)mk|=!vis[a[i]+1]; 
	if(flow>=KK){cout<<KK*2<<endl;return;}
	cout<<flow*2+min(KK-flow,kk+mk)<<endl;
}
void Init()
{
	for(int i=2;i<K;i++)
	{
		if(!vis[i])pri[++cnt]=i;
		for(int j=1;j<=cnt&&pri[j]*i<K;j++)
		{
			vis[pri[j]*i]=1;
			if(i%pri[j]==0)break;
		}
	}
}
int main()
{
    int _;cin>>_;Init();
    while(_--)Solve();
}

这程序好像有点Bug,我给组数据试试?

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 8ms
memory: 7144kb

input:

4
4 2
2 3 4 5
5 3
3 4 12 3 6
6 3
1 3 6 8 1 1
1 0
1

output:

4
3
6
0

result:

ok 4 lines

Test #2:

score: 0
Accepted
time: 41ms
memory: 34972kb

input:

110
3 3
2 2 2
3 1
1 1 1
3 2
1 1 1
3 3
1 1 1
4 2
1 1 1 1
10 7
1 1 1 2 4 6 8 10 12 14
18 1
10 5 9 4 10 7 2 4 9 1 6 1 4 9 2 4 9 7
19 5
1 6 1 8 8 1 5 10 9 2 10 2 1 9 6 10 4 3 6
14 4
6 6 8 9 5 3 4 9 2 2 1 10 10 1
15 10
5 4 2 4 10 1 8 4 10 3 10 3 7 10 4
17 5
10 7 2 4 3 1 10 8 2 8 3 4 1 1 1 1 1
20 6
8 4 6 ...

output:

0
2
3
3
4
8
2
10
8
15
10
12
10
10
12
16
10
10
12
16
14
13
13
14
17
0
19
12
12
11
16
10
19
2
12
10
5
14
10
10
13
2
18
2
4
4
11
8
11
8
0
10
10
0
10
0
8
15
12
11
13
18
10
17
9
8
20
19
13
6
4
6
11
9
13
11
6
2
8
12
17
14
6
2
20
2
18
17
6
2
10
0
7
16
2
2
0
2
18
14
11
8
4
6
0
19
890
204
2746
2372

result:

ok 110 lines

Extra Test:

score: 0
Extra Test Passed