选数（前缀和）

最新推荐文章于 2024-10-05 19:23:25 发布

zzzyyzz_

最新推荐文章于 2024-10-05 19:23:25 发布

阅读量37

点赞数

分类专栏：基础算法文章标签：算法

本文链接：https://blog.csdn.net/zzzyyzz_/article/details/130111755

版权

基础算法专栏收录该内容

7 篇文章 0 订阅

订阅专栏

文章介绍了如何使用前缀和的思想解决一道编程题目。当求解一组数中是否存在连续子集，其和能被n整除时，通过计算并存储每个位置的前缀和模n的结果，可以找到满足条件的子集。如果不存在使得前缀和模n为0的子集，那么根据鸽巢原理，一定存在两个不同的前缀和模n相等，从而找到一个合法的子集。代码实现中运用了C++的数据结构和算法，包括数组、循环和条件判断等。

摘要由CSDN通过智能技术生成

选数

选数 - 题目 - Daimayuan Online Judge

思路：我们考虑用前缀和的思想去解题，假设sum[i]为前缀和%n的结果，那么如果存在一个k是的sum[k]%n==0，那么代表从1~k着k个数是满足条件的，如果不存在一个k使得sum[k]%n==0，那么代表他只能等于1~n-1这n-1个数，但是我们前缀和其实一共有n个前缀和，那么根据抽屉原理一定有一对前缀和%n是相同的，那么假设是l,r，则我们能够知道(sum[r]-sum[l]) %n一定是等于0的，所以我们找到了一个合法的方案

#include<iostream>
#include<cstring>
#include<string>
#include<sstream>
#include<cmath>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<vector> 
#include<set>
#include<unordered_map>
#include<ctime>
#include<cstdlib>
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> PII;
typedef pair<int,pair<int,int> > PIII;
const double eps=1e-7;
const int N=5e5+7 ,M=5e5+7, INF=0x3f3f3f3f,mod=1e9+7;
const long long int llINF=0x3f3f3f3f3f3f3f3f;
inline ll read() {ll x=0,f=1;char c=getchar();while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') {x=(ll)x*10+c-'0';c=getchar();} return x*f;}
inline void write(ll x) {if(x < 0) {putchar('-'); x = -x;}if(x >= 10) write(x / 10);putchar(x % 10 + '0');}
inline void write(ll x,char ch) {write(x);putchar(ch);}
void stin() {freopen("in_put.txt","r",stdin);freopen("my_out_put.txt","w",stdout);}
bool cmp0(int a,int b) {return a>b;}
template<typename T> T gcd(T a,T b) {return b==0?a:gcd(b,a%b);}
template<typename T> T lcm(T a,T b) {return a*b/gcd(a,b);}
void hack() {printf("\n----------------------------------\n");}

int T,hackT;
int n,m,k;
int w[N];
ll sum[N];
int st[N];

void solve() {
	n=read();
	
	for(int i=1;i<=n;i++) w[i]=read();
	
	for(int i=1;i<=n;i++) sum[i]=sum[i-1]+w[i],sum[i]%=n;
	
	for(int i=1;i<=n;i++) {
		if(sum[i]==0) {
			printf("%d\n",i);
			for(int j=1;j<=i;j++) printf("%d ",j);
			return ;
		}
	}
	
	for(int i=1;i<=n;i++) {
		if(st[sum[i]]==0) st[sum[i]]=i;
		else {
			printf("%d\n",i-st[sum[i]]);
			for(int l=st[sum[i]]+1;l<=i;l++) printf("%d ",l);
			return ;
		}
	}
	printf("\n");
}   

int main() {
    // init();
    // stin();

    // scanf("%d",&T);
    T=1; 
    while(T--) hackT++,solve();
    
    return 0;       
}